from sqlalchemy import Boolean, Column, ForeignKey, String, Float, Text
from sqlalchemy.orm import relationship
from .database import Base

class User(Base):
    __tablename__ = "users"

    id = Column(String, primary_key=True, index=True)
    name = Column(String)
    department = Column(String)
    medical_allowance = Column(Float)

    claims = relationship("Claim", back_populates="owner")

class Claim(Base):
    __tablename__ = "claims"

    id = Column(String, primary_key=True, index=True)
    timestamp = Column(String)
    user_id = Column(String, ForeignKey("users.id"))
    amount_spent = Column(Float)
    amount_claimed = Column(Float)
    provider_name = Column(String)
    visit_date = Column(String)
    treatment_type = Column(String)
    cost_center = Column(String)
    declaration_signed = Column(Boolean)
    extraction_data = Column(Text) # Stored as JSON string

    owner = relationship("User", back_populates="claims")