def get_all(self):
session = database.get_session()
query = session.query(model_topic.TopicModel)
try:
topics = query.all()
return topics
except exc.NoResultFound:
return None
def get_one_by_id(self, id):
session = database.get_session()
query = session.query(model_token.TokenModel).filter_by(id=id)
try:
token = query.one()
return token
except exc.NoResultFound:
return None
def get_all(self):
session = database.get_session()
query = session.query(model_user.UserModel)
try:
users = query.all()
return users
except exc.NoResultFound:
return None
def get_one_by_id(self, id):
session = database.get_session()
query = session.query(model_user.UserModel).filter_by(id=id)
try:
user = query.one()
return user
except exc.NoResultFound:
return None
def get_one_by_id(self, id):
session = database.get_session()
query = session.query(model_category.CategoryModel).filter_by(id=id)
try:
category = query.one()
return category
except exc.NoResultFound:
return None
def get_all(self):
session = database.get_session()
query = session.query(model_category.CategoryModel)
try:
categories = query.all()
return categories
except exc.NoResultFound:
return None
def get_all(self):
session = database.get_session()
query = session.query(model_reply.ReplyModel)
try:
replies = query.all()
return replies
except exc.NoResultFound:
return None
def get_one_by_id(self, id):
session = database.get_session()
query = session.query(model_reply.ReplyModel).filter_by(id=id)
try:
reply = query.one()
return reply
except exc.NoResultFound:
return None
def get_one_by_username(self, username):
session = database.get_session()
query = session.query(
model_user.UserModel).filter_by(username=username)
try:
user = query.first()
return user
except exc.NoResultFound:
return None
def add_one(self, user):
session = database.get_session()
try:
session.add(user)
session.flush()
session.commit()
return user
except Exception:
pass
def get_one_by_name(self, name):
session = database.get_session()
query = session.query(
model_category.CategoryModel).filter_by(name=name)
try:
category = query.first()
return category
except exc.NoResultFound:
return None
def add_one(self, topic):
session = database.get_session()
try:
session.add(topic)
session.flush()
session.commit()
return topic
except Exception:
pass
def get_one_by_title(self, title):
session = database.get_session()
query = session.query(
model_topic.TopicModel).filter_by(title=title)
try:
topic = query.one()
return topic
except exc.NoResultFound:
return None
def add_one(self, reply):
session = database.get_session()
try:
session.add(reply)
session.flush()
session.commit()
return reply
except Exception:
pass
def get_list_by_category_id(self, category_id):
session = database.get_session()
query = session.query(
model_topic.TopicModel).filter_by(category_id=category_id)
try:
topics = query.all()
return topics
except exc.NoResultFound:
return None
def add_one(self, category):
session = database.get_session()
try:
session.add(category)
session.flush()
session.commit()
return category
except Exception:
pass
def get_one_by_username(self, username):
session = database.get_session()
query = session.query(
model_token.TokenModel).filter_by(username=username)
try:
token = query.one()
return token
except exc.NoResultFound:
return None
def delete_topic_by_id(self, id):
session = database.get_session()
query = session.query(model_topic.TopicModel).filter_by(id=id)
try:
topic = query.one()
session.delete(topic)
session.flush()
session.commit()
except exc.NoResultFound:
pass
def delete_category_by_id(self, id):
session = database.get_session()
query = session.query(model_category.CategoryModel).filter_by(id=id)
try:
category = query.one()
session.delete(category)
session.flush()
session.commit()
except exc.NoResultFound:
pass
def add_one_by_user_id(self, user_id):
token = model_token.TokenModel(user_id=user_id)
session = database.get_session()
try:
session.add(token)
session.flush()
session.commit()
return token
except Exception:
pass
def delete_user_by_id(self, id):
session = database.get_session()
query = session.query(model_user.UserModel).filter_by(id=id)
try:
user = query.first()
session.delete(user)
session.flush()
session.commit()
except exc.NoResultFound:
pass
def delete_reply_by_id(self, id):
session = database.get_session()
query = session.query(model_reply.ReplyModel).filter_by(id=id)
try:
reply = query.one()
session.delete(reply)
session.flush()
session.commit()
except exc.NoResultFound:
pass
def update_topic(self, topic):
session = database.get_session()
topic_id = topic.get('id')
query = session.query(model_topic.TopicModel).filter_by(id=topic_id)
try:
query.update(topic)
session.flush()
session.commit()
return topic
except exc.NoResultFound:
pass
def delete_token_by_user_id(self, user_id):
session = database.get_session()
query = session.query(
model_token.TokenModel).filter_by(user_id=user_id)
try:
token = query.first()
session.delete(token)
session.flush()
session.commit()
except exc.NoResultFound:
pass
def update_reply(self, reply):
session = database.get_session()
reply_id = reply.get('id')
query = session.query(model_reply.ReplyModel).filter_by(id=reply_id)
try:
query.update(reply)
session.flush()
session.commit()
return reply
except exc.NoResultFound:
pass
def update_category(self, category):
session = database.get_session()
category_id = category.get('id')
query = session.query(
model_category.CategoryModel).filter_by(id=category_id)
try:
query.update(category)
session.flush()
session.commit()
return category
except exc.NoResultFound:
pass
def get_list_by_topic_id(self, topic_id):
session = database.get_session()
if topic_id:
query = session.query(
model_reply.ReplyModel).filter_by(topic_id=topic_id).order_by(
model_reply.ReplyModel.updated_time.desc())
else:
query = session.query(model_reply.ReplyModel)
try:
replies = query.all()
return replies
except exc.NoResultFound:
return None
def update_token(self, token):
session = database.get_session()
token_id = token.get('id')
query = session.query(model_token.TokenModel).filter_by(id=token_id)
try:
# 'token' is a dict object here, not a model instance.
# update() return a number, not a model instance.
# in add_one(), 'token' stands a model instance.
query.update(token)
session.flush()
session.commit()
return token
except exc.NoResultFound:
pass
def update_user(self, user):
session = database.get_session()
user_id = user.get('id')
query = session.query(model_user.UserModel).filter_by(id=user_id)
try:
# 'user' is a dict object here, not a model instance.
# update() return a number, not a model instance.
# in add_one(), 'user' stands a model instance.
query.update(user)
session.flush()
session.commit()
return user
except exc.NoResultFound:
pass
def get_list_by_flag(self, flag):
session = database.get_session()
model = model_topic.TopicModel
if flag == 'agree':
query = session.query(model).order_by(model.agree.desc()).limit(10)
elif flag == 'latest':
query = session.query(model).order_by(
model.updated_time.desc()).limit(10)
elif flag == 'all':
query = session.query(model).limit(10)
elif flag == 'hot':
query = session.query(model).order_by(
model.disagree.desc()).limit(10)
else:
query = session.query(model)
try:
topics = query.all()
return topics
except exc.NoResultFound:
return None
