# ================================================================
# services/data/app/models/traffic.py
# ================================================================
"""Traffic data models"""

from sqlalchemy import Column, String, DateTime, Float, Integer, Text, Index
from sqlalchemy.dialects.postgresql import UUID
import uuid
from datetime import datetime, timezone

from shared.database.base import Base

class TrafficData(Base):
    __tablename__ = "traffic_data"
    
    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    location_id = Column(String(100), nullable=False, index=True)
    date = Column(DateTime(timezone=True), nullable=False, index=True)
    traffic_volume = Column(Integer, nullable=True)  # vehicles per hour
    pedestrian_count = Column(Integer, nullable=True)  # pedestrians per hour
    congestion_level = Column(String(20), nullable=True)  # low/medium/high
    average_speed = Column(Float, nullable=True)  # km/h
    source = Column(String(50), nullable=False, default="madrid_opendata")
    raw_data = Column(Text, nullable=True)
    created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
    updated_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc), onupdate=lambda: datetime.now(timezone.utc))
    
    __table_args__ = (
        Index('idx_traffic_location_date', 'location_id', 'date'),
    )