diff --git a/bazarr/get_episodes.py b/bazarr/get_episodes.py index 0e3f6f15f..4f80a2953 100644 --- a/bazarr/get_episodes.py +++ b/bazarr/get_episodes.py @@ -84,20 +84,10 @@ def sync_episodes(): resolution = str(episode['episodeFile']['quality']['quality']['resolution']) + 'p' videoCodec = episode['episodeFile']['mediaInfo']['videoCodec'] - if videoCodec.startswith('x264'): videoCodec = 'h264' - elif videoCodec.startswith('XviD'): videoCodec = 'XviD' - elif videoCodec.startswith('DivX'): videoCodec = 'DivX' - elif videoCodec.startswith('MPEG-1 Video'): videoCodec = 'Mpeg' - elif videoCodec.startswith('MPEG-2 Video'): videoCodec = 'Mpeg2' - elif videoCodec.startswith('MPEG-4 Video'): videoCodec = 'Mpeg4' - elif videoCodec.endswith('VP6'): videoCodec = 'VP6' - elif videoCodec.endswith('VP7'): videoCodec = 'VP7' - elif videoCodec.endswith('VP8'): videoCodec = 'VP8' - elif videoCodec.endswith('VP9'): videoCodec = 'VP9' + videoCodec = SonarrFormatVideoCodec(videoCodec) audioCodec = episode['episodeFile']['mediaInfo']['audioCodec'] - if audioCodec == 'AC-3': audioCodec = 'AC3' - elif audioCodec == 'MPEG Audio': audioCodec = 'MP3' + audioCodec = SonarrFormatAudioCodec(audioCodec) # Add episodes in sonarr to current episode list current_episodes_sonarr.append(episode['id']) @@ -152,3 +142,28 @@ def sync_episodes(): logging.debug('BAZARR All missing subtitles updated in database.') q4ws.append('Episodes sync from Sonarr ended.') + + +def SonarrFormatAudioCodec(audioCodec): + if audioCodec == 'AC-3': return 'AC3' + if audioCodec == 'E-AC-3': return 'EAC3' + if audioCodec == 'MPEG Audio': return 'MP3' + + return audioCodec + + +def SonarrFormatVideoCodec(videoCodec): + if videoCodec == 'x264' or videoCodec == 'AVC': return 'h264' + if videoCodec == 'x265' or videoCodec == 'HEVC': return 'h265' + if videoCodec.startswith('XviD'): return 'XviD' + if videoCodec.startswith('DivX'): return 'DivX' + if videoCodec == 'MPEG-1 Video': return 'Mpeg' + if videoCodec == 'MPEG-2 Video': return 'Mpeg2' + if videoCodec == 'MPEG-4 Video': return 'Mpeg4' + if videoCodec == 'VC-1': return 'VC1' + if videoCodec.endswith('VP6'): return 'VP6' + if videoCodec.endswith('VP7'): return 'VP7' + if videoCodec.endswith('VP8'): return 'VP8' + if videoCodec.endswith('VP9'): return 'VP9' + + return videoCodec