[vk] Improve login

This commit is contained in:
Sergey M․ 2016-07-13 21:52:52 +07:00
parent 2d19fb5072
commit 5f5a9d6158
No known key found for this signature in database
GPG Key ID: 2C393E0F18A9236D

View File

@ -53,13 +53,14 @@ class VKBaseIE(InfoExtractor):
# We will workaround this VK issue by resetting the remixlhk cookie to # We will workaround this VK issue by resetting the remixlhk cookie to
# the first one manually. # the first one manually.
cookies = url_handle.headers.get('Set-Cookie') cookies = url_handle.headers.get('Set-Cookie')
if sys.version_info[0] >= 3: if cookies:
cookies = cookies.encode('iso-8859-1') if sys.version_info[0] >= 3:
cookies = cookies.decode('utf-8') cookies = cookies.encode('iso-8859-1')
remixlhk = re.search(r'remixlhk=(.+?);.*?\bdomain=(.+?)(?:[,;]|$)', cookies) cookies = cookies.decode('utf-8')
if remixlhk: remixlhk = re.search(r'remixlhk=(.+?);.*?\bdomain=(.+?)(?:[,;]|$)', cookies)
value, domain = remixlhk.groups() if remixlhk:
self._set_cookie(domain, 'remixlhk', value) value, domain = remixlhk.groups()
self._set_cookie(domain, 'remixlhk', value)
login_page = self._download_webpage( login_page = self._download_webpage(
'https://login.vk.com/?act=login', None, 'https://login.vk.com/?act=login', None,