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)
