class BaseHandler(tornado.web.RequestHandler):
def prepare(self):
self.db_session = Session()
def on_finish(self):
self.db_session.remove()
def get_current_user(self):
pass
def save_session(self, app, sdata, response):
cookie_domain = self.get_cookie_domain(app)
sess = Session.query.get(sdata.sid)
if not sdata:
# Session has become empty (i.e. `if not some_dict`), so drop it
if sess:
db.session.delete(sess)
else:
if not sess:
sess = Session(sdata.sid)
db.session.add(sess)
sess.data = pickle.dumps(dict(sdata), protocol=2)
sess.last_access = datetime.now()
# Expire old(er than 30 days) sessions
Session.query.filter(Session.last_access < (datetime.now() - timedelta(days=30))).delete()
db.session.commit()
response.set_cookie(app.session_cookie_name, sess.uuid,
expires=(datetime.now() + timedelta(days=30)),
domain=cookie_domain, httponly=True)
def generate_inv_code():
sess = Session()
inv_code = util.generate_inv_code()
while sess.query(InvCode).filter(InvCode.code == inv_code).first():
inv_code = util.generate_inv_code()
sess.add(InvCode(inv_code))
sess.commit()
return inv_code
def setUp(self):
"""
Will be called before every test
"""
# Create table
db.create_all()
#Create Test registree
name = Workout_plan(name='Sob')
session = Session(pushups='2', comment='i did this in 10seconds')
#Save workout plan & session to the database
db.session.add(name)
db.session.add(session)
db.session.commit()
def prepare(self):
self.db_session = Session()
