aiteamutils 0.2.160__py3-none-any.whl → 0.2.162__py3-none-any.whl

aiteamutils/base_model.py

@@ -126,15 +126,25 @@ class BaseFileModel(BaseColumn):
     )
     mime_type: Mapped[str] = mapped_column(
         String,
-        nullable=False,
+        nullable=True,
         doc="MIME 타입"
     )
+    mime_type_main: Mapped[str] = mapped_column(
+        String,
+        nullable=True,
+        doc="MIME 타입 주 분류 (예: image, video, application 등)"
+    )
+    mime_type_sub: Mapped[str] = mapped_column(
+        String,
+        nullable=True,
+        doc="MIME 타입 부 분류 (예: jpeg, mp4, pdf 등)"
+    )
     size: Mapped[int] = mapped_column(
-        nullable=False,
+        nullable=True,
         doc="파일 크기(bytes)"
     )
     checksum: Mapped[str] = mapped_column(
         String,
-        nullable=False,
+        nullable=True,
         doc="파일 체크섬"
     )

aiteamutils/base_service.py

@@ -130,6 +130,8 @@ class BaseService(Generic[ModelType]):
                         'original_name': f['original_name'],
                         'storage_path': f['storage_path'],
                         'mime_type': f['mime_type'],
+                        'mime_type_main': f['mime_type_main'],
+                        'mime_type_sub': f['mime_type_sub'],
                         'size': f['size'],
                         'checksum': f['checksum'],
                         'column_name': f['column_name']

aiteamutils/exceptions.py

@@ -77,6 +77,12 @@ class ErrorCode(Enum):
     TOKEN_ERROR = ErrorResponse(5005, "TOKEN_ERROR", 401, "토큰 오류")
     DELETE_ERROR = ErrorResponse(5006, "DELETE_ERROR", 400, "삭제 오류")
 
+    # File 관련 에러: 6000번대
+    FILE_NOT_FOUND = ErrorResponse(6001, "FILE_NOT_FOUND", 404, "파일을 찾을 수 없습니다")
+    FILE_UPLOAD_ERROR = ErrorResponse(6002, "FILE_UPLOAD_ERROR", 500, "파일 업로드 오류")
+    FILE_DOWNLOAD_ERROR = ErrorResponse(6003, "FILE_DOWNLOAD_ERROR", 500, "파일 다운로드 오류")
+    FILE_DELETE_ERROR = ErrorResponse(6004, "FILE_DELETE_ERROR", 500, "파일 삭제 오류")
+    FILE_PROCESSING_ERROR = ErrorResponse(6005, "FILE_PROCESSING_ERROR", 500, "파일 처리 오류")
 
 class CustomException(Exception):
     """사용자 정의 예외 클래스"""

aiteamutils/files.py

@@ -48,6 +48,25 @@ class FileHandler:
                 original_error=e
             )
 
