from __future__ import annotations

from sqlalchemy import Column, DateTime, Float, Integer, String
from sqlalchemy.orm import declarative_base

Base = declarative_base()


class Session(Base):
    __tablename__ = "sessions"

    session_key = Column(Integer, primary_key=True)
    meeting_key = Column(Integer, index=True)
    session_name = Column(String(64))
    session_type = Column(String(32))
    country_name = Column(String(64))
    date_start = Column(DateTime(timezone=True))
    date_end = Column(DateTime(timezone=True))


class CarData(Base):
    __tablename__ = "car_data"

    id = Column(Integer, primary_key=True, autoincrement=True)
    session_key = Column(Integer, index=True)
    driver_number = Column(Integer, index=True)
    date = Column(DateTime(timezone=True))
    speed = Column(Float)
    throttle = Column(Float)
    brake = Column(Float)
    n_gear = Column(Integer)