55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
|
"""
|
||
|
|
Pension account and holding models for retirement pension asset allocation.
|
||
|
|
"""
|
||
|
|
import enum
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
from sqlalchemy import (
|
||
|
|
Column, Integer, String, Numeric, DateTime,
|
||
|
|
Enum as SQLEnum, ForeignKey,
|
||
|
|
)
|
||
|
|
from sqlalchemy.orm import relationship
|
||
|
|
|
||
|
|
from app.core.database import Base
|
||
|
|
|
||
|
|
|
||
|
|
class AccountType(str, enum.Enum):
|
||
|
|
DC = "dc"
|
||
|
|
IRP = "irp"
|
||
|
|
PERSONAL = "personal"
|
||
|
|
|
||
|
|
|
||
|
|
class AssetRiskType(str, enum.Enum):
|
||
|
|
SAFE = "safe"
|
||
|
|
RISKY = "risky"
|
||
|
|
|
||
|
|
|
||
|
|
class PensionAccount(Base):
|
||
|
|
__tablename__ = "pension_accounts"
|
||
|
|
|
||
|
|
id = Column(Integer, primary_key=True, index=True)
|
||
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
|
||
|
|
account_type = Column(SQLEnum(AccountType), nullable=False, index=True)
|
||
|
|
account_name = Column(String(100), nullable=False)
|
||
|
|
total_amount = Column(Numeric(16, 2), nullable=False, default=0)
|
||
|
|
birth_year = Column(Integer, nullable=False)
|
||
|
|
target_retirement_age = Column(Integer, default=60)
|
||
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
||
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||
|
|
|
||
|
|
user = relationship("User", backref="pension_accounts")
|
||
|
|
holdings = relationship("PensionHolding", back_populates="account", cascade="all, delete-orphan")
|
||
|
|
|
||
|
|
|
||
|
|
class PensionHolding(Base):
|
||
|
|
__tablename__ = "pension_holdings"
|
||
|
|
|
||
|
|
id = Column(Integer, primary_key=True, index=True)
|
||
|
|
account_id = Column(Integer, ForeignKey("pension_accounts.id"), nullable=False, index=True)
|
||
|
|
asset_name = Column(String(200), nullable=False)
|
||
|
|
asset_type = Column(SQLEnum(AssetRiskType), nullable=False)
|
||
|
|
amount = Column(Numeric(16, 2), nullable=False, default=0)
|
||
|
|
ratio = Column(Numeric(6, 2), nullable=False, default=0)
|
||
|
|
|
||
|
|
account = relationship("PensionAccount", back_populates="holdings")
|