def add_new_link(data): col = Connection.db()["links"] return col.insert({ 'movieId': data['movieId'], 'tmdbId': data['tmdbId'] })
def get_all_movies_genres(): col_movies = Connection.db()["movies"] base_movies = {} for data in col_movies.find(): base_movies[data['movieId']] = (data['genres'], data['year']) return base_movies
def get_movie_api(movie_id): # COLEÇÃO DE FILMES col_movies = Connection.db()["movies"] data = col_movies.find_one({"movieId": movie_id}) return { "movieId": data['movieId'], "title": data['title'], "genres": data['genres'], "year": data['year'], }
def get_movies(with_context): # COLEÇÃO DE AVALIAÇÕES col_ratings = Connection.db()["ratings"] # INICIA A BASE DE AVALIACOES base_ratings = {} for data in col_ratings.find(): base_ratings.setdefault(data['userId'], {}) # APLICAÇÃO DO CONTEXTO -> FILTRA AS AVALIACOES PELO CONTEXTO(SE É FIM DE SEMANA OU NÃO) if with_context and not Util.is_context_dayweek( datetime.fromtimestamp(float( data['timestamp'])).weekday()): continue base_ratings[data['userId']][data['movieId']] = float( data['rating']) return base_ratings
def get_all_movies_per_genre(genres): # COLEÇÃO DE FILMES col_movies = Connection.db()["movies"] return col_movies.find({"genres": {"$" "in": genres}})
def get_rating(movie_id): # COLEÇÃO DE FILMES col_movies = Connection.db()["ratings"] return col_movies.find_one({"movieId": movie_id})
def get_all_movies(): col_movies = Connection.db()["movies"] return col_movies.find()
def get_movie_link(movie_id): col_links = Connection.db()["links"] return col_links.find_one({'movieId': movie_id})
def get_movies_links(): col_links = Connection.db()["links"] return col_links.find()
def get_movie_genre(genre): col_movies = Connection.db()["movies"] return col_movies.find_one({"genres": genre})
def get_categories(movieId): col_tabs = Connection.db()["movies"] return col_tabs.find_one({"movieId": movieId})["genres"]
def get_tags_movie(movieId): col_tabs = Connection.db()["tags"] return col_tabs.find({"movieId": movieId})
def get_tmdb_exist(tmdbId): col = Connection.db()["links"] return col.find_one({"tmdbId": tmdbId})
def get_tmdb_id(movieId): col = Connection.db()["links"] return col.find_one({"movieId": movieId})
def get_user_ratings(user_id): # COLEÇÃO DE AVALIAÇÕES DO USUARIO col_movies = Connection.db()["ratings"] return col_movies.find({"userId": user_id})
def add(data): col_users = Connection.db()["users"] return col_users.insert(data)