discord_intro_quiz_bot/web/db.py
2024-08-03 19:29:46 +09:00

31 lines
888 B
Python

import os
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, scoped_session
from flask import g, current_app
from models import Base, Account, ActiveSession, ChannelId, QuizMaster
def get_engine():
return create_engine(f"sqlite:///{current_app.config['DATABASE']}")
def get_db():
if 'db' not in g:
engine = get_engine()
g.db = scoped_session(sessionmaker(bind=engine))
return g.db
def close_connection(exception):
db = g.pop('db', None)
if db is not None:
db.remove()
def initialize_db(logger):
engine = get_engine()
Base.metadata.create_all(engine)
logger.info("データベースの初期化が完了しました")
def reset_active_sessions(logger):
db = get_db()
db.query(ActiveSession).delete()
db.commit()
logger.info("アクティブセッションをリセットしました")