Exemple #1
0
    def get_qprecs(self, algo, movies, nr_recs, items, user_id):
        dbManager = DbManager()
        #user_id = dbManager.insert_and_get_min_user_id()
        rating_value = 5
        # remove all ratings from this user
        dbManager.remove_ratings(user_id)

        # insert ratings (value: 10) for user 0 for each movie
        for m in movies:
            dbManager.insert_rating(user_id, m, rating_value)

        # get recs
        all_recs = self.get_recs_from_recserver([user_id], nr_recs, algo,
                                                items)
        print(all_recs)
        for user_recs in all_recs:
            if user_recs['recs'] != None:
                for rec in user_recs['recs']:
                    rec['title'] = dbManager.get_movie(
                        rec['item']).loc[0, 'title']
                    rec['image_url'] = self.get_image_url(rec['item'])

        return all_recs