- 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
42 lines
978 B
Python
42 lines
978 B
Python
# This file is being contributed to pyasn1-modules software.
|
|
#
|
|
# Created by Russ Housley
|
|
#
|
|
# Copyright (c) 2020, Vigil Security, LLC
|
|
# License: http://snmplabs.com/pyasn1/license.html
|
|
#
|
|
# HSS/LMS Hash-based Signature Algorithm for CMS
|
|
#
|
|
# ASN.1 source from:
|
|
# https://www.rfc-editor.org/rfc/rfc8708.txt
|
|
|
|
|
|
from pyasn1.type import univ
|
|
|
|
from pyasn1_modules import rfc5280
|
|
|
|
|
|
# Object Identifiers
|
|
|
|
id_alg_hss_lms_hashsig = univ.ObjectIdentifier('1.2.840.113549.1.9.16.3.17')
|
|
|
|
id_alg_mts_hashsig = id_alg_hss_lms_hashsig
|
|
|
|
|
|
# Signature Algorithm Identifier
|
|
|
|
sa_HSS_LMS_HashSig = rfc5280.AlgorithmIdentifier()
|
|
sa_HSS_LMS_HashSig['algorithm'] = id_alg_hss_lms_hashsig
|
|
# sa_HSS_LMS_HashSig['parameters'] is alway absent
|
|
|
|
|
|
# Public Key
|
|
|
|
class HSS_LMS_HashSig_PublicKey(univ.OctetString):
|
|
pass
|
|
|
|
|
|
pk_HSS_LMS_HashSig = rfc5280.SubjectPublicKeyInfo()
|
|
pk_HSS_LMS_HashSig['algorithm'] = sa_HSS_LMS_HashSig
|
|
# pk_HSS_LMS_HashSig['parameters'] CONTAINS a DER-encoded HSS_LMS_HashSig_PublicKey
|