fastapi_app/app/models.py
2025-12-10 22:59:09 +00:00

57 lines
2 KiB
Python

from sqlalchemy import Column, Integer, String, Float, Date, ForeignKey
from sqlalchemy.orm import relationship
from app.database import Base
class Customer(Base):
__tablename__ = "customers"
customer_id = Column(String(20), primary_key=True, index=True)
full_name = Column(String(100), index=True)
email = Column(String(100), unique=True)
phone = Column(String(50))
date_of_birth = Column(Date)
age = Column(Integer)
gender = Column(String(20))
street_address = Column(String(100))
city = Column(String(50))
state = Column(String(10))
zip_code = Column(String(20))
home_branch_id = Column(Integer)
customer_since = Column(Date)
employment_status = Column(String(50))
annual_income = Column(Float)
credit_score = Column(Integer)
preferred_contact_method = Column(String(20))
accounts = relationship("Account", back_populates="customer", cascade="all, delete-orphan")
class Account(Base):
__tablename__ = "accounts"
account_id = Column(String(20), primary_key=True, index=True)
account_number = Column(String(20), unique=True, index=True)
customer_id = Column(String(20), ForeignKey("customers.customer_id"))
account_type = Column(String(20))
open_date = Column(Date)
balance = Column(Float)
branch_id = Column(Integer)
customer = relationship("Customer", back_populates="accounts")
transactions = relationship("Transaction", back_populates="account", cascade="all, delete-orphan")
class Transaction(Base):
__tablename__ = "transactions"
transaction_id = Column(String(30), primary_key=True, index=True)
account_id = Column(String(20), ForeignKey("accounts.account_id"))
branch_id = Column(Integer)
transaction_type = Column(String(50))
amount = Column(Float)
date = Column(Date)
balance_after = Column(Float)
vendor = Column(String(100))
transaction_location = Column(String(100))
account = relationship("Account", back_populates="transactions")