From 5841fe78cc562d5bb3e93177bfeced6b9f90b50c Mon Sep 17 00:00:00 2001
From: Backend IM Bot <bot@backend.im>
Date: Wed, 26 Mar 2025 13:03:24 +0000
Subject: [PATCH] Add User model

---
 models/user.py | 34 +++++++++++++++++++++++++++-------
 1 file changed, 27 insertions(+), 7 deletions(-)

diff --git a/models/user.py b/models/user.py
index faf81eb..bd6bcbb 100644
--- a/models/user.py
+++ b/models/user.py
@@ -1,11 +1,31 @@
-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)
+    company_name = Column(String, nullable=True)
+    role = Column(String, nullable=False, default="user")
+    phone = Column(String, nullable=True)
+    address = Column(String, nullable=True)
+    is_active = Column(Boolean, default=True)
+    
+    # Balance and transactions
+    account_balance = Column(Float, default=0.0)
+    
+    # Relationships
+    products = relationship("Product", back_populates="owner")
+    orders = relationship("Order", back_populates="user")
+    transactions = relationship("Transaction", back_populates="user")
+    
+    # Timestamps
+    created_at = Column(DateTime, default=func.now())
+    updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
\ No newline at end of file