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()