add re-enter cookies

catch-the-fish.py

@@ -10,7 +10,8 @@ total_count = 0
 my_catch_count = 0
 
 flarum_remember = "izvHQwoKddgwYWC9e1dFLfcADCOjMnhl2xIEk2BP"
-flarum_session = "c8zulxRXQjrOMFYNQOUv0ze2H9Lg57OLWtFLYCow"
+flarum_session = "Ue1XwF5tYIAJO2FcY7cDu3KtbsB5Mzv47GIJOmyY"
+X_Csrf_Token = "PnU3bv9gJn8Azv6oqnJvaHH3V7zrGt4RiVyULzGZ"
 
 cookies = {
 #   "flarum_remember": "izvHQwoKddgwYWC9e1dFLfcADCOjMnhl2xIEk2BP",
@@ -52,9 +53,19 @@ headers1 = {
     "Sec-Fetch-Mode": "cors",
     "Sec-Fetch-Site": "same-origin",
     "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/113.0.0.0 Safari/537.36",
-    "X-Csrf-Token": "ymGMXtNP0LQ371ogq2Du3exUiWX1JU4B8BWtG15i"
+    "X-Csrf-Token": f"{X_Csrf_Token}"
 }
 
+def input_cookies():
+    global cookies, headers1
+
+    flarum_remember = input("请输入flarum_remember的值：")
+    flarum_session = input("请输入flarum_session的值：")
+    csrf_token = input("请输入X-Csrf-Token的值：")
+
+    cookies["flarum_remember"] = flarum_remember
+    cookies["flarum_session"] = flarum_session
+    headers1["X-Csrf-Token"] = csrf_token
 
 for offset in range(0, 2001, 20):
     url = base_url + str(offset)
@@ -97,9 +108,13 @@ for offset in range(0, 2001, 20):
             response = requests.post(fish_url, json=data, headers=headers1)
 
             if response.status_code != 200:
-                print("                                                                                 ", end="\r", flush=True)
-                print("Cookies过期")
-                sys.exit()
+                print("Cookies过期，请重新输入")
+                input_cookies()
+                response = requests.post(fish_url, json=data, headers=headers1)
+
+                if response.status_code != 200:
+                    print("重新输入的Cookies无效，请检查输入是否正确")
+                    sys.exit()
 
 #            print(response.status_code)  # 打印响应状态码
 #            print(response.json())  # 打印响应内容