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