initial commit

utils/platform_tools.py

@@ -0,0 +1,23 @@
+import os
+from utils.logger import logger
+import requests
+import json
+
+
+def mark_not_available():
+    headers = {"Content-Type": "application/json"}
+    if os.getenv("LEADERBOARD_API_TOKEN"):
+        headers["Authorization"] = "Bearer " + os.getenv("LEADERBOARD_API_TOKEN")
+
+    logger.info("更改为产品不可用...")
+    try:
+        submit_id = str(os.getenv("SUBMIT_ID", "-1"))
+        resp = requests.post(
+            os.getenv("UPDATE_SUBMIT_URL", "http://contest.4pd.io:8080/submit/update"),
+            data=json.dumps({submit_id: {"product_avaliable": 0}}),
+            headers=headers,
+            timeout=600,
+        )
+        logger.info(resp.json())
+    except Exception as e:
+        logger.error(f"change product available error, {e}")