Daniel Bedeleanu
ea49cd6e4a
- Create backend/services/image_storage.py with 4 core functions:
- sanitize_filename(): remove unsafe chars, limit to 255 chars, convert to lowercase
- get_unique_filename(): handle collisions with UUID suffix (format: {name}_{uuid8}_{variant}.jpg)
- ensure_image_directories(): create /images/ root and category subdirs on startup
- save_image(): save bytes to /images/{category}/{filename}, returns relative path
- Create comprehensive test suite (22 tests) covering all functionality
- Integrate ensure_image_directories() into FastAPI startup event
- Directory structure: /images/{category}/{filename}
- Collision handling: auto-suffix with UUID if filename exists
- All tests passing, pathlib.Path for safe operations
62 lines
1.4 KiB
Python
62 lines
1.4 KiB
Python
#
|
|
# The Python Imaging Library.
|
|
# $Id$
|
|
#
|
|
# read files from within a tar file
|
|
#
|
|
# History:
|
|
# 95-06-18 fl Created
|
|
# 96-05-28 fl Open files in binary mode
|
|
#
|
|
# Copyright (c) Secret Labs AB 1997.
|
|
# Copyright (c) Fredrik Lundh 1995-96.
|
|
#
|
|
# See the README file for information on usage and redistribution.
|
|
#
|
|
from __future__ import annotations
|
|
|
|
import io
|
|
|
|
from . import ContainerIO
|
|
|
|
|
|
class TarIO(ContainerIO.ContainerIO[bytes]):
|
|
"""A file object that provides read access to a given member of a TAR file."""
|
|
|
|
def __init__(self, tarfile: str, file: str) -> None:
|
|
"""
|
|
Create file object.
|
|
|
|
:param tarfile: Name of TAR file.
|
|
:param file: Name of member file.
|
|
"""
|
|
self.fh = open(tarfile, "rb")
|
|
|
|
while True:
|
|
s = self.fh.read(512)
|
|
if len(s) != 512:
|
|
self.fh.close()
|
|
|
|
msg = "unexpected end of tar file"
|
|
raise OSError(msg)
|
|
|
|
name = s[:100].decode("utf-8")
|
|
i = name.find("\0")
|
|
if i == 0:
|
|
self.fh.close()
|
|
|
|
msg = "cannot find subfile"
|
|
raise OSError(msg)
|
|
if i > 0:
|
|
name = name[:i]
|
|
|
|
size = int(s[124:135], 8)
|
|
|
|
if file == name:
|
|
break
|
|
|
|
self.fh.seek((size + 511) & (~511), io.SEEK_CUR)
|
|
|
|
# Open region
|
|
super().__init__(self.fh, self.fh.tell(), size)
|