This commit is contained in:
Louis Vézina 2019-09-12 06:34:01 -04:00
parent 90609b6433
commit 4e7e3a39d2
3 changed files with 66 additions and 54 deletions

View file

@ -56,12 +56,16 @@ def sync_episodes():
r.raise_for_status() r.raise_for_status()
except requests.exceptions.HTTPError as errh: except requests.exceptions.HTTPError as errh:
logging.exception("BAZARR Error trying to get episodes from Sonarr. Http error.") logging.exception("BAZARR Error trying to get episodes from Sonarr. Http error.")
return
except requests.exceptions.ConnectionError as errc: except requests.exceptions.ConnectionError as errc:
logging.exception("BAZARR Error trying to get episodes from Sonarr. Connection Error.") logging.exception("BAZARR Error trying to get episodes from Sonarr. Connection Error.")
return
except requests.exceptions.Timeout as errt: except requests.exceptions.Timeout as errt:
logging.exception("BAZARR Error trying to get episodes from Sonarr. Timeout Error.") logging.exception("BAZARR Error trying to get episodes from Sonarr. Timeout Error.")
return
except requests.exceptions.RequestException as err: except requests.exceptions.RequestException as err:
logging.exception("BAZARR Error trying to get episodes from Sonarr.") logging.exception("BAZARR Error trying to get episodes from Sonarr.")
return
else: else:
for episode in r.json(): for episode in r.json():
if 'hasFile' in episode: if 'hasFile' in episode:
@ -127,69 +131,69 @@ def sync_episodes():
'video_codec': videoCodec, 'video_codec': videoCodec,
'audio_codec': audioCodec}) 'audio_codec': audioCodec})
# Update existing episodes in DB # Update existing episodes in DB
episode_in_db_list = [] episode_in_db_list = []
episodes_in_db = TableEpisodes.select( episodes_in_db = TableEpisodes.select(
TableEpisodes.sonarr_series_id, TableEpisodes.sonarr_series_id,
TableEpisodes.sonarr_episode_id, TableEpisodes.sonarr_episode_id,
TableEpisodes.title, TableEpisodes.title,
TableEpisodes.path, TableEpisodes.path,
TableEpisodes.season, TableEpisodes.season,
TableEpisodes.episode, TableEpisodes.episode,
TableEpisodes.scene_name, TableEpisodes.scene_name,
TableEpisodes.monitored, TableEpisodes.monitored,
TableEpisodes.format, TableEpisodes.format,
TableEpisodes.resolution, TableEpisodes.resolution,
TableEpisodes.video_codec, TableEpisodes.video_codec,
TableEpisodes.audio_codec TableEpisodes.audio_codec
).dicts() ).dicts()
for item in episodes_in_db: for item in episodes_in_db:
episode_in_db_list.append(item) episode_in_db_list.append(item)
episodes_to_update_list = [i for i in episodes_to_update if i not in episode_in_db_list] episodes_to_update_list = [i for i in episodes_to_update if i not in episode_in_db_list]
for updated_episode in episodes_to_update_list: for updated_episode in episodes_to_update_list:
TableEpisodes.update( TableEpisodes.update(
updated_episode updated_episode
).where( ).where(
TableEpisodes.sonarr_episode_id == updated_episode['sonarr_episode_id'] TableEpisodes.sonarr_episode_id == updated_episode['sonarr_episode_id']
).execute() ).execute()
altered_episodes.append([updated_episode['sonarr_episode_id'], altered_episodes.append([updated_episode['sonarr_episode_id'],
updated_episode['path'], updated_episode['path'],
updated_episode['sonarr_series_id']]) updated_episode['sonarr_series_id']])
# Insert new episodes in DB # Insert new episodes in DB
for added_episode in episodes_to_add: for added_episode in episodes_to_add:
TableEpisodes.insert( TableEpisodes.insert(
added_episode added_episode
).on_conflict_ignore().execute() ).on_conflict_ignore().execute()
altered_episodes.append([added_episode['sonarr_episode_id'], altered_episodes.append([added_episode['sonarr_episode_id'],
added_episode['path'], added_episode['path'],
added_episode['sonarr_series_id']]) added_episode['sonarr_series_id']])
# Remove old episodes from DB # Remove old episodes from DB
removed_episodes = list(set(current_episodes_db_list) - set(current_episodes_sonarr)) removed_episodes = list(set(current_episodes_db_list) - set(current_episodes_sonarr))
for removed_episode in removed_episodes: for removed_episode in removed_episodes:
TableEpisodes.delete().where( TableEpisodes.delete().where(
TableEpisodes.sonarr_episode_id == removed_episode TableEpisodes.sonarr_episode_id == removed_episode
).execute() ).execute()
# Store subtitles for added or modified episodes # Store subtitles for added or modified episodes
for altered_episode in altered_episodes: for altered_episode in altered_episodes:
store_subtitles(path_replace(altered_episode[1])) store_subtitles(path_replace(altered_episode[1]))
list_missing_subtitles(altered_episode[2]) list_missing_subtitles(altered_episode[2])
logging.debug('BAZARR All episodes synced from Sonarr into database.') logging.debug('BAZARR All episodes synced from Sonarr into database.')
# Search for desired subtitles if no more than 5 episodes have been added. # Search for desired subtitles if no more than 5 episodes have been added.
if len(altered_episodes) <= 5: if len(altered_episodes) <= 5:
logging.debug("BAZARR No more than 5 episodes were added during this sync then we'll search for subtitles.") logging.debug("BAZARR No more than 5 episodes were added during this sync then we'll search for subtitles.")
for altered_episode in altered_episodes: for altered_episode in altered_episodes:
episode_download_subtitles(altered_episode[0]) episode_download_subtitles(altered_episode[0])
else: else:
logging.debug("BAZARR More than 5 episodes were added during this sync then we wont search for subtitles right now.") logging.debug("BAZARR More than 5 episodes were added during this sync then we wont search for subtitles right now.")
def SonarrFormatAudioCodec(audioCodec): def SonarrFormatAudioCodec(audioCodec):

