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
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
|
|
__all__ = ['BaseLoader', 'FullLoader', 'SafeLoader', 'Loader', 'UnsafeLoader']
|
|
|
|
from .reader import *
|
|
from .scanner import *
|
|
from .parser import *
|
|
from .composer import *
|
|
from .constructor import *
|
|
from .resolver import *
|
|
|
|
class BaseLoader(Reader, Scanner, Parser, Composer, BaseConstructor, BaseResolver):
|
|
|
|
def __init__(self, stream):
|
|
Reader.__init__(self, stream)
|
|
Scanner.__init__(self)
|
|
Parser.__init__(self)
|
|
Composer.__init__(self)
|
|
BaseConstructor.__init__(self)
|
|
BaseResolver.__init__(self)
|
|
|
|
class FullLoader(Reader, Scanner, Parser, Composer, FullConstructor, Resolver):
|
|
|
|
def __init__(self, stream):
|
|
Reader.__init__(self, stream)
|
|
Scanner.__init__(self)
|
|
Parser.__init__(self)
|
|
Composer.__init__(self)
|
|
FullConstructor.__init__(self)
|
|
Resolver.__init__(self)
|
|
|
|
class SafeLoader(Reader, Scanner, Parser, Composer, SafeConstructor, Resolver):
|
|
|
|
def __init__(self, stream):
|
|
Reader.__init__(self, stream)
|
|
Scanner.__init__(self)
|
|
Parser.__init__(self)
|
|
Composer.__init__(self)
|
|
SafeConstructor.__init__(self)
|
|
Resolver.__init__(self)
|
|
|
|
class Loader(Reader, Scanner, Parser, Composer, Constructor, Resolver):
|
|
|
|
def __init__(self, stream):
|
|
Reader.__init__(self, stream)
|
|
Scanner.__init__(self)
|
|
Parser.__init__(self)
|
|
Composer.__init__(self)
|
|
Constructor.__init__(self)
|
|
Resolver.__init__(self)
|
|
|
|
# UnsafeLoader is the same as Loader (which is and was always unsafe on
|
|
# untrusted input). Use of either Loader or UnsafeLoader should be rare, since
|
|
# FullLoad should be able to load almost all YAML safely. Loader is left intact
|
|
# to ensure backwards compatibility.
|
|
class UnsafeLoader(Reader, Scanner, Parser, Composer, Constructor, Resolver):
|
|
|
|
def __init__(self, stream):
|
|
Reader.__init__(self, stream)
|
|
Scanner.__init__(self)
|
|
Parser.__init__(self)
|
|
Composer.__init__(self)
|
|
Constructor.__init__(self)
|
|
Resolver.__init__(self)
|