def get(self, quote_id):
session = Session()
rows = session.query(Quotes).filter_by(id=quote_id).first()
session.delete(rows)
session.commit()
session.close()
self.redirect('/quotes')
def get(self, user_id):
session = Session()
rows = session.query(User).filter_by(id=user_id).first()
session.delete(rows)
session.commit()
session.close()
self.redirect('/administrator')
def add_post(img_url, thumb_url, username):
session = Session()
user = session.query(User).filter_by(username=username).first()
post = Post(image_url=img_url, thumb_url=thumb_url, user_id=user.id)
session.add(post)
session.commit()
post_id = post.id
session.close()
return post_id
class BaseHandler(tornado.web.RequestHandler, SessionMixin):
def get_current_user(self):
return self.session.get('username_session')
def initialize(self):
self.db_session = Session()
self.orm = OrmHandler(self.db_session)
def on_finish(self):
self.db_session.close()
class BaseHandler(tornado.web.RequestHandler, SessionMixin):
def get_current_user(self):
return self.session.get('tudo_user', None)
def prepare(self):
self.db_session = Session()
logger.info('db_session instance {}'.format(self))
self.orm = HandlerORM(self.db_session)
def on_finish(self):
self.db_session.close()
logger.info('db_session close')
def register(username, password1, password2):
ret = {'msg': 'other error', 'user_id': None}
if username and password1 and password2:
if password1 == password2:
session = Session()
user = session.query(User).filter_by(username=username).all()
if user:
msg = 'username is exists'
else:
new_user = User(username=username, password=hash(password1))
session.add(new_user)
session.commit()
ret['user_id'] = new_user.id
session.close()
msg = 'ok'
else:
msg = 'password1 != password2'
else:
msg = 'username or password is empty'
ret['msg'] = msg
return ret
def is_exists(cls, username, password):
session = Session()
ret = session.query(exists().where(
and_(User.name == username, User.password == password))).scalar()
session.close()
return ret
def is_exists(cls, username, password):
session = Session()
ret = session.query(exists().where(and_(cls.username==username, cls.password==password))).scalar()
# ret = bool(user)/
session.close()
return ret
