def test_save_empty_movies_to_json(self): movies = [] filename = os.path.join(os.path.join(TESTDATA_PATH, 'exports'), 'TEST_empty_movies.json') file_impex.save_movies_to_json(movies, os.path.join(TESTDATA_PATH, 'exports'), 'TEST_empty_movies.json') with open(filename) as file: self.assertEqual(movies, json.load(file)) os.remove(filename)
def test_save_single_movie_to_json(self): movies = [self.movie] movies_json = [m for m in movies] filename = os.path.join(os.path.join(TESTDATA_PATH, 'exports'), 'TEST_single_movie.json') file_impex.save_movies_to_json(movies, os.path.join(TESTDATA_PATH, 'exports'), 'TEST_single_movie.json') with open(filename) as file: self.assertEqual(movies_json, json.load(file)) os.remove(filename)
def test_save_empty_movies_to_json_in_new_folder(self): movies = [] filename = os.path.join(TESTDATA_NEW_PATH, 'TEST_empty_movies.json') file_impex.save_movies_to_json(movies, TESTDATA_NEW_PATH, 'TEST_empty_movies.json') with open(filename) as file: self.assertEqual(movies, json.load(file)) os.remove(filename) os.removedirs(TESTDATA_NEW_PATH)
def parse_data_from_source(parser): movies = parser.parse() json_filename = '%s_%s.json' % (TIMESTAMP, type(parser.site).__name__) file_impex.save_movies_to_json(movies, folder=EXPORTS_FOLDER, filename=json_filename) sys.stdout.write('\r\n===== %s: saved %i parsed movies to %s/%s\r\n' % (type(parser.site).__name__, len(movies), EXPORTS_FOLDER, json_filename)) sys.stdout.flush() return movies
def test_save_single_movie_to_json(self): movies = [self.movie] movies_json = movies filename = os.path.join(os.path.join(TESTDATA_PATH, "exports"), "TEST_single_movie.json") file_impex.save_movies_to_json(movies, os.path.join(TESTDATA_PATH, "exports"), "TEST_single_movie.json") with open(filename) as file: self.assertEqual(movies_json, json.load(file)) os.remove(filename)
def parse_data_from_source(parser): movies = parser.parse() json_filename = '{timestamp}_{sitename}.json'.format(timestamp=TIMESTAMP, sitename=type(parser.site).__name__) file_impex.save_movies_to_json(movies, folder=EXPORTS_FOLDER, filename=json_filename) sys.stdout.write('\r\n===== {site_displayname}: saved {parsed_movies_count} parsed movies to ' '{folder}/{filename}\r\n'.format( site_displayname=parser.site.site_displayname, parsed_movies_count=len(movies), folder=EXPORTS_FOLDER, filename=json_filename )) sys.stdout.flush() return movies
def _handle_failed_movies(self, movies): if self.args and self.args.verbose and self.args.verbose >= 1: self._print_failed_movies() if len(self.failed_movies) > 0: file_impex.save_movies_to_json( movies, folder=self.exports_folder, filename=self.failed_movies_filename) sys.stdout.write( '===== %s: export data for %i failed movies to %s/%s\r\n' % (self.site.site_displayname, len(self.failed_movies), self.exports_folder, self.failed_movies_filename)) sys.stdout.flush()
def _handle_failed_movies(self, movies): if self.args and self.args.verbose and self.args.verbose >= 1: self._print_failed_movies() if len(self.failed_movies) > 0: file_impex.save_movies_to_json(movies, folder=self.exports_folder, filename=self.failed_movies_filename) sys.stdout.write('===== {site_displayname}: export data for {failed_number} failed movies to ' '{folder}/{filename}\r\n'.format( site_displayname=self.site.site_displayname, failed_number=len(self.failed_movies), folder=self.exports_folder, filename=self.failed_movies_filename )) sys.stdout.flush()
def _handle_failed_movies(self, movies): for failed_movie in self.failed_movies: sys.stdout.write('FAILED TO FIND: %s (%i)\r\n' % (failed_movie['title'], failed_movie['year'])) if len(self.failed_movies) > 0: file_impex.save_movies_to_json( movies, folder=self.exports_folder, filename=self.failed_movies_filename) sys.stdout.write( '===== %s: export data for %i failed movies to %s/%s\r\n' % (self.site.site_name, len(self.failed_movies), self.exports_folder, self.failed_movies_filename)) sys.stdout.flush()
def parse_data_from_source(parser): try: movies = parser.parse() except RatSException as e: command_line.error(str(e)) sys.exit(1) json_filename = f"{TIMESTAMP}_{type(parser.site).__name__}.json" file_impex.save_movies_to_json(movies, folder=EXPORTS_FOLDER, filename=json_filename) sys.stdout.write( f"\r\n===== {parser.site.site_displayname}: saved {len(movies)} parsed movies to " f"{EXPORTS_FOLDER}/{json_filename}\r\n") sys.stdout.flush() return movies
def test_save_multiple_movies_to_json(self): movie2 = dict() movie2['title'] = 'The Matrix' movie2['year'] = 1999 movie2['imdb'] = dict() movie2['imdb']['id'] = 'tt0133093' movie2['imdb']['url'] = 'http://www.imdb.com/title/tt0133093' movie2['trakt'] = dict() movie2['trakt']['id'] = '481' movie2['trakt']['url'] = 'https://trakt.tv/movies/the-matrix-1999' movie2['trakt']['my_rating'] = 9 movies = [self.movie, movie2] movies_json = movies filename = os.path.join(os.path.join(TESTDATA_PATH, 'exports'), 'TEST_multiple_movies.json') file_impex.save_movies_to_json(movies, os.path.join(TESTDATA_PATH, 'exports'), 'TEST_multiple_movies.json') with open(filename) as file: self.assertEqual(movies_json, json.load(file)) os.remove(filename)
def test_save_multiple_movies_to_json(self): movie2 = dict() movie2["title"] = "The Matrix" movie2["year"] = 1999 movie2["imdb"] = dict() movie2["imdb"]["id"] = "tt0133093" movie2["imdb"]["url"] = "https://www.imdb.com/title/tt0133093" movie2["trakt"] = dict() movie2["trakt"]["id"] = "481" movie2["trakt"]["url"] = "https://trakt.tv/movies/the-matrix-1999" movie2["trakt"]["my_rating"] = 9 movies = [self.movie, movie2] movies_json = movies filename = os.path.join(os.path.join(TESTDATA_PATH, "exports"), "TEST_multiple_movies.json") file_impex.save_movies_to_json(movies, os.path.join(TESTDATA_PATH, "exports"), "TEST_multiple_movies.json") with open(filename) as file: self.assertEqual(movies_json, json.load(file)) os.remove(filename)