database.py

from sqlalchemy import create_engine, Column, Integer, String, Date, JSON, DateTime, Enum
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from datetime import datetime
import enum

Base = declarative_base()

class SessionStatus(enum.Enum):
    AVAILABLE = "available"
    BOOKED = "booked" 
    COMPLETED = "completed"
    CANCELLED = "cancelled"

class HostWindow(Base):
    __tablename__ = "host_windows"
    
    id = Column(Integer, primary_key=True)
    host_id = Column(String, nullable=False)  # Discord user ID
    host_name = Column(String, nullable=False)
    date = Column(Date, nullable=False)
    start_hour = Column(Integer, nullable=False)  # 24hr format
    end_hour = Column(Integer, nullable=False)
    created_at = Column(DateTime, default=datetime.utcnow)

class Session(Base):
    __tablename__ = "sessions"
    
    id = Column(Integer, primary_key=True)
    host_id = Column(String, nullable=False)
    host_name = Column(String, nullable=False)
    date = Column(Date, nullable=False)
    start_hour = Column(Integer, nullable=False)
    end_hour = Column(Integer, nullable=False)
    participants = Column(JSON, default=list)  # List of Discord user IDs
    status = Column(Enum(SessionStatus), default=SessionStatus.BOOKED)
    discord_event_id = Column(String, nullable=True)
    created_at = Column(DateTime, default=datetime.utcnow)

from config import settings  # assuming settings has a DB URL like settings.database_url

engine = create_engine(settings.database_url, echo=False, future=True)
SessionLocal = sessionmaker(bind=engine, autocommit=False, autoflush=False)

# --- DB bootstrap (engine, SessionLocal, create_all) ---
from config import settings
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

# If using SQLite, enable check_same_thread=False for use across threads (bot + API)
connect_args = {}
if settings.database_url.startswith("sqlite"):
    connect_args = {"check_same_thread": False}

engine = create_engine(settings.database_url, echo=False, future=True, connect_args=connect_args)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

# Create tables if they don't exist
Base.metadata.create_all(bind=engine)