def updateBlogVisible(blog_id, user_id, visible=0):
db = mysqlModel()
res = db.updateBlogVisible(blog_id=blog_id,
user_id=user_id,
visible=visible)
db.close
return res
def verifyUser(usermail, password):
'''
验证用户，返回用户id
'''
db = mysqlModel()
res = db.verifyUser(usermail=usermail, password=password)
db.close()
return res
def deleteCategory(category_id, user_id):
'''
'''
db = mysqlModel()
res = db.deleteCategory(category_id=category_id, user_id=user_id)
db.close()
return res
def getUserMessage(user_id):
'''
获取用户基础信息
'''
db = mysqlModel()
res = db.getUserMessage(user_id=user_id)
db.close()
return res
def getUserLogo(nickname):
'''
获取用户头像
目前禁用
'''
db = mysqlModel()
res = db.getUserLogo(nickname=nickname)
db.close()
return res
def updateCommentReplyCount(manager_id, comment_id, method='add'):
'''
'''
db = mysqlModel()
res = db.updateCommentReplyCount(manager_id=manager_id,
comment_id=comment_id,
method=method)
db.close()
return res
def addComment(blog_id, user_id, content, manager_id):
'''
'''
db = mysqlModel()
res = db.addComment(blog_id=blog_id,
user_id=user_id,
content=content,
manager_id=manager_id)
db.close()
return res
def deleteComment(comment_id, user_id, blog_id, manager_id):
'''
'''
db = mysqlModel()
res = db.deleteComment(comment_id=comment_id,
user_id=user_id,
blog_id=blog_id,
manager_id=manager_id)
db.close()
return res
def addUser(usermail, password, nickname, picture):
'''
'''
db = mysqlModel()
res = db.addUser(usermail=usermail,
password=password,
nickname=nickname,
picture=picture)
db.close()
return res
def getCategoryList(user_id):
db = mysqlModel()
res = db.getCategories(user_id=user_id)
db.close()
if res is not None:
category_list = []
for r in res:
category_list.append(r['category'])
return category_list
return None
def addBlog(title, cover_img_url, content, category, tag, user_id):
db = mysqlModel()
res = db.addBlog(title=title,
cover_img_url=cover_img_url,
content=content,
category=category,
tag=tag,
user_id=user_id)
db.close()
return res
def getCategories(user_id):
db = mysqlModel()
res = db.getCategories(user_id=user_id)
for r in res:
r['tags'] = []
tags = db.getTag(user_id=user_id, category_id=r['id'])
if tags:
for tag in tags:
r['tags'].append(tag['tag'])
db.close()
return res
def updateManagerMessage(user_id):
'''
更新用户的博客数、阅读数和评论数
建议每日离线更新
:param user_id:
:return:
'''
db = mysqlModel()
res = db.updateManagerMessage(user_id=user_id)
db.close()
return res
def getTagDict(user_id):
db = mysqlModel()
tags = db.getTags(user_id=user_id)
tag_dict = {}
if tags and len(tags) > 0:
for tag in tags:
if tag['category'] in tag_dict.keys():
tag_dict[tag['category']].append(tag['tag'])
else:
tag_dict[tag['category']] = [tag['tag']]
db.close()
return tag_dict
def getReplyList(manager_id, comment_id, sort_by='update_time'):
'''
'''
db = mysqlModel()
res = db.getReplyList(manager_id=manager_id, sort_by=sort_by)
if res is not None:
result = []
for r in res:
if r['comment_id'] == comment_id:
result.append(r)
res = result
db.close()
return res
def updateCategoryTag(category, tags, user_id):
db = mysqlModel()
if db.updateCategoryCount(category=category, user_id=user_id):
tag_list = tags.split(',')
for tag in tag_list:
if len(tag) <= 0:
continue
if not db.updateTagCount(
category=category, tag=tag, user_id=user_id):
db.close()
return False
db.close()
return True
db.close()
return False
def getComments(manager_id, blog_id, sort_by='update_time'):
'''
'''
db = mysqlModel()
res = db.getComments(manager_id=manager_id, sort_by=sort_by)
if res is not None:
result = []
for r in res:
if r['blog_id'] == blog_id:
reply_list = getReplyList(manager_id=manager_id,
sort_by=sort_by,
comment_id=r['id'])
r['reply_list'] = reply_list
result.append(r)
res = result
db.close()
return res
def addReply(comment_id,
user_id,
content,
manager_id,
blog_id,
replied_id=None,
replied_user_id=None):
'''
'''
db = mysqlModel()
res = db.addReply(comment_id=comment_id,
user_id=user_id,
content=content,
manager_id=manager_id,
blog_id=blog_id,
replied_id=replied_id,
replied_user_id=replied_user_id)
db.close()
return res
def getReplyList(manager_id,
sort_by='update_time',
blog_title=None,
comment_id=None):
'''
'''
db = mysqlModel()
res = db.getReplyList(manager_id=manager_id, sort_by=sort_by)
if blog_title:
blog_titles = blog_title.split(' ')
result = []
for r in res:
if any([i in r['title'] for i in blog_titles]):
result.append(r)
res = result
if comment_id:
ids = [int(i) for i in comment_id.split(' ')]
result = []
for r in res:
if r['comment_id'] in ids:
result.append(r)
res = result
db.close()
return res
#coding:utf-8
import sys
sys.path.append('../')
from tools.mysql_tools.mysql_models import mysqlModel
from tools.mysql_tools.mysql_tools import *
if __name__ == '__main__':
client = mysqlModel()
# print(client.getTags(user_id=1))
# print(client.addTag(category='技术', tag='django'))
# print(client.addCategory('动漫'))
# print(client.addBlog(title='第一条博客', cover_img_url='http://127.0.0.1:5000/show/picture/?filename=dongman_1.jpg', content='# 第一条博客\n### 用于测试', category='生活', tag='随记'))
# print(client.updateCategoryCount(category='生活'))
# print(client.updateTagCount(category='生活', tag='随记', user_id=1))
# print(client.updateManagerMessage(user_id=1))
# print(client.getComments(manager_id=1))
# print(client.updateBlogCommentCount(manager_id=1, blog_id=7))
print(client.getReplyList(manager_id=1))
# print(client.deleteReply(reply_id=4, comment_id=1, user_id=4, manager_id=1, blog_id=2))
# print(getCategoryList(user_id=1))
# print(updateAllCategories(user_id=1))
# print(getTagDict(user_id=1))
# print(updateAllTags(user_id=1))
def getBlog(blog_id, user_id):
db = mysqlModel()
res = db.getBlog(blog_id=blog_id, user_id=user_id)
db.close()
return res
def getTags(user_id):
db = mysqlModel()
res = db.getTags(user_id=user_id)
db.close()
return res
def verifyManager(username, password):
db = mysqlModel()
res = db.verifyManager(username=username, password=password)
db.close()
return res
def getCategories(user_id):
db = mysqlModel()
res = db.getCategories(user_id=user_id)
db.close()
return res
def addCategory(category, user_id):
db = mysqlModel()
res = db.addCategory(category=category, user_id=user_id)
db.close()
return res
def verifyMailExist(usermail):
db = mysqlModel()
res = db.verifyMailExist(usermail=usermail)
db.close()
return res
def updateCategory(category, user_id):
db = mysqlModel()
res = db.updateCategoryCount(category=category, user_id=user_id)
db.close()
return res
def updateBlogRead(blog_id, user_id):
db = mysqlModel()
res = db.updateBlogRead(blog_id=blog_id, user_id=user_id)
db.close()
return res
def getBlogList(user_id, sort_by='id'):
db = mysqlModel()
blog_list = db.getBlogs(user_id=user_id, sort_by=sort_by)
db.close()
return blog_list
def updateBlogCommentCount(manager_id, blog_id):
db = mysqlModel()
res = db.updateBlogCommentCount(manager_id=manager_id, blog_id=blog_id)
db.close()
return res
