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("切换用户失败!")