+    @staticmethod
+    def _split_mime_type(mime_type: str) -> Tuple[str, str]:
+        """MIME 타입을 주 타입과 부 타입으로 분리
+        
+        Args:
+            mime_type (str): MIME 타입 문자열 (예: "image/jpeg")
+            
+        Returns:
+            Tuple[str, str]: (주 타입, 부 타입) 튜플
+        """
+        try:
+            if not mime_type or '/' not in mime_type:
+                return 'application', 'octet-stream'
+            main_type, sub_type = mime_type.split('/', 1)
+            return main_type.lower(), sub_type.lower()
+        except Exception as e:
+            logger.error(f"[MIME 타입 분리 실패] {mime_type}: {str(e)}")
+            return 'application', 'octet-stream'
+
     @staticmethod
     async def _calculate_checksum(file: BinaryIO) -> str:
         """파일의 SHA-256 체크섬 계산
@@ -131,10 +150,17 @@ class FileHandler:
             # 파일 포인터 복구
             file.seek(current_position)
             
+            # MIME 타입 처리
+            mime_type = FileHandler._get_mime_type(original_name)
+            mime_type_main, mime_type_sub = FileHandler._split_mime_type(mime_type)
+            logger.info(f"[MIME 타입 분리] {mime_type} -> {mime_type_main}/{mime_type_sub}")
+            
             file_info = {
                 "original_name": original_name,
                 "storage_path": storage_path,
-                "mime_type": FileHandler._get_mime_type(original_name),
+                "mime_type": mime_type,
+                "mime_type_main": mime_type_main,
+                "mime_type_sub": mime_type_sub,
                 "size": os.path.getsize(storage_path),
                 "checksum": checksum,
                 "entity_name": entity_name,
@@ -202,12 +228,12 @@ class FileHandler:
                         text("""
                             INSERT INTO files (
                                 ulid, entity_name, entity_ulid, original_name, storage_path,
-                                mime_type, size, checksum, column_name, created_at, updated_at,
-                                is_deleted
+                                mime_type, mime_type_main, mime_type_sub, size, checksum, 
+                                column_name, created_at, updated_at, is_deleted
                             ) VALUES (
                                 :ulid, :entity_name, :entity_ulid, :original_name, :storage_path,
-                                :mime_type, :size, :checksum, :column_name, :created_at, :updated_at,
-                                :is_deleted
+                                :mime_type, :mime_type_main, :mime_type_sub, :size, :checksum,
+                                :column_name, :created_at, :updated_at, :is_deleted
                             ) RETURNING *
                         """),
                         {
@@ -217,6 +243,8 @@ class FileHandler:
                             "original_name": file_info["original_name"],
                             "storage_path": file_info["storage_path"],
                             "mime_type": file_info["mime_type"],
+                            "mime_type_main": file_info["mime_type_main"],
+                            "mime_type_sub": file_info["mime_type_sub"],
                             "size": file_info["size"],
                             "checksum": file_info["checksum"],
                             "column_name": column_name,
@@ -234,6 +262,8 @@ class FileHandler:
                         "original_name": row.original_name,
                         "storage_path": row.storage_path,
                         "mime_type": row.mime_type,
+                        "mime_type_main": row.mime_type_main,
+                        "mime_type_sub": row.mime_type_sub,
                         "size": row.size,
                         "checksum": row.checksum,
                         "column_name": row.column_name,

aiteamutils/version.py

@@ -1,2 +1,2 @@
 """버전 정보"""
-__version__ = "0.2.160"
+__version__ = "0.2.162"

{aiteamutils-0.2.160.dist-info → aiteamutils-0.2.162.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiteamutils
-Version: 0.2.160
+Version: 0.2.162
 Summary: AI Team Utilities
 Project-URL: Homepage, https://github.com/yourusername/aiteamutils
 Project-URL: Issues, https://github.com/yourusername/aiteamutils/issues

aiteamutils-0.2.162.dist-info/RECORD

@@ -0,0 +1,16 @@
+aiteamutils/__init__.py,sha256=kRBpRjark0M8ZwFfmKiMFol6CbIILN3WE4f6_P6iIq0,1089
+aiteamutils/base_model.py,sha256=Jw5Fyfrtdc_jyg-LmC35c-D6w3-2_dh2CBeiKjM9GPE,4497
+aiteamutils/base_repository.py,sha256=Oy2zE1i5qx60Xf1tnsaKLyFWapiPqt5JH8NejwNrPWg,4647
+aiteamutils/base_service.py,sha256=JIeRtFn1Ll4Qcq3v88pgS9lmEQLQoVyyOKqYR8n02S4,21192
+aiteamutils/cache.py,sha256=07xBGlgAwOTAdY5mnMOQJ5EBxVwe8glVD7DkGEkxCtw,1373
+aiteamutils/config.py,sha256=YdalpJb70-txhGJAS4aaKglEZAFVWgfzw5BXSWpkUz4,3232
+aiteamutils/database.py,sha256=msvBKtxWeQVOo0v2Q9i2azuTNtnUItuNNar52gdRZTo,20418
+aiteamutils/enums.py,sha256=7WLqlcJqQWtETAga2WAxNp3dJTQIAd2TW-4WzkoHHa8,2498
+aiteamutils/exceptions.py,sha256=sgIVulllKMM9InTltaB7VD6i7DiQvCoycexsV-BiIBY,16570
+aiteamutils/files.py,sha256=fxnCu9rErd4vCovMi0jy4adLUiA7rx_q4RdOL4wSgsU,14258
+aiteamutils/security.py,sha256=McUl3t5Z5SyUDVUHymHdDkYyF4YSeg4g9fFMML4W6Kw,11630
+aiteamutils/validators.py,sha256=_WHN6jqJQzKM5uPTg-Da8U2qqevS84XeKMkCCF4C_lY,9591
+aiteamutils/version.py,sha256=vAvqE9V77cH7vOSwbqingOdyP4KA12olJ1Dj_9Br2jQ,43
+aiteamutils-0.2.162.dist-info/METADATA,sha256=LS01DV_fpROE_i2VnNwkeRx0KQ82VBeRAi4XveycukA,1743
+aiteamutils-0.2.162.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aiteamutils-0.2.162.dist-info/RECORD,,

aiteamutils/models.py

@@ -1 +0,0 @@
- 

aiteamutils-0.2.160.dist-info/RECORD

@@ -1,17 +0,0 @@
-aiteamutils/__init__.py,sha256=kRBpRjark0M8ZwFfmKiMFol6CbIILN3WE4f6_P6iIq0,1089
-aiteamutils/base_model.py,sha256=yBZqzTDF9PA4wCAvmYfG12FdVwLtxOEUCcA3z2i6fXU,4176
-aiteamutils/base_repository.py,sha256=Oy2zE1i5qx60Xf1tnsaKLyFWapiPqt5JH8NejwNrPWg,4647
-aiteamutils/base_service.py,sha256=5YSRpulLI-tO6zQ75LBTsyzYoMzayPITWhE5-8rie3s,21068
-aiteamutils/cache.py,sha256=07xBGlgAwOTAdY5mnMOQJ5EBxVwe8glVD7DkGEkxCtw,1373
-aiteamutils/config.py,sha256=YdalpJb70-txhGJAS4aaKglEZAFVWgfzw5BXSWpkUz4,3232
-aiteamutils/database.py,sha256=msvBKtxWeQVOo0v2Q9i2azuTNtnUItuNNar52gdRZTo,20418
-aiteamutils/enums.py,sha256=7WLqlcJqQWtETAga2WAxNp3dJTQIAd2TW-4WzkoHHa8,2498
-aiteamutils/exceptions.py,sha256=pgf3ersezObyl17wAO3I2fb8m9t2OzWDX1mSjwAWm2Y,16035
-aiteamutils/files.py,sha256=Qq2w3VAEOzvsDirYtxRTN48LnIzf4TPUH2LftmyYtQk,12831
-aiteamutils/models.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
-aiteamutils/security.py,sha256=McUl3t5Z5SyUDVUHymHdDkYyF4YSeg4g9fFMML4W6Kw,11630
-aiteamutils/validators.py,sha256=_WHN6jqJQzKM5uPTg-Da8U2qqevS84XeKMkCCF4C_lY,9591
-aiteamutils/version.py,sha256=Dv_RPFPM_ljwt7FQn4GsgrSYOBxg7SL2-aDSf-jrp30,43
-aiteamutils-0.2.160.dist-info/METADATA,sha256=-mvX5XwdZbNh-6odVtUghcR6k9YU_P2eOv8XruEj5Gg,1743
-aiteamutils-0.2.160.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aiteamutils-0.2.160.dist-info/RECORD,,