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 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(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
def shutdown_session(response):
db_session.remove()
return response
