def rate_movie():
if current_user.is_authenticated is False:
return api_error('user not login!')
movie_id = request.values.get('movie_id')
score = request.values.get('score')
if movie_id is None or score is None:
return api_error('missing args: movie_id or score')
movie_id = int(movie_id)
score = float(score)
rated = UserRatedMovie.query.filter_by(movie_id=movie_id, user_id=current_user.id).first()
mov = Movie.query.filter_by(id=movie_id).first()
if mov is None:
return api_error("movie_id error, no such movie")
if score > 10 or score < 0:
return api_error("score out of range")
if rated is None:
# 如果评分不存在
rated = UserRatedMovie(movie_id=movie_id, user_id=current_user.id, score=score)
mov.vote_average = (mov.vote_average * mov.vote_count + score) / (mov.vote_count + 1)
mov.vote_count += 1
else:
# 如果评分存在
mov.vote_average = (mov.vote_average * mov.vote_count - rated.score + score) / mov.vote_count
rated.score = score
try:
db.session.add(mov)
db.session.add(rated)
db.session.commit()
except:
db.session.rollback()
return api_error("database reported a error, fail to commit")
return api_success(None)
def general_recommend():
rank = get_rank()[:20]
q = list()
for m_id in rank:
q.append(Movie.query.filter_by(id=m_id).first())
movie_items = [{'movie_id': i.id, 'title': i.title,
'tagline': i.tagline, 'poster_link': i.poster_link}
for i in q]
return api_success({'movieItems': movie_items})
def related_recommend():
movie_id = request.values.get("movie_id")
if movie_id is None:
return api_error('missing args: movie_id')
recomm = get_recomm_by_movie_id(movie_id)[:12]
q = Movie.query.filter(Movie.id.in_(recomm))
movie_items = [{'movie_id': i.id, 'title': i.title,
'tagline': i.tagline, 'poster_link': i.poster_link}
for i in q]
return api_success({'movieItems': movie_items})
def get_user_rate():
if current_user.is_authenticated is False:
return api_error('user not login!')
movie_id = request.args.get('movie_id')
if movie_id is None:
return api_error('missing args: movie_id')
movie_id = int(movie_id)
rated = UserRatedMovie.query.filter_by(movie_id=movie_id, user_id=current_user.id).first()
if rated is None:
return api_error("user did not rate this movie")
else:
result = {'score': round(rated.score, 1), 'movie_id': rated.movie_id}
return api_success(result)
def movie_list():
page_num = request.args.get('page_num')
genre_id = request.args.get('genre_id')
# if page_num is None:
# return api_error('missing args: page_num')
if genre_id is None or genre_id is '':
q_movies = Movie.query.order_by(Movie.release_date.desc())[:30]
else:
q = Genre.query.filter_by(id=genre_id)
if q.count() == 0:
return api_error('genre_id error')
q = q.first()
q_movies = q.movies.order_by(Movie.release_date.desc())[:30]
movie_items = [{'movie_id': i.id, 'title': i.title,
'tagline': i.tagline, 'poster_link': i.poster_link}
for i in q_movies]
return api_success({'movieItems': movie_items})
def user_recommend():
if current_user.is_authenticated is False:
return api_error('user not login!')
# 根据user的rated movies 选择相关推荐，根据cf推荐，根据user关注的用户的高rating推荐。
random.seed(datetime.now().timestamp())
movie_id_list = get_recomm_by_user(current_user.id, threshold=5.0)[:5]
temp_list = random.sample(current_user.rated_movies, 3)
for item in temp_list:
movie_id_list.extend(random.sample(get_recomm_by_movie_id(item.id), 2))
# current_user.followed
print(movie_id_list)
movie_id_list = list(set(movie_id_list))
movie_id_list = random.sample(movie_id_list, 8)
q = Movie.query.filter(Movie.id.in_(movie_id_list))
movie_items = [{'movie_id': i.id, 'title': i.title,
'tagline': i.tagline, 'poster_link': i.poster_link}
for i in q]
return api_success({'movieItems': movie_items})
def genres():
q = Genre.query.all()
genres_list = [{'id': i.id, 'name': i.name} for i in q]
return api_success({'genres': genres_list})
def refresh_recomm():
if not current_user.can(Permission.ADMIN):
return api_error('Permission denial')
with MessageQueue() as mq:
mq.send_refresh_recomm_signal()
return api_success('Refreshing in seconds')