def load_auth(self, username):
auth_file = os.path.join(Config.get_tmp_path(username), 'auth.json')
if os.path.exists(auth_file):
if time.time() - os.stat(auth_file).st_mtime < DELTA:
with open(auth_file) as fh:
c, tokens = json.load(fh)
cookie = RequestCookie(c)
return cookie, tokens
return None, None
def load_auth(self, username):
auth_file = Config.check_auth_file(username)
# 如果缓存授权信息没过期, 就直接读取它.
if auth_file:
if time.time() - os.stat(auth_file).st_mtime < DELTA:
with open(auth_file) as fh:
c, tokens = json.load(fh)
cookie = RequestCookie(c)
return cookie, tokens
return None, None
def urlopen(url,
headers={},
data=None,
retries=RETRIES,
timeout=TIMEOUT,
cookie=RequestCookie()):
'''打开一个http连接, 并返回Request.
headers 是一个dict. 默认提供了一些项目, 比如User-Agent, Referer等, 就
不需要重复加入了.
这个函数只能用于http请求, 不可以用于下载大文件.
如果服务器支持gzip压缩的话, 就会使用gzip对数据进行压缩, 然后在本地自动
解压.
req.data 里面放着的是最终的http数据内容, 通常都是UTF-8编码的文本.
'''
headers_merged = default_headers.copy()
for key in headers.keys():
headers_merged[key] = headers[key]
redirectHandler = myRedirctHandler(cookie)
opener = urllib.request.build_opener(ForbiddenHandler, redirectHandler)
opener.addheaders = [(k, v) for k, v in headers_merged.items()]
for i in range(retries):
try:
req = opener.open(url, data=data, timeout=timeout)
encoding = req.headers.get('Content-encoding')
req.data = req.read()
if encoding == 'gzip':
req.data = gzip.decompress(req.data)
elif encoding == 'deflate':
req.data = zlib.decompress(req.data, -zlib.MAX_WBITS)
return req
except OSError as e:
logger.error(traceback.format_exc())
except:
logger.error(traceback.format_exc())
return None
def signin(self):
def on_get_bdstoken(bdstoken, error=None):
if error or not bdstoken:
logger.error('SigninDialog.on_get_bdstoken: %s, %s' %
(bdstoken, error))
self.signin_failed(_('Failed to get bdstoken!'))
else:
nonlocal tokens
tokens['bdstoken'] = bdstoken
self.update_profile(username,
password,
cookie,
tokens,
dump=True)
def on_post_login(info, error=None):
if error or not info:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('Login failed, please try again'))
else:
errno, query = info
if errno == 0:
cookie.load_list(query)
self.signin_button.set_label(_('Get bdstoken...'))
gutil.async_call(auth.get_bdstoken,
cookie,
callback=on_get_bdstoken)
# 257: 需要输入验证码
elif errno == 257:
nonlocal verifycode
nonlocal codeString
vcodetype = query['vcodetype']
codeString = query['codeString']
dialog = SigninVcodeDialog(self, username, cookie,
tokens['token'], codeString,
vcodetype)
response = dialog.run()
verifycode = dialog.get_vcode()
codeString = dialog.codeString
dialog.destroy()
self.signin_button.set_label(_('Get bdstoken...'))
gutil.async_call(auth.post_login,
cookie,
tokens,
username,
password_enc,
rsakey,
verifycode,
codeString,
callback=on_post_login)
# 密码错误
elif errno == 4:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('Password error, please try again'))
# 验证码错误
elif errno == 6:
self.signin_failed(
_('Verfication code error, please try again'))
# 需要短信验证
elif errno == 400031:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(
_('Does not support SMS/Email verification!'))
# 登录失败,请在弹出的窗口操作,或重新登录
elif errno == 120021:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('NET:登录失败,请在弹出的窗口操作,或重新登录'))
elif errno == 120019:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(
_('NET:近期登录次数过多, 请先通过 passport.baidu.com 解除锁定'))
elif errno == 500010:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('NET:登录过于频繁,请24小时后再试'))
elif errno == 400031:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('NET:账号异常，请在当前网络环境下在百度网页端正常登录一次'))
else:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('Unknown error, please try again'))
def on_get_public_key(info, error=None):
if not info or error:
logger.error('SigninDialog.on_get_public_key: %s, %s' %
(info, error))
self.signin_failed(
_('Failed to request public key, please try again'))
else:
pubkey = info['pubkey']
nonlocal rsakey
rsakey = info['key']
nonlocal password_enc
password_enc = util.RSA_encrypt(pubkey, password)
gutil.async_call(auth.post_login,
cookie,
tokens,
username,
password_enc,
rsakey,
verifycode,
codeString,
callback=on_post_login)
def on_check_login(info, error=None):
if not info or error:
logger.error('SigninDialog.on_check_login: %s, %s' %
(info, error))
self.signin_failed(
_('Failed to check login, please try again'))
else:
ubi_cookie, status = info
cookie.load_list(ubi_cookie)
nonlocal codeString
nonlocal verifycode
codeString = status['data']['codeString']
vcodetype = status['data']['vcodetype']
if codeString:
dialog = SigninVcodeDialog(self, username, cookie, tokens,
codeString, vcodetype)
response = dialog.run()
verifycode = dialog.get_vcode()
codeString = dialog.codeString
dialog.destroy()
'''中文验证码长度为2,英文验证码长度为4'''
if not verifycode or (len(verifycode) != 4
and len(verifycode) != 2):
self.signin_failed(
_('Please input verification code!'))
return
else:
gutil.async_call(auth.get_public_key,
cookie,
tokens,
callback=on_get_public_key)
else:
gutil.async_call(auth.get_public_key,
cookie,
tokens,
callback=on_get_public_key)
def on_get_UBI(ubi_cookie, error=None):
if error or not ubi_cookie:
logger.error('SigninDialog.on_getUBI: %s, %s' %
(ubi_cookie, error))
self.signin_failed(_('Failed to get UBI, please try again.'))
else:
cookie.load_list(ubi_cookie)
self.signin_button.set_label(_('Check login'))
gutil.async_call(auth.check_login,
cookie,
tokens,
username,
callback=on_check_login)
def on_get_token(info, error=None):
if error or not info:
logger.error('SigninDialog.on_get_token: %s, %s' %
(info, error))
self.signin_failed(_('Failed to get token, please try again.'))
else:
nonlocal tokens
hosupport, token = info
cookie.load_list(hosupport)
cookie.load('cflag=65535%3A1; PANWEB=1;')
tokens['token'] = token
self.signin_button.set_label(_('Get UBI...'))
gutil.async_call(auth.get_UBI,
cookie,
tokens,
callback=on_get_UBI)
def on_get_BAIDUID(uid_cookie, error=None):
if error or not uid_cookie:
logger.error('SigninDialog.on_get_BAIDUID: %s, %s' %
(uid_cookie, error))
self.signin_failed(
_('Failed to get BAIDUID cookie, please try again.'))
else:
cookie.load_list(uid_cookie)
self.signin_button.set_label(_('Get TOKEN...'))
gutil.async_call(auth.get_token, cookie, callback=on_get_token)
username = self.username_combo.get_child().get_text()
password = self.password_entry.get_text()
# 使用本地的缓存token, 有效期是三天
if not self.password_changed and self.signin_check.get_active():
cookie, tokens = self.load_auth(username)
if cookie and tokens:
self.update_profile(username, password, cookie, tokens)
return
cookie = RequestCookie()
tokens = {}
verifycode = ''
codeString = ''
password_enc = ''
rsakey = ''
self.signin_button.set_label(_('Get BAIDUID...'))
gutil.async_call(auth.get_BAIDUID, callback=on_get_BAIDUID)
def signin(self):
def on_get_bdstoken(bdstokens, error=None):
if error or not bdstokens:
self.signin_failed(_('Error: Failed to get bdstokens!'))
else:
nonlocal tokens
for token in bdstokens:
tokens[token] = bdstokens[token]
self.update_profile(username,
password,
cookie,
tokens,
dump=True)
def on_get_bduss(result, error=None):
status, info = result
if status == 4:
self.signin_failed(
_('Please check username and password are correct!'))
elif status == 0:
cookie.load_list(info)
self.signin_button.set_label(_('Get bdstoken...'))
gutil.async_call(auth.get_bdstoken,
cookie,
callback=on_get_bdstoken)
elif status == 257:
vcodetype, codeString = info
dialog = SigninVcodeDialog(self, username, cookie,
tokens['token'], codeString,
vcodetype)
dialog.run()
vcode = dialog.get_vcode()
dialog.destroy()
if not vcode or len(vcode) != 4:
self.signin_failed(_('Please input verification code!'))
return
self.signin_button.set_label(_('Get bduss...'))
gutil.async_call(auth.get_bduss,
cookie,
tokens['token'],
username,
password,
vcode,
codeString,
callback=on_get_bduss)
else:
self.signin_failed(
_('Unknown err_no {0}, please try again!').format(status))
def on_check_login(status, error=None):
if error or not status:
self.signin_failed(
_('Failed to get check login, please try again.'))
elif len(status['data']['codeString']):
codeString = status['data']['codeString']
vcodetype = status['data']['vcodetype']
dialog = SigninVcodeDialog(self, username, cookie,
tokens['token'], codeString,
vcodetype)
dialog.run()
vcode = dialog.get_vcode()
dialog.destroy()
if not vcode or len(vcode) != 4:
self.signin_failed(_('Please input verification code!'))
return
self.signin_button.set_label(_('Get bduss...'))
gutil.async_call(auth.get_bduss,
cookie,
tokens['token'],
username,
password,
vcode,
codeString,
callback=on_get_bduss)
else:
self.signin_button.set_label(_('Get bduss...'))
gutil.async_call(auth.get_bduss,
cookie,
tokens['token'],
username,
password,
callback=on_get_bduss)
def on_get_UBI(ubi_cookie, error=None):
if error or not ubi_cookie:
self.signin_failed(
_('Failed to get UBI cookie, please try again.'))
else:
cookie.load_list(ubi_cookie)
self.signin_button.set_label(_('Get token...'))
gutil.async_call(auth.check_login,
cookie,
tokens['token'],
username,
callback=on_check_login)
def on_get_token(token, error=None):
if error or not token:
self.signin_failed(
_('Failed to get tokens, please try again.'))
else:
nonlocal tokens
tokens['token'] = token
self.signin_button.set_label(_('Get UBI...'))
gutil.async_call(auth.get_UBI,
cookie,
token,
callback=on_get_UBI)
def on_get_BAIDUID(uid_cookie, error=None):
if error or not uid_cookie:
self.signin_failed(
_('Failed to get BAIDUID cookie, please try agin.'))
else:
cookie.load_list(uid_cookie)
self.signin_button.set_label(_('Get BAIDUID...'))
gutil.async_call(auth.get_token, cookie, callback=on_get_token)
username = self.username_combo.get_child().get_text()
password = self.password_entry.get_text()
# 使用本地的缓存token, 有效期是三天
if self.signin_check.get_active():
cookie, tokens = self.load_auth(username)
if cookie and tokens:
self.update_profile(username, password, cookie, tokens)
return
cookie = RequestCookie()
tokens = {}
cookie.load('cflag=65535%3A1; PANWEB=1;')
self.signin_button.set_label(_('Get cookie...'))
gutil.async_call(auth.get_BAIDUID, callback=on_get_BAIDUID)
def signin(self):
def on_get_bdstoken(bdstoken, error=None):
if error or not bdstoken:
logger.error('SigninDialog.on_get_bdstoken: %s, %s' %
(bdstoken, error))
self.signin_failed(_('Failed to get bdstoken!'))
else:
nonlocal tokens
tokens['bdstoken'] = bdstoken
self.update_profile(username,
password,
cookie,
tokens,
dump=True)
def on_post_login(info, error=None):
if error or not info:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('Login failed, please try again'))
else:
errno, query = info
if errno == 0:
cookie.load_list(query)
self.signin_button.set_label(_('Get bdstoken...'))
gutil.async_call(auth.get_bdstoken,
cookie,
callback=on_get_bdstoken)
# 257: 需要输入验证码
elif errno == 257:
nonlocal verifycode
nonlocal codeString
vcodetype = query['vcodetype']
codeString = query['codeString']
dialog = SigninVcodeDialog(self, username, cookie,
tokens['token'], codeString,
vcodetype)
response = dialog.run()
verifycode = dialog.get_vcode()
codeString = dialog.codeString
dialog.destroy()
if not verifycode or len(verifycode) != 4:
self.signin_failed(
_('Please input verification code!'))
return
else:
self.signin_button.set_label(_('Get bdstoken...'))
gutil.async_call(auth.post_login,
cookie,
tokens,
username,
password_enc,
rsakey,
verifycode,
codeString,
callback=on_post_login)
# 密码错误
elif errno == 4:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('Password error, please try again'))
# 验证码错误
elif errno == 6:
self.signin_failed(
_('Verfication code error, please try again'))
# 需要短信验证
elif errno == 400031:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(
_('Does not support SMS/Email verification!'))
else:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('Unknown error, please try again'))
def on_get_public_key(info, error=None):
if not info or error:
logger.error('SigninDialog.on_get_public_key: %s, %s' %
(info, error))
self.signin_failed(
_('Failed to request public key, please try again'))
else:
pubkey = info['pubkey']
nonlocal rsakey
rsakey = info['key']
nonlocal password_enc
password_enc = util.RSA_encrypt(pubkey, password)
gutil.async_call(auth.post_login,
cookie,
tokens,
username,
password_enc,
rsakey,
verifycode,
codeString,
callback=on_post_login)
def on_check_login(info, error=None):
if not info or error:
logger.error('SigninDialog.on_check_login: %s, %s' %
(info, error))
self.signin_failed(
_('Failed to check login, please try again'))
else:
ubi_cookie, status = info
cookie.load_list(ubi_cookie)
nonlocal codeString
nonlocal verifycode
codeString = status['data']['codeString']
vcodetype = status['data']['vcodetype']
if codeString:
dialog = SigninVcodeDialog(self, username, cookie, tokens,
codeString, vcodetype)
response = dialog.run()
verifycode = dialog.get_vcode()
codeString = dialog.codeString
dialog.destroy()
if not verifycode or len(verifycode) != 4:
self.signin_failed(
_('Please input verification code!'))
return
else:
gutil.async_call(auth.get_public_key,
cookie,
tokens,
callback=on_get_public_key)
else:
gutil.async_call(auth.get_public_key,
cookie,
tokens,
callback=on_get_public_key)
def on_get_UBI(ubi_cookie, error=None):
if error or not ubi_cookie:
logger.error('SigninDialog.on_getUBI: %s, %s' %
(ubi_cookie, error))
self.signin_failed(_('Failed to get UBI, please try again.'))
else:
cookie.load_list(ubi_cookie)
self.signin_button.set_label(_('Check login'))
gutil.async_call(auth.check_login,
cookie,
tokens,
username,
callback=on_check_login)
def on_get_token(info, error=None):
if error or not info:
logger.error('SigninDialog.on_get_token: %s, %s' %
(info, error))
self.signin_failed(_('Failed to get token, please try again.'))
else:
nonlocal tokens
hosupport, token = info
cookie.load_list(hosupport)
cookie.load('cflag=65535%3A1; PANWEB=1;')
tokens['token'] = token
self.signin_button.set_label(_('Get UBI...'))
gutil.async_call(auth.get_UBI,
cookie,
tokens,
callback=on_get_UBI)
def on_get_BAIDUID(uid_cookie, error=None):
if error or not uid_cookie:
logger.error('SigninDialog.on_get_BAIDUID: %s, %s' %
(uid_cookie, error))
self.signin_failed(
_('Failed to get BAIDUID cookie, please try agin.'))
else:
cookie.load_list(uid_cookie)
self.signin_button.set_label(_('Get TOKEN...'))
gutil.async_call(auth.get_token, cookie, callback=on_get_token)
username = "******"
password = "******"
# 使用本地的缓存token, 有效期是三天
cookie = RequestCookie()
tokens = {}
verifycode = ''
codeString = ''
password_enc = ''
rsakey = ''
self.signin_button.set_label(_('Get BAIDUID...'))
gutil.async_call(auth.get_BAIDUID, callback=on_get_BAIDUID)
def signin(self):
def on_get_bdstoken(bdstoken, error=None):
if error or not bdstoken:
print('Error in get bdstoken:', bdstoken, error)
print('Please check your username and passowrd')
self.signin_failed(_('Error: Failed to get bdstokens!'))
else:
nonlocal tokens
tokens['bdstoken'] = bdstoken
self.update_profile(username,
password,
cookie,
tokens,
dump=True)
def on_wap_signin(cookie_str, error):
if not cookie_str or error:
print('Error in on_wap_signin():', cookie_str, error)
self.signin_failed(_('Failed to signin, please try again.'))
else:
cookie.load_list(cookie_str)
self.signin_button.set_label(_('Get bdstoken...'))
gutil.async_call(auth.get_bdstoken,
cookie,
callback=on_get_bdstoken)
def on_get_wap_passport(info, error=None):
if error or not info:
print('Error occurs in on_get_wap_passport:', info, error)
self.signin_failed(
_('Failed to get WAP page, please try again.'))
cookie_str, _form = info
if not cookie_str or not _form:
print('Error occurs in on_get_wap_passport:', info, error)
self.signin_failed(
_('Failed to get WAP page, please try again.'))
else:
nonlocal form
form = _form
form['username'] = username
form['password'] = password
cookie.load_list(cookie_str)
if len(form.get('vcodestr', '')):
dialog = SigninVcodeDialog(self, cookie, form)
dialog.run()
dialog.destroy()
if len(form.get('verifycode', '')) != 4:
print('verifycode length is not 4!')
return
self.signin_button.set_label(_('Signin...'))
gutil.async_call(auth.wap_signin,
cookie,
form,
callback=on_wap_signin)
def on_get_token(token, error=None):
if error or not token:
print('Error in get token():', token, error)
self.signin_failed(
_('Failed to get tokens, please try again.'))
else:
nonlocal tokens
tokens['token'] = token
self.signin_button.set_label(_('Get WAP page...'))
gutil.async_call(auth.get_wap_passport,
callback=on_get_wap_passport)
def on_get_BAIDUID(uid_cookie, error=None):
if error or not uid_cookie:
print('Error in get BAIDUID():', uid_cookie, error)
self.signin_failed(
_('Failed to get BAIDUID cookie, please try agin.'))
else:
cookie.load_list(uid_cookie)
self.signin_button.set_label(_('Get TOKEN...'))
gutil.async_call(auth.get_token, cookie, callback=on_get_token)
username = self.username_combo.get_child().get_text()
password = self.password_entry.get_text()
# 使用本地的缓存token, 有效期是三天
if not self.password_changed and self.signin_check.get_active():
cookie, tokens = self.load_auth(username)
if cookie and tokens:
self.update_profile(username, password, cookie, tokens)
return
cookie = RequestCookie()
cookie.load('cflag=65535%3A1; PANWEB=1;')
tokens = {}
form = {}
self.signin_button.set_label(_('Get BAIDUID...'))
gutil.async_call(auth.get_BAIDUID, callback=on_get_BAIDUID)
def signin(self, auto_signin=True):
self.profile = None
self.screen.nodelay(0)
self.screen.addstr(2, 2, "Baidu Yun Login")
self.screen.addstr(4, 2, "Please input your ID and Password")
self.screen.addstr(6, 2, "Press Any Key to continue ... ")
self.screen.refresh()
self.screen.getch()
curses.echo()
win = curses.newwin(height, width, begin_y, begin_x)
win.border(0)
win.addstr(2, 2, "UserName:"******"Cookie is valid. Press any key to continue.")
win.refresh()
win.getch()
return True
win.addstr(4, 2, "Password: (Won't be shown)")
win.refresh()
curses.noecho()
password = win.getstr(5, 2, 20).decode(encoding='utf-8')
curses.echo()
self.cookie = RequestCookie()
self.tokens = {}
verifycode = ''
codeString = ''
password_enc = ''
rsakey = ''
win.addstr(7, 2, "Get BaiduID...")
win.refresh()
uid_cookie = auth.get_BAIDUID()
if not uid_cookie:
win.addstr(8, 2, " Get BaiduID failed, press any key to exit")
win.refresh()
win.getch()
return False
self.cookie.load_list(uid_cookie)
win.addstr(8, 2, "Get token...")
win.refresh()
info = auth.get_token(self.cookie)
if not info:
win.addstr(9, 2, " Get Token failed, press any key to exit")
win.refresh()
win.getch()
return False
hosupport, token = info
self.cookie.load_list(hosupport)
self.cookie.load('cflag=65535%3A1; PANWEB=1;')
self.tokens['token'] = token
win.addstr(9, 2, "Get UBI...")
win.refresh()
ubi_cookie = auth.get_UBI(self.cookie, self.tokens)
self.cookie.load_list(ubi_cookie)
win.addstr(10, 2, "Check login...")
win.refresh()
ubi_cookie, status = auth.check_login(self.cookie, self.tokens,
username)
self.cookie.load_list(ubi_cookie)
codeString = status['data']['codeString']
vcodetype = status['data']['vcodetype']
if codeString:
win.addstr(11, 2, "Vcode is needed but not supported")
win.addstr(12, 2, "Press any key to quit")
win.refresh()
win.getch()
return False
win.addstr(11, 2, "Get public key...")
win.refresh()
info = auth.get_public_key(self.cookie, self.tokens)
pubkey = info['pubkey']
rsakey = info['key']
password_enc = util.RSA_encrypt(pubkey, password)
win.addstr(12, 2, "login...")
win.refresh()
info = auth.post_login(self.cookie, self.tokens, username,
password_enc, rsakey)
errno, query = info
if errno == 0:
self.cookie.load_list(query)
win.addstr(13, 2, "Get bds_token ...")
win.refresh()
bdstoken = auth.get_bdstoken(self.cookie)
self.tokens['bdstoken'] = bdstoken
win.addstr(14, 2, "Login finished, press any key to continue")
win.refresh()
win.getch()
elif errno == 257:
win.addstr(13, 2, "need vcode, but not supported")
win.addstr(14, 2, "Press any key to quit")
win.refresh()
win.getch()
return False
elif errno == 4:
win.addstr(13, 2, "invalid password!")
win.addstr(14, 2, "Press any key to quit")
win.refresh()
win.getch()
return False
else:
win.addstr(13, 2, "Something is wrong")
win.addstr(14, 2, "Press any key to quit")
win.refresh()
win.getch()
return False
self.dump_auth(self.username, self.cookie, self.tokens)
return True
def __init__(self, cookies=RequestCookie()):
self.cookies = cookies
def signin(self):
def on_get_bdstoken(bdstoken, error=None):
if error or not bdstoken:
logger.error('SigninDialog.on_get_bdstoken: %s, %s' %
(bdstoken, error))
self.signin_failed(_('Failed to get bdstoken!'))
else:
nonlocal tokens
tokens['bdstoken'] = bdstoken
self.update_profile(username,
password,
cookie,
tokens,
dump=True)
def on_get_auth_cookie(info, error=None):
if error or not info:
logger.error('SigninDialog.on_get_auth_cookie: %s, %s' %
(info, error))
self.signin_failed(
_('Fail to get auth cookie, please try again'))
else:
cookie.load_list(info)
self.signin_button.set_label(_('Get bdstoken...'))
gutil.async_call(auth.get_bdstoken,
cookie,
callback=on_get_bdstoken)
def on_post_login(info, error=None):
if error or not info:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('Login failed, please try again'))
else:
errno, query = info
if errno == 0:
cookie.load_list(query)
self.signin_button.set_label(_('Get auth cookie...'))
gutil.async_call(auth.get_auth_cookie,
cookie,
callback=on_get_auth_cookie)
# 257: 需要输入验证码
elif errno == 257:
nonlocal verifycode
nonlocal codeString
vcodetype = query['vcodetype']
codeString = query['codeString']
dialog = SigninVcodeDialog(self, username, cookie, tokens,
codeString, vcodetype)
response = dialog.run()
verifycode = dialog.get_vcode()
codeString = dialog.codeString
dialog.destroy()
if not (verifycode or len(verifycode) == 2
or len(verifycode) == 4):
self.signin_failed(
_('Please input verification code!'))
return
else:
self.signin_button.set_label(_('Get bdstoken...'))
gutil.async_call(auth.post_login,
cookie,
tokens,
username,
password_enc,
rsakey,
verifycode,
codeString,
callback=on_post_login)
# 密码错误
elif errno == 4:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('Password error, please try again'))
# 验证码错误
elif errno == 6:
self.signin_failed(
_('Verfication code error, please try again'))
# 需要手机号验证
elif errno == 18:
self.signin_failed(_('Real name verification required!'))
# 需要安全验证
elif errno == 120021:
self.signin_failed(_('Security verification required!'))
# 需要短信验证
elif errno == 400031:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(
_('Does not support SMS/Email verification!'))
else:
logger.error('SigninDialog.on_post_login: %s, %s' %
(info, error))
self.signin_failed(_('Unknown error, please try again'))
def on_get_public_key(info, error=None):
if not info or error:
logger.error('SigninDialog.on_get_public_key: %s, %s' %
(info, error))
self.signin_failed(
_('Failed to request public key, please try again'))
else:
pubkey = info['pubkey']
nonlocal rsakey
rsakey = info['key']
nonlocal password_enc
password_enc = util.RSA_encrypt(pubkey, password)
gutil.async_call(auth.post_login,
cookie,
tokens,
username,
password_enc,
rsakey,
verifycode,
codeString,
dv,
callback=on_post_login)
def on_check_login(info, error=None):
if not info or error:
logger.error('SigninDialog.on_check_login: %s, %s' %
(info, error))
self.signin_failed(
_('Failed to check login, please try again'))
else:
ubi_cookie, status = info
cookie.load_list(ubi_cookie)
nonlocal codeString
nonlocal verifycode
codeString = status['data']['codeString']
vcodetype = status['data']['vcodetype']
if codeString:
dialog = SigninVcodeDialog(self, username, cookie, tokens,
codeString, vcodetype)
response = dialog.run()
verifycode = dialog.get_vcode()
codeString = dialog.codeString
dialog.destroy()
if not (verifycode or len(verifycode) == 2
or len(verifycode) == 4):
self.signin_failed(
_('Please input verification code!'))
return
else:
gutil.async_call(auth.get_public_key,
cookie,
tokens,
callback=on_get_public_key)
else:
gutil.async_call(auth.get_public_key,
cookie,
tokens,
callback=on_get_public_key)
def on_get_UBI(ubi_cookie, error=None):
if error or not ubi_cookie:
logger.error('SigninDialog.on_getUBI: %s, %s' %
(ubi_cookie, error))
self.signin_failed(_('Failed to get UBI, please try again.'))
else:
cookie.load_list(ubi_cookie)
self.signin_button.set_label(_('Check login'))
gutil.async_call(auth.check_login,
cookie,
tokens,
username,
dv,
callback=on_check_login)
def on_get_token(info, error=None):
if error or not info:
logger.error('SigninDialog.on_get_token: %s, %s' %
(info, error))
self.signin_failed(_('Failed to get token, please try again.'))
else:
nonlocal tokens
hosupport, token = info
cookie.load_list(hosupport)
cookie.load('cflag=15%3A3; PANWEB=1;')
tokens['token'] = token
self.signin_button.set_label(_('Get UBI...'))
gutil.async_call(auth.get_UBI,
cookie,
tokens,
callback=on_get_UBI)
def on_get_BAIDUID(uid_cookie, error=None):
if error or not uid_cookie:
logger.error('SigninDialog.on_get_BAIDUID: %s, %s' %
(uid_cookie, error))
self.signin_failed(
_('Failed to get BAIDUID cookie, please try again.'))
else:
cookie.load_list(uid_cookie)
self.signin_button.set_label(_('Get TOKEN...'))
gutil.async_call(auth.get_token, cookie, callback=on_get_token)
username = self.username_combo.get_child().get_text()
password = self.password_entry.get_text()
# 判断是否使用缓存授权信息
if self.cache_login:
cookie, tokens = self.load_auth(username)
if cookie and tokens:
self.update_profile(username, password, cookie, tokens)
return
cookie = RequestCookie()
tokens = {}
dv = DV.get_new_dv()
verifycode = ''
codeString = ''
password_enc = ''
rsakey = ''
self.signin_button.set_label(_('Get BAIDUID...'))
gutil.async_call(auth.get_BAIDUID, callback=on_get_BAIDUID)
