File size: 2,943 Bytes
26c9046 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | """
User models for SAMAAN platform
"""
from sqlalchemy import Column, Integer, String, DateTime, Boolean, Enum, ForeignKey
from sqlalchemy.orm import relationship
from sqlalchemy.sql import func
import enum
from database import Base
class UserRole(str, enum.Enum):
"""User roles enumeration"""
BENEFICIARY = "beneficiary"
STATE_OFFICER = "state_officer"
BANK_OFFICER = "bank_officer"
ADMIN = "admin"
class User(Base):
"""User model for authentication"""
__tablename__ = "users"
id = Column(Integer, primary_key=True, index=True)
mobile = Column(String(10), unique=True, index=True, nullable=False)
name = Column(String(100), nullable=False)
password_hash = Column(String(255), nullable=False)
role = Column(Enum(UserRole), nullable=False, default=UserRole.BENEFICIARY)
state = Column(String(50), nullable=True)
district = Column(String(50), nullable=True)
is_active = Column(Boolean, default=True)
created_at = Column(DateTime(timezone=True), server_default=func.now())
last_login = Column(DateTime(timezone=True), nullable=True)
# Relationships
loan_proofs = relationship("LoanProof", foreign_keys="LoanProof.beneficiary_id", back_populates="beneficiary")
credit_scores = relationship("CreditScore", back_populates="beneficiary")
consumption_data = relationship("ConsumptionData", back_populates="beneficiary")
loan_records = relationship("LoanRecord", back_populates="beneficiary")
review_decisions = relationship("LoanProof", foreign_keys="LoanProof.reviewer_id", back_populates="reviewer")
dbt_cases_assigned = relationship("DBTCase", foreign_keys="DBTCase.assigned_officer_id", back_populates="assigned_officer")
notifications = relationship("Notification", back_populates="user", cascade="all, delete-orphan")
class BeneficiaryProfile(Base):
"""Extended profile for beneficiaries"""
__tablename__ = "beneficiary_profiles"
id = Column(Integer, primary_key=True, index=True)
user_id = Column(Integer, ForeignKey("users.id"), unique=True)
aadhaar_last4 = Column(String(4), nullable=True)
caste_category = Column(String(50), nullable=True) # SC/ST/OBC/General
annual_income_estimate = Column(Integer, nullable=True)
family_size = Column(Integer, nullable=True)
occupation = Column(String(100), nullable=True)
bank_account_last4 = Column(String(4), nullable=True)
bank_name = Column(String(100), nullable=True)
ifsc_code = Column(String(11), nullable=True)
address = Column(String(500), nullable=True)
pincode = Column(String(6), nullable=True)
created_at = Column(DateTime(timezone=True), server_default=func.now())
updated_at = Column(DateTime(timezone=True), onupdate=func.now())
# Relationship
user = relationship("User", back_populates="profile")
User.profile = relationship("BeneficiaryProfile", back_populates="user", uselist=False)
|