def create_post(category_id, title, summary, path, create_time, user_id):
import datetime
now = datetime.datetime.now()
print now
get_session().add(Post(
category_id=category_id, title=title, summary=summary, path=path, create_time=now, user_id=user_id))
get_session().flush()
def find_posts_by_category_id(category_id, start=None, stop=None):
if start is not None and stop is not None:
return get_session().query(Post).filter_by(category_id=category_id).\
order_by(Post.create_time.desc()).slice(start, stop).all()
else:
return get_session().query(Post).filter_by(category_id=category_id).\
order_by(Post.create_time.desc()).all()
def wrapper(*args, **kwargs):
try:
results = func(*args, **kwargs)
return results
except Exception as e:
logging.error(e)
get_session().rollback()
args[0].render('error.html')
raise
finally:
get_session().commit()
def find_category_by_id(category_id):
return get_session().query(Category).filter_by(category_id=category_id).one()
def find_all_categories():
return get_session().query(Category).all()
def __init__(self, *args, **kwargs):
super(BaseHandler, self).__init__(*args, **kwargs)
self.session = session.Session(self.application.session_manager, self)
self.db_session = get_session()
def find_user_by_id(user_id):
return get_session().query(User).filter_by(user_id=user_id).one()
def create_user(user):
return get_session().add(user)
def find_all_users():
return get_session().query(User).all()
def find_posts_count(category_id=None):
conditions = dict()
if category_id:
conditions['category_id'] = category_id
return get_session().query(Post).filter_by(**conditions).count()
def update_user(user):
u = get_session().query(User).filter_by(username=user.username).one()
if u:
u.password = user.password
def find_all_users():
return get_session().query(User).all()
if __name__ == '__main__':
# user = find_user_by_id(1)
# print user.username
# user = User(username='******', password='******')
# create_user(user)
# get_session().commit()
# users = find_all_users()
# print users
user = find_user_by_username('admin')
print user.password
user.password = '******'
get_session().commit()
# update_user()
print find_user_by_username('admin').password
def find_all_posts(start=None, stop=None):
if start is not None and stop is not None:
return get_session().query(Post).order_by(Post.create_time.desc()).slice(start, stop).all()
else:
return get_session().query(Post).order_by(Post.create_time.desc()).all()
def delete_post_by_post_id(post_id):
get_session().delete(find_post_by_post_id(post_id))
return
def find_post_by_post_id(post_id):
return get_session().query(Post).filter_by(post_id=post_id).first()
def delete_category_by_id(category_id):
return get_session().delete(find_category_by_id(category_id))
def create_category(categoryname):
return get_session().add(Category(categoryname=categoryname))
def find_user_by_username(username):
return get_session().query(User).filter_by(username=username).first()
