models.py 948 B

123456789101112131415161718192021222324252627282930
  1. from sqlalchemy import Boolean, Column, ForeignKey, String, Float, Text
  2. from sqlalchemy.orm import relationship
  3. from .database import Base
  4. class User(Base):
  5. __tablename__ = "users"
  6. id = Column(String, primary_key=True, index=True)
  7. name = Column(String)
  8. department = Column(String)
  9. medical_allowance = Column(Float)
  10. claims = relationship("Claim", back_populates="owner")
  11. class Claim(Base):
  12. __tablename__ = "claims"
  13. id = Column(String, primary_key=True, index=True)
  14. timestamp = Column(String)
  15. user_id = Column(String, ForeignKey("users.id"))
  16. amount_spent = Column(Float)
  17. amount_claimed = Column(Float)
  18. provider_name = Column(String)
  19. visit_date = Column(String)
  20. treatment_type = Column(String)
  21. cost_center = Column(String)
  22. declaration_signed = Column(Boolean)
  23. extraction_data = Column(Text) # Stored as JSON string
  24. owner = relationship("User", back_populates="claims")