mirror of
https://github.com/morpheus65535/bazarr.git
synced 2025-02-23 14:26:02 +08:00
Fix for 90609b6433
This commit is contained in:
parent
90609b6433
commit
4e7e3a39d2
3 changed files with 66 additions and 54 deletions
|
@ -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):
|
||||||
|
|
|
@ -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(
|
||||||
|
|
|
@ -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(
|
||||||
|
|
Loading…
Reference in a new issue