slight refactor/formatting

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

View File

@ -62,12 +62,11 @@ class NiconicoBaseIE(InfoExtractor):
}, data=urlencode_postdata({
'mail_tel': username,
'password': password,
}),
)
}))
if self.is_logged_in:
return
if err_msg := traverse_obj(webpage, (
elif err_msg := traverse_obj(webpage, (
{find_element(cls='notice error')}, {find_element(cls='notice__text')}, {clean_html},
)):
self._raise_login_error(err_msg, 'Invalid username or password')
@ -80,18 +79,18 @@ class NiconicoBaseIE(InfoExtractor):
'Content-Type': 'application/x-www-form-urlencoded',
}, data=urlencode_postdata({
'otp': self._get_tfa_info('6 digit number shown on app'),
}),
)
if 'error-code' in parse_qs(urlh.url):
}))
if self.is_logged_in:
return
elif 'error-code' in parse_qs(urlh.url):
err_msg = traverse_obj(mfa, ({find_element(cls='pageMainMsg')}, {clean_html}))
self._raise_login_error(err_msg, 'MFA session expired')
if 'formError' in mfa:
elif 'formError' in mfa:
err_msg = traverse_obj(mfa, (
{find_element(cls='formError')}, {find_element(tag='div')}, {clean_html}))
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)
class NiconicoIE(NiconicoBaseIE):