update

utils/leaderboard.py

@@ -0,0 +1,38 @@
+from utils.request import requests_retry_session
+import os
+import json
+import traceback
+from utils.logger import logger
+
+lb_headers = {"Content-Type":"application/json"}
+if os.getenv("LEADERBOARD_API_TOKEN"):
+    lb_headers['Authorization'] = 'Bearer ' + os.getenv("LEADERBOARD_API_TOKEN")
+
+
+def change_product_unavailable() -> 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(traceback.format_exc())
+        logger.error(f"change product avaliable error, {e}")
+
+
+def mark_evaluating(task_id) -> None:
+    logger.info("上报EVALUATING状态...")
+    job_id = os.getenv('JOB_ID') or "-1"
+    url = os.getenv("REGISTER_MARK_TASK_URL", "http://contest.4pd.io:8080/job/register_mark_task") + "/" + job_id
+    try:
+        requests_retry_session().post(
+            url,
+            data=json.dumps({"task_id": task_id}),
+            headers=lb_headers,
+        )
+    except Exception as e:
+        logger.error(traceback.format_exc())
+        logger.error(f"mark evaluating error, {e}")