artem/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2026-05-02 05:25:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

slight refactor/formatting

Browse Source
This commit is contained in:
bashonly 2025-04-28 22:26:36 +00:00 committed by GitHub
parent 20be1c0fff
commit d29862b6e9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
yt_dlp/extractor/niconico.py
View File

@ -62,12 +62,11 @@ class NiconicoBaseIE(InfoExtractor):
}, data=urlencode_postdata({ }, data=urlencode_postdata({
'mail_tel': username, 'mail_tel': username,
'password': password, 'password': password,
}), }))
)
if self.is_logged_in: if self.is_logged_in:
return return
elif err_msg := traverse_obj(webpage, (
if err_msg := traverse_obj(webpage, (
{find_element(cls='notice error')}, {find_element(cls='notice__text')}, {clean_html}, {find_element(cls='notice error')}, {find_element(cls='notice__text')}, {clean_html},
)): )):
self._raise_login_error(err_msg, 'Invalid username or password') self._raise_login_error(err_msg, 'Invalid username or password')
@ -80,17 +79,17 @@ class NiconicoBaseIE(InfoExtractor):
'Content-Type': 'application/x-www-form-urlencoded', 'Content-Type': 'application/x-www-form-urlencoded',
}, data=urlencode_postdata({ }, data=urlencode_postdata({
'otp': self._get_tfa_info('6 digit number shown on app'), 'otp': self._get_tfa_info('6 digit number shown on app'),
}), }))
) if self.is_logged_in:
return
if 'error-code' in parse_qs(urlh.url): elif 'error-code' in parse_qs(urlh.url):
err_msg = traverse_obj(mfa, ({find_element(cls='pageMainMsg')}, {clean_html})) err_msg = traverse_obj(mfa, ({find_element(cls='pageMainMsg')}, {clean_html}))
self._raise_login_error(err_msg, 'MFA session expired') self._raise_login_error(err_msg, 'MFA session expired')
if 'formError' in mfa: elif 'formError' in mfa:
err_msg = traverse_obj(mfa, ( err_msg = traverse_obj(mfa, (
{find_element(cls='formError')}, {find_element(tag='div')}, {clean_html})) {find_element(cls='formError')}, {find_element(tag='div')}, {clean_html}))
self._raise_login_error(err_msg, 'MFA challenge failed') self._raise_login_error(err_msg, 'MFA challenge failed')
else:
self._raise_login_error(None, 'Unexpected login error', expected=False) self._raise_login_error(None, 'Unexpected login error', expected=False)