from __future__ import annotations

from typing import Optional
from datetime import datetime
from sqlmodel import SQLModel, Field


class SessionBase(SQLModel):
    start_time: datetime
    end_time: Optional[datetime] = None


class Session(SessionBase, table=True):
    __tablename__ = "sessions"

    session_id: Optional[int] = Field(default=None, primary_key=True)


class SessionCreate(SessionBase):
    pass