def save_cookies(): """ 录入账号和cookies :return: """ print('*' * 40) for website in ACCOUNTS.keys(): print('%s正在获取cookies...' % website) acc = RedisClient('accounts', website) coo = RedisClient('cookies', website) for username, password in acc.get_all().items(): log = eval(website.capitalize() + 'Login()') log.login(username, password) cookies = log.get_cookies() coo.set(username, cookies) print('%s的所有cookies已保存成功...' % website) print('*' * 40)
def run(): for website in ACCOUNTS.keys(): print('*' * 40) print('%s正在检测cookies...' % website) acc = RedisClient('accounts', website) coo = RedisClient('cookies', website) log = eval(website.capitalize() + 'Login()') for username, password in acc.get_all().items(): # 第一次运行时,可能数据还未录入就开始测试,因此加异常处理 try: cookies = coo.get(username) if log.good_cookies(cookies): print('\t%s\t\tCookies已通过检测...' % username) else: print('\t%s\t\tCookies未通过检测!!!' % username) coo.delete(username) print('\t%s\t\tCookies已删除!!!' % username) log.login(username, password) cookies = log.get_cookies() coo.set(username, cookies) except Exception as e: print('数据库为空,请等数据录入之后再进行测试:', e.args)