[yandexmusic] Improve error handling

pull/2/head
Sergey M․ 9 years ago
parent 0cbcbdd89d
commit eebe6b382e
No known key found for this signature in database
GPG Key ID: 2C393E0F18A9236D

@ -18,9 +18,10 @@ from ..utils import (
class YandexMusicBaseIE(InfoExtractor): class YandexMusicBaseIE(InfoExtractor):
@staticmethod @staticmethod
def _handle_error(response): def _handle_error(response):
error = response.get('error') if isinstance(response, dict):
if error: error = response.get('error')
raise ExtractorError(error, expected=True) if error:
raise ExtractorError(error, expected=True)
def _download_webpage(self, *args, **kwargs): def _download_webpage(self, *args, **kwargs):
webpage = super(YandexMusicBaseIE, self)._download_webpage(*args, **kwargs) webpage = super(YandexMusicBaseIE, self)._download_webpage(*args, **kwargs)

Loading…
Cancel
Save