26 lines
921 B
Python
26 lines
921 B
Python
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()
|