24 lines
1.0 KiB
Python
24 lines
1.0 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
from sqlalchemy import Column, String, Integer, DateTime, Numeric, ForeignKey
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import relationship
|
|
from app.database import Base
|
|
|
|
class Bankroll(Base):
|
|
__tablename__ = "bankrolls"
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"), unique=True, nullable=False)
|
|
monthly_budget = Column(Numeric(10, 2), nullable=False, default=500)
|
|
weekly_limit = Column(Numeric(10, 2), default=150)
|
|
daily_limit = Column(Numeric(10, 2), default=50)
|
|
bet_max_pct = Column(Numeric(5, 2), default=5.0)
|
|
current_balance = Column(Numeric(10, 2), default=0)
|
|
month_spent = Column(Numeric(10, 2), default=0)
|
|
week_spent = Column(Numeric(10, 2), default=0)
|
|
day_spent = Column(Numeric(10, 2), default=0)
|
|
reset_day = Column(Integer, default=1)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
user = relationship("User", back_populates="bankroll")
|