31 lines
888 B
Python
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("アクティブセッションをリセットしました")
|