def save_rec(path, db):
"""
将推荐的结果存放到redis中
存储的数据结构{"hot": [(*,*),(*,*)]}
:param path: 推荐结果存放的路径
:param db: 推荐结果存放的db
:return: null
"""
rec_dict = np.load(path, allow_pickle=True).item()
r = config.conn()
for key, value in rec_dict.items():
r.zadd(key, dict(value))
r.close()
def save_metadata(path, db):
"""
将壁纸的信息存储到redis中
{"id": (wp_id, wp_type, wp_url)}
:param path: 壁纸data存放的路径
:param db: 存的的db
:return: null
"""
rec = np.load(path, allow_pickle=True).item()
r = config.conn()
metadata_dict = np.load(path, allow_pickle=True).item()
for key, value in metadata_dict.items():
(wp_id, wp_url) = value[0], value[2]
j = json.dumps({"wp_id": wp_id, "wp_url": wp_url})
r.hset("metadata", wp_id, j)
r.close()
import json
from app import app, db
from flask import request, jsonify, Response
from app.dbUtils import config
from app.models import WellPaper, UserPaper, UserLoad, User
# redis
redis = config.conn()
@app.route('/getPaper', methods=['POST'])
def getPaper():
data = json.loads(request.data)
try:
mid = db.session.query(WellPaper).filter(
WellPaper.wp_type == data['type']).all()
favorate = db.session.query(UserPaper).filter(
UserPaper.u_id == data['u_id']).all()
except:
db.session.rollback()
mid = db.session.query(WellPaper).filter(
WellPaper.wp_type == data['type']).all()
favorate = db.session.query(UserPaper).filter(
UserPaper.u_id == data['u_id']).all()
db.session.close()
result = dict()
fav = list()
for i, element in enumerate(favorate):
fav.append(element.to_dict().get('wp_id'))