コード例 #1
0
ファイル: app_.py プロジェクト: LouisCauchy/new_api_versions
def recommendations():
    data = request.args
    try:
        access_key = str(dict(data)['access_key'][0])
    except:
        return accessKeyRequired()

    agent = Agent.query.filter_by(access_key=access_key).first()
    if agent is None: return accessKeyRequired(not_in_db=True)
    try:
        agent_user_id = str(data['user_id'])
    except:
        return badRequest()

    user = User.query.filter_by(agent_user_id=agent_user_id,
                                agent_id=agent.id).first()
    if user is None: return badRequest(user=True)

    user_movies = get_user_movies(user.id)
    if not user_movies:
        return 'User has no movies'  #, Response(403) jsonify([])

    if agent.agent_name == 'vkino':
        user_tmdb_movies = []
        for movie in user_movies:
            tmdbid = Vkino.query.filter_by(vkino_id=movie).first().tmdb_fk_id
            if tmdbid is None:
                continue
            else:
                user_tmdb_movies.append(tmdbid)
    if not user_tmdb_movies:
        return jsonify(recommendations=[])

    min_num_of_recs = 3

    rs = Recommender(user_tmdb_movies, agent.agent_name + '_movies',
                     agent.agent_name + '_id', user.id)
    rs.tmdb_input_info()
    rs.form_characteristics()
    result = rs.get_recommendations()

    if not result:
        return jsonify(recommendations=result)

    agent_premiere_idx = [movie['pk_id'] for movie in result[:min_num_of_recs]]
    if type(agent_premiere_idx) is list:
        r = Recommendation(user.id, agent.id, datetime.now())
        db.session.add(r)
        db.session.commit()
        for movie in agent_premiere_idx:
            mr = MovieRecommendation(r.id, movie, datetime.now())
            db.session.add(mr)
            db.session.commit()

    return jsonify(recommendations=agent_premiere_idx, recommendation_id=r.id)
コード例 #2
0
def recommendations():
    print('2')
    # print(request.args)
    data = request.args
    try:
        access_key = str(dict(data)['access_key'][0])
    except:
        return accessKeyRequired()

    agent = Agent.query.filter_by(access_key=access_key).first()
    if agent is None: return accessKeyRequired(not_in_db=True)
    try:
        agent_user_id = str(data['user_id'])
    except:
        return badRequest()

    user = User.query.filter_by(agent_user_id=agent_user_id,
                                agent_id=agent.id).first()
    if user is None: return badRequest(user=True)

    user_movies = get_user_movies(user.id)
    logger.info(user_movies)
    print(user_movies)
    if not user_movies:
        return 'User has no movies'  #, Response(403) jsonify([])

    if not Recommendation.query.filter_by(user_id=user.id).all():
        print('No recommendations before')
    else:
        print('Some movies were recommender before')
        prev_rec_dates = []
        for rec in Recommendation.query.filter_by(user_id=user.id).all():
            prev_rec_dates.append(
                (datetime.now() - rec.created_at).total_seconds() / 3600)
        last_rec_date = min(prev_rec_dates)
        print(last_rec_date)
        if last_rec_date <= 24:
            print('Latest recommendation was today. Return empty list')
            return jsonify(recommendations=[])

    if agent.agent_name == 'vkino':
        user_tmdb_movies = []
        for movie in user_movies:
            tmdbid = Vkino.query.filter_by(vkino_id=movie).first().tmdb_fk_id
            if tmdbid is None:
                continue
            else:
                user_tmdb_movies.append(tmdbid)
    if not user_tmdb_movies:
        return jsonify(recommendations=[])
    print(user_tmdb_movies)
    min_num_of_recs = 3

    rs = Recommender(user_tmdb_movies, agent.agent_name + '_movies',
                     agent.agent_name + '_id', user.id)
    rs.tmdb_input_info()
    rs.form_characteristics()
    result = rs.get_recommendations()
    #print(result[0])
    if not result:
        return jsonify(recommendations=result)

    agent_premiere_idx = [movie['pk_id'] for movie in result[:min_num_of_recs]]
    print(agent_premiere_idx)
    if type(agent_premiere_idx) is list:
        r = Recommendation(user.id, agent.id, datetime.now())
        db.session.add(r)
        db.session.commit()
        for movie in agent_premiere_idx:
            mr = MovieRecommendation(r.id, movie, datetime.now())
            db.session.add(mr)
            db.session.commit()

    return jsonify(recommendations=agent_premiere_idx, recommendation_id=r.id)