def init_app():
global _db, _logs_dir, _k8s, _scheduler
logger.info('configuration dump\n%s', config.dump_yaml())
if config.httpdb.db_type == 'sqldb':
logger.info('using SQLDB')
_db = SQLDB(config.httpdb.dsn)
else:
logger.info('using FileRunDB')
_db = FileRunDB(config.httpdb.dirpath)
_db.connect()
_logs_dir = Path(config.httpdb.logs_path)
try:
_k8s = K8sHelper()
except Exception:
pass
# @yaronha - Initialize here
task = periodic.Task()
periodic.schedule(task, 60)
_scheduler = Scheduler()
for data in _db.list_schedules():
if 'schedule' not in data:
logger.warning('bad scheduler data - %s', data)
continue
_submit(data)
def get_run_db_instance(db_session: Session):
db = get_db()
if isinstance(db, SQLDB):
run_db = SQLRunDB(db.dsn, db_session)
else:
run_db = db.db
run_db.connect()
return run_db
def get_run_db_instance(db_session: Session,
leader_session: typing.Optional[str] = None):
db = get_db()
if isinstance(db, SQLDB):
run_db = SQLRunDB(db.dsn, db_session, leader_session)
else:
run_db = db.db
run_db.connect()
return run_db
