def sign(self, args):
"""签到 + 抽奖"""
if '-a' in args or '--all' in args:
old_user = self.who()
for user in config.get_users_name():
self.su([user, ])
sleep(0.5)
self._disk.user_sign()
sleep(0.5)
self.su([old_user, ])
else:
self._disk.user_sign()
def su(self, args):
"""列出、切换用户"""
users = config.get_users_name()
def list_user():
for i, user in enumerate(users):
user_info = config.get_user_info(user)
methord = "用户名+密码 登录" if user_info[2] else "Cookie 登录"
print(f"[{i}] 用户名: {user}, {methord}")
if args:
if args[0] == '-l':
list_user()
return None
elif args[0] in users:
select_user = args[0]
else:
error(f"用户名 {args[0]} 无效")
return None
else:
list_user()
select = input("请输入用户序号, [0、1 ... ]: ")
if select.isnumeric():
select = int(select)
if select > len(users):
error(f"序号 {select} 无效!")
return None
select_user = users[select]
else:
error(f"序号 {select} 无效!")
return None
config.work_id = self._work_id # 保存旧的工作目录
result = config.change_user(select_user)
if result and self._disk.login_by_cookie(config) == Cloud189.SUCCESS:
info(f"成功切换至用户 {config.username}")
self.refresh(config.work_id)
else:
error("切换用户失败!")
