miralabs-backend/ai-service/app/models/user_document.py

from typing import Any

from sqlalchemy import JSON, Column, Integer, String
from sqlalchemy.dialects.postgresql import TIMESTAMP

from app.models.base import Base
from app.utils.utility import to_dict


class UserDocument(Base):
    __tablename__ = "UserDocuments"

    userDocumentId = Column(Integer, primary_key=True)
    userId = Column(Integer)
    documentTypeId = Column(Integer)
    documentSubTypeId = Column(Integer)
    documentStatusId = Column(Integer)
    URL = Column(String)
    date = Column(TIMESTAMP)
    applicationId = Column(Integer)
    userProfileId = Column(Integer)
    remarks = Column(String)
    documentName = Column(String)
    documentDetails = Column(JSON)
    documentSize = Column(Integer)
    quoteNumber = Column(String)

    def dict(self) -> dict[str, Any]:
        return to_dict(self)