[globo] handle login errors

This commit is contained in:
Remita Amine 2018-05-21 15:07:24 +01:00
parent db2058f63e
commit e518749300
1 changed files with 20 additions and 11 deletions

View File

@ -8,7 +8,10 @@ import random
import re import re
from .common import InfoExtractor from .common import InfoExtractor
from ..compat import compat_str from ..compat import (
compat_HTTPError,
compat_str,
)
from ..utils import ( from ..utils import (
ExtractorError, ExtractorError,
float_or_none, float_or_none,
@ -71,6 +74,7 @@ class GloboIE(InfoExtractor):
if email is None: if email is None:
return return
try:
self._download_json( self._download_json(
'https://login.globo.com/api/authentication', None, data=json.dumps({ 'https://login.globo.com/api/authentication', None, data=json.dumps({
'payload': { 'payload': {
@ -81,6 +85,11 @@ class GloboIE(InfoExtractor):
}).encode(), headers={ }).encode(), headers={
'Content-Type': 'application/json; charset=utf-8', 'Content-Type': 'application/json; charset=utf-8',
}) })
except ExtractorError as e:
if isinstance(e.cause, compat_HTTPError) and e.cause.code == 401:
resp = self._parse_json(e.cause.read(), None)
raise ExtractorError(resp.get('userMessage') or resp['id'], expected=True)
raise
self._LOGGED_IN = True self._LOGGED_IN = True
def _real_extract(self, url): def _real_extract(self, url):