def set_session_budget(budget, session_id, dbsession=DBSESSION):
"""
delete the existing entry for this session_id in the sessionbudget table,
then enter a new row
"""
print("Deleting old budget")
dbsession.query(SessionBudget).filter_by(session_id=session_id).delete()
dbsession.commit()
print("Setting budget for {} to {}".format(session_id, budget))
sb = SessionBudget(session_id=session_id, budget=budget)
dbsession.add(sb)
dbsession.commit()
return True
def reset_session_squad(session_id, dbsession=DBSESSION):
"""
remove any rows with the given session ID and add a new budget of
100M
"""
# remove all players with this session id
dbsession.query(SessionSquad).filter_by(session_id=session_id).delete()
dbsession.commit()
# now remove the budget, and make a new one
dbsession.query(SessionBudget).filter_by(session_id=session_id).delete()
dbsession.commit()
sb = SessionBudget(session_id=session_id, budget=1000)
dbsession.add(sb)
dbsession.commit()
return True