import os
from sqlalchemy import create_engine, Column, Integer, String, DateTime, Boolean
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

Base = declarative_base()

class DBGroupInstance(Base):
    __tablename__ = "GroupInstances"
    object_num = Column(Integer, primary_key=True, autoincrement=True)
    timestamp = Column(DateTime, nullable=False)
    instance_id = Column(String, nullable=False)
    world_name = Column(String, nullable=False)
    member_count = Column(Integer, nullable=False)
    delivered = Column(Boolean, nullable=False)

def get_session():
    sqlite_file = os.getenv("SQLITE_FILE")
    if not sqlite_file:
        raise Exception("SQLite File not defined!")
    database_uri = f'sqlite:///{sqlite_file}'
    engine = create_engine(database_uri)
    Base.metadata.create_all(engine)
    Session = sessionmaker(bind=engine)
    return Session()