def check_invalid(): # 获取存放路径 config_path = os.path.join(os.getcwd(), "..\\common\\config.ini") config = robot.read_config(config_path) save_data_path = robot.get_config(config, "SAVE_DATA_PATH", "info/save.data", 3) save_data_dir = os.path.dirname(save_data_path) fee_save_data_path = os.path.join(save_data_dir, "fee.data") # 读取村存档中的收费相册列表 if not os.path.exists(fee_save_data_path): log.step("收费相册存档不存在") return fee_save_data_file = open(fee_save_data_path, "r") fee_save_data = fee_save_data_file.read() fee_save_data_file.close() fee_album_id_list = fee_save_data.strip().split(" ") new_fee_album_id_list = [] # 循环访问,判断相册是否已经被删除 for fee_album_id in fee_album_id_list: album_url = "http://meituzz.com/album/browse?albumID=%s" % fee_album_id album_page_return_code, album_page = tool.http_request(album_url)[:2] if album_page_return_code == 1: if album_page.find("<title>相册已被删除</title>") == -1: new_fee_album_id_list.append(fee_album_id) else: log.step("第%s页相册已被删除" % fee_album_id) # 重新保存 fee_save_data_file = open(fee_save_data_path, "w") fee_save_data_file.write(" ".join(new_fee_album_id_list) + " ") fee_save_data_file.close()
def get_auth_token(): from common import robot config = robot.read_config(os.path.join(os.getcwd(), "..\\common\\config.ini")) # 操作系统&浏览器 browser_type = robot.get_config(config, "BROWSER_TYPE", 2, 1) # cookie is_auto_get_cookie = robot.get_config(config, "IS_AUTO_GET_COOKIE", True, 4) if is_auto_get_cookie: cookie_path = robot.tool.get_default_browser_cookie_path(browser_type) else: cookie_path = robot.get_config(config, "COOKIE_PATH", "", 0) return tool.get_cookie_value_from_browser("auth_token", cookie_path, browser_type, (".twitter.com",))
def set_session_id(): global SESSION_ID from common import robot config = robot.read_config(os.path.join(os.getcwd(), "..\\common\\config.ini")) # 操作系统&浏览器 browser_type = robot.get_config(config, "BROWSER_TYPE", 2, 1) # cookie is_auto_get_cookie = robot.get_config(config, "IS_AUTO_GET_COOKIE", True, 4) if is_auto_get_cookie: cookie_path = robot.tool.get_default_browser_cookie_path(browser_type) else: cookie_path = robot.get_config(config, "COOKIE_PATH", "", 0) session_id = tool.get_cookie_value_from_browser("sessionid", cookie_path, browser_type, ("instagram.com",)) if session_id: SESSION_ID = session_id