31 lines
1.4 KiB
Python
31 lines
1.4 KiB
Python
# ================================================================
|
|
# 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'),
|
|
)
|