def _buildMovieObj(self, movieInfo): movie = Movie(movieInfo["name"]) movie.movieYear = movieInfo["year"] movie.pictureLink = movieInfo["picture"] movie.magnetLink = movieInfo["magnet"] movie.quality = movieInfo["quality"] movie.seeders = movieInfo["seeders"] movie.imdb = movieInfo["imdb"] movie.ranking = movieInfo["rating"] movie.summary = movieInfo["summary"] return movie
def get_movie(root, path): def get_movie_files(movie_path): return [movie_file for movie_file in os.listdir(movie_path) if os.path.isfile(os.path.join(movie_path, movie_file)) and (os.path.splitext(movie_file)[1] in utils.get_movie_extensions())] if not os.path.exists(os.path.join(root, path)): raise OSError('Path do not exist') movie = Movie() movie.path = path rx = regex.get_movie(movie.path) movie.search_year = rx['year'] movie.imdb = rx['imdbID'] movie.search_title = rx['title'] movie.search_alternative_title = utils.replace(rx['title']) movie.path = os.path.join(root, movie.path) files = get_movie_files(movie.path) if len(files) < 1: return movie # todo: this *should* not longer be needed, if 'new.mkv' in files and len(files) > 1: os.remove(os.path.join(movie.path, 'new.mkv')) files = get_movie_files(movie.path) # todo: implement multi cd support if len(files) == 1: movie.files = [files[0]] if len(files) > 1: # get only files that are tagged as CD for _file in [f for f in files if regex.get_cd(f) == '']: files.remove(_file) movie.files = files return movie