From 92e0d13913cca747b6620888d429c770c3a4b3b8 Mon Sep 17 00:00:00 2001 From: vitiko98 Date: Fri, 25 Jun 2021 23:22:39 -0400 Subject: [PATCH] Fix Subsynchro provider (Fix #1450) --- libs/subliminal_patch/providers/subsynchro.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/libs/subliminal_patch/providers/subsynchro.py b/libs/subliminal_patch/providers/subsynchro.py index d3feab733..e05e7c4e7 100644 --- a/libs/subliminal_patch/providers/subsynchro.py +++ b/libs/subliminal_patch/providers/subsynchro.py @@ -3,7 +3,7 @@ import io import logging import os -from zipfile import ZipFile +from zipfile import ZipFile, is_zipfile from requests import Session from guessit import guessit @@ -144,15 +144,10 @@ class SubsynchroProvider(Provider): ) response.raise_for_status() - if subtitle.file_type.endswith(".zip"): + stream = io.BytesIO(response.content) + if is_zipfile(stream): logger.debug("Zip file found") - subtitle_ = self.get_file(ZipFile(io.BytesIO(response.content))) - - elif subtitle.file_type.endswith(".srt"): - logger.debug("Srt file found") - subtitle_ = response.content - + subtitle_ = self.get_file(ZipFile(stream)) + subtitle.content = fix_line_ending(subtitle_) else: - raise APIThrottled(f"Unknown file type: {subtitle.file_type}") - - subtitle.content = fix_line_ending(subtitle_) + raise APIThrottled(f"Unknown file type: {subtitle.download_url}")