def shutdown_session(exception=None): """ Do sth after each request. """ db_session.remove() # get page execute time. req_pass_time = (time.time() - g.req_start_time) * 1000
def get_list_of_scripts_to_run(challenges, num_benign_scripts): # we want to run all the set flags first, then a random mix of benign and get flags set_flag_scripts = [] get_flag_scripts = [] benign_scripts = [] for challenge in challenges: results = Script.query.filter_by(challenge=challenge, is_working=True) logger.debug("Getting scripts for challenge {} --> {}".format( challenge, [r.__str__() for r in results])) benigns = [] for result in results: type = str(result.type) if type == 'ScriptType.getflag': get_flag_scripts.append(result.id) elif type == 'ScriptType.setflag': set_flag_scripts.append(result.id) elif type == 'ScriptType.benign': benigns.append(result.id) elif type == 'ScriptType.exploit': assert False, "In this version, we should never be running exploits" for i in range(num_benign_scripts): if benigns: benign_script = random.choice(benigns) benign_scripts.append(benign_script) logger.debug("Benign Scripts: {}".format(benign_scripts)) logger.debug("Getflag Scripts: {}".format(get_flag_scripts)) logger.debug("Setflag Scripts: {}".format(set_flag_scripts)) random.shuffle(set_flag_scripts) other_scripts = [*get_flag_scripts, *benign_scripts] # combine lists random.shuffle(other_scripts) db_session.remove() return [*set_flag_scripts, *other_scripts]
def shutdown_session(exception=None): db_session.remove()
def shutdown_session(response): db_session.remove() return response
def close_session(*args, **kwargs): # Flask SQLAlchemy will automatically create new sessions for you from # a scoped session factory, given that we are maintaining the same app # context, this ensures tasks have a fresh session (e.g. session errors # won't propagate across tasks) db.remove()
def game_ad_running(): event = Event.query.order_by(Event.id.desc()).first() db_session.remove() # session.expire_all() # XXX return event.attack_defense_start < datetime.now() < event.end
def tick_exists(): db_session.remove() tick = Tick.query.filter_by( event=Event.query.order_by(Event.id.desc()).first()).first() return True if tick is not None else False