Upload core/models/case.py with huggingface_hub

core/models/case.py

@@ -8,7 +8,17 @@ fraud case management and investigation.
 import json
 import uuid
 
-from sqlalchemy import JSON, Boolean, Column, DateTime, Float, ForeignKey, Index, Integer, String
+from sqlalchemy import (
+    JSON,
+    Boolean,
+    Column,
+    DateTime,
+    Float,
+    ForeignKey,
+    Index,
+    Integer,
+    String,
+)
 from sqlalchemy.orm import relationship
 
 from .base import Base, CasePriority, CaseStatus, CaseType, EncryptedString, utc_now
@@ -18,7 +28,9 @@ class Case(Base):
     __tablename__ = "cases"
 
     id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()))
-    project_id = Column(String, ForeignKey("projects.id"), index=True, default="default")
+    project_id = Column(
+        String, ForeignKey("projects.id"), index=True, default="default"
+    )
     title = Column(String, nullable=False, index=True)
     description = Column(EncryptedString)
     status = Column(String, default=CaseStatus.OPEN, index=True)
@@ -43,13 +55,27 @@ class Case(Base):
     project = relationship("Project", back_populates="cases")
     assignee = relationship("User", back_populates="cases", foreign_keys=[assignee_id])
     creator = relationship("User", foreign_keys=[created_by])
-    transactions = relationship("Transaction", back_populates="case", cascade="all, delete-orphan")
-    evidence = relationship("Evidence", back_populates="case", cascade="all, delete-orphan")
-    notes = relationship("CaseNote", back_populates="case", cascade="all, delete-orphan")
-    activities = relationship("CaseActivity", back_populates="case", cascade="all, delete-orphan")
-    alerts = relationship("FraudAlert", back_populates="case", cascade="all, delete-orphan")
-    trade_transactions = relationship("TradeTransaction", back_populates="case", cascade="all, delete-orphan")
-    crypto_transactions = relationship("CryptoTransaction", back_populates="case", cascade="all, delete-orphan")
+    transactions = relationship(
+        "Transaction", back_populates="case", cascade="all, delete-orphan"
+    )
+    evidence = relationship(
+        "Evidence", back_populates="case", cascade="all, delete-orphan"
+    )
+    notes = relationship(
+        "CaseNote", back_populates="case", cascade="all, delete-orphan"
+    )
+    activities = relationship(
+        "CaseActivity", back_populates="case", cascade="all, delete-orphan"
+    )
+    alerts = relationship(
+        "FraudAlert", back_populates="case", cascade="all, delete-orphan"
+    )
+    trade_transactions = relationship(
+        "TradeTransaction", back_populates="case", cascade="all, delete-orphan"
+    )
+    crypto_transactions = relationship(
+        "CryptoTransaction", back_populates="case", cascade="all, delete-orphan"
+    )
 
 
 class Transaction(Base):
@@ -77,7 +103,9 @@ class Transaction(Base):
     # Relationships
     case = relationship("Case", back_populates="transactions")
 
-    __table_args__ = (Index("idx_transactions_case_date_amount", "case_id", "date", "amount"),)
+    __table_args__ = (
+        Index("idx_transactions_case_date_amount", "case_id", "date", "amount"),
+    )
 
 
 class Evidence(Base):
@@ -98,7 +126,9 @@ class Evidence(Base):
     uploaded_by = Column(String, index=True)
 
     # Processing information
-    processing_status = Column(String, default="pending", index=True)  # pending, processing, completed, failed
+    processing_status = Column(
+        String, default="pending", index=True
+    )  # pending, processing, completed, failed
     processed_at = Column(DateTime, index=True)
 
     # Content analysis
@@ -144,7 +174,9 @@ class CaseActivity(Base):
     id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()))
     case_id = Column(String, ForeignKey("cases.id"), index=True)
     user_id = Column(String, ForeignKey("users.id"), index=True)
-    activity_type = Column(String, nullable=False, index=True)  # created, updated, viewed, etc.
+    activity_type = Column(
+        String, nullable=False, index=True
+    )  # created, updated, viewed, etc.
     description = Column(String, nullable=False)
     activity_metadata = Column(JSON, default=dict)
     created_at = Column(DateTime, default=utc_now, index=True)
@@ -160,7 +192,9 @@ class FraudAlert(Base):
     id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()))
     case_id = Column(String, ForeignKey("cases.id"), index=True)
     alert_type = Column(String, nullable=False, index=True)
-    severity = Column(String, default="medium", index=True)  # low, medium, high, critical
+    severity = Column(
+        String, default="medium", index=True
+    )  # low, medium, high, critical
     title = Column(String, nullable=False)
     description = Column(EncryptedString)
     alert_metadata = Column(JSON, default=dict)
@@ -229,7 +263,9 @@ class EvidenceChain(Base):
     timestamp = Column(DateTime, default=utc_now, index=True)
     chain_metadata = Column(EncryptedString, default="{}")
 
-    __table_args__ = (Index("idx_evidence_package_timestamp", "package_id", "timestamp"),)
+    __table_args__ = (
+        Index("idx_evidence_package_timestamp", "package_id", "timestamp"),
+    )
 
 
 __all__ = [