jon1scr
/
yt-dlc
forked from minhngoc25a/yt-dlc
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[dramafever] Fix authentication (closes #16067)

This commit is contained in:
Sergey M․ 2018-04-02 22:19:42 +07:00
parent 86693c4930
commit 8bd1df3c31
No known key found for this signature in database
GPG Key ID: 2C393E0F18A9236D
1 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
youtube_dl/extractor/dramafever.py
View File

@ -2,9 +2,11 @@
from __future__ import unicode_literals from __future__ import unicode_literals
import itertools import itertools
import json
from .common import InfoExtractor from .common import InfoExtractor
from ..compat import ( from ..compat import (
compat_HTTPError,
compat_str, compat_str,
compat_urlparse, compat_urlparse,
) )
@ -14,14 +16,11 @@ from ..utils import (
int_or_none, int_or_none,
parse_age_limit, parse_age_limit,
parse_duration, parse_duration,
sanitized_Request,
unified_timestamp, unified_timestamp,
urlencode_postdata
) )
class DramaFeverBaseIE(InfoExtractor): class DramaFeverBaseIE(InfoExtractor):
_LOGIN_URL = 'https://www.dramafever.com/accounts/login/'
_NETRC_MACHINE = 'dramafever' _NETRC_MACHINE = 'dramafever'
_CONSUMER_SECRET = 'DA59dtVXYLxajktV' _CONSUMER_SECRET = 'DA59dtVXYLxajktV'
@ -39,8 +38,8 @@ class DramaFeverBaseIE(InfoExtractor):
'consumer secret', default=self._CONSUMER_SECRET) 'consumer secret', default=self._CONSUMER_SECRET)
def _real_initialize(self): def _real_initialize(self):
self._login()
self._consumer_secret = self._get_consumer_secret() self._consumer_secret = self._get_consumer_secret()
self._login()
def _login(self): def _login(self):
(username, password) = self._get_login_info() (username, password) = self._get_login_info()
@ -52,18 +51,28 @@ class DramaFeverBaseIE(InfoExtractor):
'password': password, 'password': password,
} }
request = sanitized_Request( try:
self._LOGIN_URL, urlencode_postdata(login_form)) response = self._download_json(
response = self._download_webpage( 'https://www.dramafever.com/api/users/login', None, 'Logging in',
request, None, 'Logging in') data=json.dumps(login_form).encode('utf-8'), headers={
'x-consumer-key': self._consumer_secret,
})
except ExtractorError as e:
if isinstance(e.cause, compat_HTTPError) and e.cause.code in (403, 404):
response = self._parse_json(
e.cause.read().decode('utf-8'), None)
else:
raise
if all(logout_pattern not in response # Successful login
for logout_pattern in ['href="/accounts/logout/"', '>Log out<']): if response.get('result') or response.get('guid') or response.get('user_guid'):
error = self._html_search_regex( return
r'(?s)<h\d[^>]+\bclass="hidden-xs prompt"[^>]*>(.+?)</h\d',
response, 'error message', default=None) errors = response.get('errors')
if error: if errors and isinstance(errors, list):
raise ExtractorError('Unable to login: %s' % error, expected=True) error = errors[0]
message = error.get('message') or error['reason']
raise ExtractorError('Unable to login: %s' % message, expected=True)
raise ExtractorError('Unable to log in') raise ExtractorError('Unable to log in')