RuyiLuo
229 lines
8.0 KiB
Python
229 lines
8.0 KiB
Python
import sys
|
|
sys.path.append("./")
|
|
import json
|
|
from flask_cors import *
|
|
from flask import Flask, jsonify, request
|
|
import snowflake.client
|
|
from dao.mysql_server import MysqlServer
|
|
from dao.entity.register_user import RegisterUser
|
|
from dao.entity.logitem import LogItem
|
|
from dao.entity.user_likes import UserLikes
|
|
from dao.entity.user_collections import UserCollections
|
|
from controller.user_action_controller import UserAction
|
|
from controller.log_controller import LogController
|
|
from recprocess.online import OnlineServer as RecsysServer
|
|
|
|
app = Flask(__name__)
|
|
|
|
# 允许跨域访问
|
|
CORS(app, supports_credentials=True)
|
|
|
|
# 定义推荐服务的实例, 是一个单例类
|
|
recsys_server = RecsysServer()
|
|
|
|
@app.route('/recsys/register', methods=["POST"])
|
|
def register():
|
|
"""用户注册
|
|
"""
|
|
request_str = request.get_data()
|
|
request_dict = json.loads(request_str)
|
|
# print(request_dict)
|
|
|
|
user = RegisterUser()
|
|
user.username = request_dict["username"]
|
|
user.passwd = request_dict["passwd"]
|
|
|
|
# 查询当前用户名是否已经被用过了
|
|
result = UserAction().user_is_exist(user, "register")
|
|
|
|
if result != 0:
|
|
return jsonify({"code": 500, "mgs": "this username is exists"})
|
|
|
|
user.userid = snowflake.client.get_guid() # 雪花算法
|
|
|
|
user.age = request_dict["age"]
|
|
user.gender = request_dict["gender"]
|
|
user.city = request_dict["city"]
|
|
|
|
# 添加注册用户
|
|
save_res = UserAction().save_user(user)
|
|
if not save_res:
|
|
return jsonify({"code": 500, "mgs": "register fail."})
|
|
|
|
return jsonify({"code": 200, "msg": "register success."})
|
|
|
|
|
|
@app.route('/recsys/login', methods=["POST"])
|
|
def login():
|
|
"""用户登录
|
|
"""
|
|
request_str = request.get_data()
|
|
request_dict = json.loads(request_str)
|
|
|
|
user = RegisterUser()
|
|
user.username = request_dict["username"]
|
|
user.passwd = request_dict["passwd"]
|
|
|
|
# 查询数据库中的用户名或者密码是否存在
|
|
try:
|
|
result = UserAction().user_is_exist(user, "login")
|
|
# print(result,"login")
|
|
if result == 1:
|
|
return jsonify({"code": 200, "msg": "login success"})
|
|
elif result == 2:
|
|
# 密码错误
|
|
return jsonify({"code": 500, "msg": "passwd is error"})
|
|
else:
|
|
return jsonify({"code": 500, "msg": "this username is not exist!"})
|
|
except Exception as e:
|
|
return jsonify({"code": 500, "mgs": "login fail."})
|
|
|
|
|
|
@app.route('/recsys/rec_list', methods=["GET"])
|
|
def rec_list():
|
|
"""推荐页
|
|
"""
|
|
user_name = request.args.get('user_id')
|
|
page_id = request.args.get('page_id')
|
|
|
|
# 查询用户的id
|
|
user_id = UserAction().get_user_id_by_name(user_name)
|
|
if not user_id:
|
|
return False
|
|
|
|
if user_id is None or page_id is None:
|
|
return jsonify({"code": 2000, "msg": "user_id or page_id is none!"})
|
|
try:
|
|
rec_news_list = recsys_server.get_rec_list(user_id, page_id)
|
|
if len(rec_news_list) == 0:
|
|
return jsonify({"code": 500, "msg": "rec_list data is empty."})
|
|
return jsonify({"code": 200, "msg": "request rec_list success.", "data": rec_news_list, "user_id": user_id})
|
|
except Exception as e:
|
|
print(str(e))
|
|
return jsonify({"code": 500, "msg": "redis fail."})
|
|
|
|
|
|
@app.route('/recsys/hot_list', methods=["GET"])
|
|
def hot_list():
|
|
"""热门页面
|
|
"""
|
|
if request.method == "GET":
|
|
user_name = request.args.get('user_id')
|
|
page_id = request.args.get('page_id')
|
|
|
|
if user_name is None or page_id is None:
|
|
return jsonify({"code": 2000, "msg": "user_name or page_id is none!"})
|
|
|
|
# 查询用户的id
|
|
user_id = UserAction().get_user_id_by_name(user_name)
|
|
if not user_id:
|
|
return False
|
|
|
|
try:
|
|
# 这里需要改成get_hot_list, 当前get_hot_list方法还没有实现
|
|
rec_news_list = recsys_server.get_hot_list(user_id)
|
|
# 下面这个接口是用来前端测试的
|
|
# rec_news_list = recsys_server.get_rec_list(user_id, page_id)
|
|
|
|
if len(rec_news_list) == 0:
|
|
return jsonify({"code": 200, "msg": "request redis data fail."})
|
|
# rec_news_list = recsys_server.get_hot_list(user_id, page_id)
|
|
return jsonify({"code": 200, "msg": "request hot_list success.", "data": rec_news_list, "user_id": user_id})
|
|
except Exception as e:
|
|
print(str(e))
|
|
return jsonify({"code": 2000, "msg": "request hot_list fail."})
|
|
|
|
|
|
@app.route('/recsys/news_detail', methods=["GET"])
|
|
def news_detail():
|
|
"""一篇文章的详细信息
|
|
"""
|
|
user_name = request.args.get('user_name')
|
|
news_id = request.args.get('news_id')
|
|
|
|
user_id = UserAction().get_user_id_by_name(user_name)
|
|
|
|
# if news_id is None or user_id is None:
|
|
if news_id is None or user_name is None:
|
|
return jsonify({"code": 2000, "msg": "news_id is none or user_name is none!"})
|
|
try:
|
|
news_detail = recsys_server.get_news_detail(news_id)
|
|
|
|
# recsys_server.save_user_consume(user_id,news_id) # 记录用户消费的
|
|
|
|
if UserAction().get_likes_counts_by_user(user_id,news_id) > 0:
|
|
news_detail["likes"] = True
|
|
else:
|
|
news_detail["likes"] = False
|
|
|
|
if UserAction().get_coll_counts_by_user(user_id,news_id) > 0:
|
|
news_detail["collections"] = True
|
|
else:
|
|
news_detail["collections"] = False
|
|
# print("test",news_detail)
|
|
return jsonify({"code": 0, "msg": "request news_detail success.", "data": news_detail})
|
|
except Exception as e:
|
|
print(str(e))
|
|
return jsonify({"code": 2000, "msg": "error"})
|
|
|
|
|
|
@app.route('/recsys/action', methods=["POST"])
|
|
def actions():
|
|
"""用户的行为：阅读，点赞，收藏
|
|
"""
|
|
request_str = request.get_data()
|
|
request_dict = json.loads(request_str)
|
|
|
|
username = request_dict.get('user_name')
|
|
newsid = request_dict.get('news_id')
|
|
actiontype = request_dict.get("action_type")
|
|
actiontime = request_dict.get("action_time")
|
|
|
|
userid = UserAction().get_user_id_by_name(username) # 获取用户 id
|
|
if not userid:
|
|
return jsonify({"code": 2000, "msg": "user not register"})
|
|
|
|
#TODO 先判断当前的action_type是否是取消的意思，如果是的话，需要将数据库中对应的操作删掉
|
|
action_type_list = actiontype.split(":")
|
|
# print(actiontype)
|
|
if len(action_type_list) == 2:
|
|
_action_type = action_type_list[0]
|
|
if action_type_list[1] == "false": # 如果数据库中这个参数为false的话
|
|
# 删除数据
|
|
if _action_type=="likes":
|
|
UserAction().del_likes_by_user(userid,newsid) # 删除用户喜欢记录
|
|
elif _action_type=="collections":
|
|
UserAction().del_coll_by_user(userid,newsid) # 删除用户收藏记录
|
|
else:
|
|
if _action_type=="likes":
|
|
userlikes = UserLikes()
|
|
userlikes.new(userid,username,newsid)
|
|
UserAction().save_one_action(userlikes) # 记录用户喜欢记录
|
|
elif _action_type=="collections":
|
|
usercollections = UserCollections()
|
|
usercollections.new(userid,username,newsid)
|
|
UserAction().save_one_action(usercollections) # 记录用户收藏记录
|
|
|
|
try:
|
|
# 落日志
|
|
logitem = LogItem()
|
|
logitem.new(userid,newsid,action_type_list[0])
|
|
LogController().save_one_log(logitem)
|
|
|
|
# 更新redis中的展示数据 新闻侧
|
|
# if action_type_list[0] in ["read","likes","collections"]:
|
|
recsys_server.update_news_dynamic_info(news_id=newsid,action_type=action_type_list)
|
|
return jsonify({"code": 200, "msg": "action success"})
|
|
|
|
except Exception as e:
|
|
print(str(e))
|
|
return jsonify({"code": 2000, "msg": "action error"})
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# 允许服务器被公开访问
|
|
app.run(debug=True, host='0.0.0.0', port=3000, threaded=True)
|
|
# 只能被自己的机子访问
|
|
# app.run(debug=True, host='127.0.0.1', port=10086, threaded=True)
|
|
|