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
104 lines
2.6 KiB
Python
104 lines
2.6 KiB
Python
#
|
|
# The Python Imaging Library.
|
|
# $Id$
|
|
#
|
|
# IM Tools support for PIL
|
|
#
|
|
# history:
|
|
# 1996-05-27 fl Created (read 8-bit images only)
|
|
# 2001-02-17 fl Use 're' instead of 'regex' (Python 2.1) (0.2)
|
|
#
|
|
# Copyright (c) Secret Labs AB 1997-2001.
|
|
# Copyright (c) Fredrik Lundh 1996-2001.
|
|
#
|
|
# See the README file for information on usage and redistribution.
|
|
#
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
|
|
from . import Image, ImageFile
|
|
|
|
#
|
|
# --------------------------------------------------------------------
|
|
|
|
field = re.compile(rb"([a-z]*) ([^ \r\n]*)")
|
|
|
|
|
|
##
|
|
# Image plugin for IM Tools images.
|
|
|
|
|
|
class ImtImageFile(ImageFile.ImageFile):
|
|
format = "IMT"
|
|
format_description = "IM Tools"
|
|
|
|
def _open(self) -> None:
|
|
# Quick rejection: if there's not a LF among the first
|
|
# 100 bytes, this is (probably) not a text header.
|
|
|
|
assert self.fp is not None
|
|
|
|
buffer = self.fp.read(100)
|
|
if b"\n" not in buffer:
|
|
msg = "not an IM file"
|
|
raise SyntaxError(msg)
|
|
|
|
xsize = ysize = 0
|
|
|
|
while True:
|
|
if buffer:
|
|
s = buffer[:1]
|
|
buffer = buffer[1:]
|
|
else:
|
|
s = self.fp.read(1)
|
|
if not s:
|
|
break
|
|
|
|
if s == b"\x0c":
|
|
# image data begins
|
|
self.tile = [
|
|
ImageFile._Tile(
|
|
"raw",
|
|
(0, 0) + self.size,
|
|
self.fp.tell() - len(buffer),
|
|
self.mode,
|
|
)
|
|
]
|
|
|
|
break
|
|
|
|
else:
|
|
# read key/value pair
|
|
if b"\n" not in buffer:
|
|
buffer += self.fp.read(100)
|
|
lines = buffer.split(b"\n")
|
|
s += lines.pop(0)
|
|
buffer = b"\n".join(lines)
|
|
if len(s) == 1 or len(s) > 100:
|
|
break
|
|
if s[0] == ord(b"*"):
|
|
continue # comment
|
|
|
|
m = field.match(s)
|
|
if not m:
|
|
break
|
|
k, v = m.group(1, 2)
|
|
if k == b"width":
|
|
xsize = int(v)
|
|
self._size = xsize, ysize
|
|
elif k == b"height":
|
|
ysize = int(v)
|
|
self._size = xsize, ysize
|
|
elif k == b"pixel" and v == b"n8":
|
|
self._mode = "L"
|
|
|
|
|
|
#
|
|
# --------------------------------------------------------------------
|
|
|
|
Image.register_open(ImtImageFile.format, ImtImageFile)
|
|
|
|
#
|
|
# no extension registered (".im" is simply too common)
|