from sqlalchemy import Column, Integer, String
from sqlalchemy.dialects.postgresql import TIMESTAMP
from typing import Any
from app.utils.utility import to_dict
from sqlalchemy.orm import declarative_base

Base = declarative_base()


class UserDocument(Base):
    __tablename__ = "UserDocuments"

    userDocumentId = Column(Integer, primary_key=True)
    documentTypeId = Column(Integer)
    documentSubTypeId = Column(Integer)
    documentStatusId = Column(Integer)
    URL = Column(String)
    date = Column(TIMESTAMP)
    applicationId = Column(Integer)
    userProfileId = Column(Integer)
    userId = Column(Integer)
    remarks = Column(String)
    documentName = Column(String)
    
    def dict(self) -> dict[str, Any]:
        return to_dict(self)