View file

@ -43,12 +43,16 @@ def update_movies():
r.raise_for_status() r.raise_for_status()
except requests.exceptions.HTTPError as errh: except requests.exceptions.HTTPError as errh:
logging.exception("BAZARR Error trying to get movies from Radarr. Http error.") logging.exception("BAZARR Error trying to get movies from Radarr. Http error.")
return
except requests.exceptions.ConnectionError as errc: except requests.exceptions.ConnectionError as errc:
logging.exception("BAZARR Error trying to get movies from Radarr. Connection Error.") logging.exception("BAZARR Error trying to get movies from Radarr. Connection Error.")
return
except requests.exceptions.Timeout as errt: except requests.exceptions.Timeout as errt:
logging.exception("BAZARR Error trying to get movies from Radarr. Timeout Error.") logging.exception("BAZARR Error trying to get movies from Radarr. Timeout Error.")
return
except requests.exceptions.RequestException as err: except requests.exceptions.RequestException as err:
logging.exception("BAZARR Error trying to get movies from Radarr.") logging.exception("BAZARR Error trying to get movies from Radarr.")
return
else: else:
# Get current movies in DB # Get current movies in DB
current_movies_db = TableMovies.select( current_movies_db = TableMovies.select(

View file

@ -34,12 +34,16 @@ def update_series():
r.raise_for_status() r.raise_for_status()
except requests.exceptions.HTTPError as errh: except requests.exceptions.HTTPError as errh:
logging.exception("BAZARR Error trying to get series from Sonarr. Http error.") logging.exception("BAZARR Error trying to get series from Sonarr. Http error.")
return
except requests.exceptions.ConnectionError as errc: except requests.exceptions.ConnectionError as errc:
logging.exception("BAZARR Error trying to get series from Sonarr. Connection Error.") logging.exception("BAZARR Error trying to get series from Sonarr. Connection Error.")
return
except requests.exceptions.Timeout as errt: except requests.exceptions.Timeout as errt:
logging.exception("BAZARR Error trying to get series from Sonarr. Timeout Error.") logging.exception("BAZARR Error trying to get series from Sonarr. Timeout Error.")
return
except requests.exceptions.RequestException as err: except requests.exceptions.RequestException as err:
logging.exception("BAZARR Error trying to get series from Sonarr.") logging.exception("BAZARR Error trying to get series from Sonarr.")
return
else: else:
# Get current shows in DB # Get current shows in DB
current_shows_db = TableShows.select( current_shows_db = TableShows.select(