update

utils/update_submit.py

@@ -0,0 +1,26 @@
+import json
+import os
+import sys
+
+from utils.logger import logger
+from utils.request import requests_retry_session
+
+lb_headers = {"Content-Type": "application/json"}
+if os.getenv("LEADERBOARD_API_TOKEN"):
+    lb_headers["Authorization"] = "Bearer " + os.getenv("LEADERBOARD_API_TOKEN")
+
+
+def change_product_available() -> None:
+    logger.info("更改为产品不可用...")
+    submit_id = str(os.getenv("SUBMIT_ID", -1))
+    try:
+        requests_retry_session().post(
+            os.getenv(
+                "UPDATE_SUBMIT_URL", "http://contest.4pd.io:8080/submit/update"
+            ),
+            data=json.dumps({submit_id: {"product_avaliable": 0}}),
+            headers=lb_headers,
+        )
+    except Exception as e:
+        logger.error(f"change product avaliable error, {e}")
+        sys.exit(1)