60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
"""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")
|