Spaces:
Running on Zero
Running on Zero
| """ | |
| Document model for file management | |
| """ | |
| from sqlalchemy import Column, Integer, String, DateTime, Text, ForeignKey, Boolean, Enum as SQLEnum | |
| from sqlalchemy.orm import relationship | |
| from datetime import datetime | |
| import enum | |
| from app.core.database import Base | |
| class DocumentType(str, enum.Enum): | |
| """Document type - نوع المستند""" | |
| CONTRACT = "contract" # عقد | |
| COURT_FILING = "court_filing" # صحيفة دعوى | |
| EVIDENCE = "evidence" # دليل | |
| CORRESPONDENCE = "correspondence" # مراسلات | |
| POWER_OF_ATTORNEY = "power_of_attorney" # وكالة | |
| ID_DOCUMENT = "id_document" # وثيقة هوية | |
| OTHER = "other" # أخرى | |
| class DocumentStatus(str, enum.Enum): | |
| """Document status - حالة المستند""" | |
| DRAFT = "draft" # مسودة | |
| PENDING_REVIEW = "pending_review" # قيد المراجعة | |
| APPROVED = "approved" # معتمد | |
| ARCHIVED = "archived" # مؤرشف | |
| class Document(Base): | |
| """ | |
| Document model - نموذج المستند | |
| For managing uploaded documents | |
| """ | |
| __tablename__ = "documents" | |
| id = Column(Integer, primary_key=True, index=True) | |
| # Basic info | |
| title = Column(String(500), nullable=False) | |
| title_ar = Column(String(500), nullable=True) | |
| description = Column(Text, nullable=True) | |
| description_ar = Column(Text, nullable=True) | |
| # File info | |
| file_name = Column(String(500), nullable=False) | |
| file_path = Column(String(1000), nullable=False) | |
| file_size = Column(Integer, nullable=True) # Size in bytes | |
| file_type = Column(String(50), nullable=True) # MIME type | |
| file_extension = Column(String(10), nullable=True) # .pdf, .docx, etc. | |
| # Classification | |
| document_type = Column(SQLEnum(DocumentType), default=DocumentType.OTHER, nullable=False) | |
| status = Column(SQLEnum(DocumentStatus), default=DocumentStatus.DRAFT, nullable=False) | |
| # Related entities | |
| client_id = Column(Integer, ForeignKey("clients.id"), nullable=True) | |
| case_id = Column(Integer, ForeignKey("cases.id"), nullable=True) | |
| uploaded_by_id = Column(Integer, ForeignKey("users.id"), nullable=False) | |
| # OCR/Extracted text (for search) | |
| extracted_text = Column(Text, nullable=True) | |
| extracted_text_ar = Column(Text, nullable=True) | |
| # Metadata | |
| tags = Column(String(500), nullable=True) | |
| category = Column(String(100), nullable=True) | |
| # Version control | |
| version = Column(Integer, default=1) | |
| is_latest_version = Column(Boolean, default=True) | |
| parent_document_id = Column(Integer, ForeignKey("documents.id"), nullable=True) | |
| # Security | |
| is_confidential = Column(Boolean, default=False) | |
| is_encrypted = Column(Boolean, default=False) | |
| # Timestamps | |
| created_at = Column(DateTime, default=datetime.utcnow, nullable=False) | |
| updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow) | |
| # Relationships | |
| client = relationship("Client", back_populates="documents") | |
| case = relationship("Case", back_populates="documents") | |
| uploaded_by = relationship("User") | |
| versions = relationship("Document", backref="parent", remote_side=[id]) | |
| def __repr__(self): | |
| return f"<Document {self.id}: {self.file_name}>" | |
| def file_size_mb(self) -> float: | |
| """Get file size in MB""" | |
| if self.file_size: | |
| return round(self.file_size / (1024 * 1024), 2) | |
| return 0.0 |