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