38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
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}") |