51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
"""Account (wallet) model"""
|
|
|
|
from sqlalchemy import Column, Integer, String, Float, DateTime, Boolean, ForeignKey, Enum
|
|
from sqlalchemy.orm import relationship
|
|
from datetime import datetime
|
|
from enum import Enum as PyEnum
|
|
from app.db.database import Base
|
|
|
|
|
|
class AccountType(str, PyEnum):
|
|
"""Types of accounts"""
|
|
CARD = "card"
|
|
CASH = "cash"
|
|
DEPOSIT = "deposit"
|
|
GOAL = "goal"
|
|
OTHER = "other"
|
|
|
|
|
|
class Account(Base):
|
|
"""Account model - represents a user's wallet or account"""
|
|
|
|
__tablename__ = "accounts"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
family_id = Column(Integer, ForeignKey("families.id"), nullable=False, index=True)
|
|
owner_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
|
|
|
|
name = Column(String(255), nullable=False)
|
|
account_type = Column(Enum(AccountType), default=AccountType.CARD)
|
|
description = Column(String(500), nullable=True)
|
|
|
|
# Balance
|
|
balance = Column(Float, default=0.0)
|
|
initial_balance = Column(Float, default=0.0)
|
|
|
|
# Status
|
|
is_active = Column(Boolean, default=True)
|
|
is_archived = Column(Boolean, default=False)
|
|
|
|
# Timestamps
|
|
created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
|
|
# Relationships
|
|
family = relationship("Family", back_populates="accounts")
|
|
owner = relationship("User", back_populates="accounts")
|
|
transactions = relationship("Transaction", back_populates="account")
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<Account(id={self.id}, name={self.name}, balance={self.balance})>"
|