Initial commit: SIBU 2.0 MISSION

backend/app/models/user.py

@@ -0,0 +1,59 @@
+"""User and DriverProfile models."""
+from sqlmodel import SQLModel, Field, Relationship
+from typing import Optional
+from enum import Enum
+from uuid import UUID, uuid4
+from datetime import datetime
+from sqlalchemy import Column, DateTime, func
+
+
+class UserRole(str, Enum):
+    ADMIN = "ADMIN"
+    PASSENGER = "PASSENGER"
+    DRIVER = "DRIVER"
+    PROMOTER = "PROMOTER"
+
+
+class VehicleType(str, Enum):
+    TAXI = "taxi"
+    BUS = "bus"
+
+
+class User(SQLModel, table=True):
+    __tablename__ = "users"
+
+    id: Optional[UUID] = Field(default_factory=uuid4, primary_key=True)
+    email: str = Field(unique=True, index=True)
+    hashed_password: str
+    full_name: str = Field(index=True)
+    role: UserRole = Field(default=UserRole.PASSENGER)
+    is_active: bool = Field(default=True)
+    is_verified: bool = Field(default=False)  # For drivers/admins verification
+    profile_photo_url: Optional[str] = None
+    created_at: Optional[datetime] = Field(
+        sa_column=Column(DateTime, server_default=func.now())
+    )
+    
+    # Relationships
+    driver_profile: Optional["DriverProfile"] = Relationship(
+        back_populates="user", sa_relationship_kwargs={"uselist": False}
+    )
+
+
+class DriverProfile(SQLModel, table=True):
+    __tablename__ = "driver_profiles"
+
+    id: Optional[UUID] = Field(default_factory=uuid4, primary_key=True)
+    user_id: UUID = Field(foreign_key="users.id")
+    cedula: str
+    vehicle_type: VehicleType
+    license_plate: str
+    photo_url: Optional[str] = None
+    vehicle_photo_url: Optional[str] = None
+    cooperative_name: Optional[str] = None  # Specifically for Bus
+    shift: Optional[str] = None  # For Taxi schedules (e.g. "Dia,Noche")
+    payment_methods: Optional[str] = None # e.g. "Efectivo,Yappi"
+    speaks_english: bool = Field(default=False)
+    
+    # Relationship
+    user: User = Relationship(back_populates="driver_profile")