From befcce0c27180536f95a5737845f97827041a5f5 Mon Sep 17 00:00:00 2001 From: Backend IM Bot Date: Wed, 26 Mar 2025 17:09:23 +0000 Subject: [PATCH] Add User model --- models/user.py | 24 +++++++++++++++++------- 1 file changed, 17 insertions(+), 7 deletions(-) diff --git a/models/user.py b/models/user.py index faf81eb..27239b3 100644 --- a/models/user.py +++ b/models/user.py @@ -1,11 +1,21 @@ -from sqlalchemy import Column, String, Boolean +from sqlalchemy import Column, String, Integer, Boolean, DateTime, ForeignKey, Float +from sqlalchemy.orm import relationship +from sqlalchemy.sql import func from core.database import Base +import uuid +from datetime import datetime class User(Base): __tablename__ = "users" - - id = Column(String, primary_key=True) - username = Column(String, unique=True, index=True) - email = Column(String, unique=True, index=True) - hashed_password = Column(String) - disabled = Column(Boolean, default=False) + id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4())) + username = Column(String, unique=True, nullable=False, index=True) + email = Column(String, unique=True, nullable=False) + password = Column(String, nullable=False) + first_name = Column(String, nullable=True) + last_name = Column(String, nullable=True) + monthly_budget = Column(Float, default=0.0) + currency = Column(String, default="USD") + is_active = Column(Boolean, default=True) + expenses = relationship("Expense", back_populates="user", cascade="all, delete-orphan") + created_at = Column(DateTime, default=func.now()) + updated_at = Column(DateTime, default=func.now(), onupdate=func.now()) \ No newline at end of file