From 21666700c2686e5e70e14b7b8ff706f2b051eb53 Mon Sep 17 00:00:00 2001 From: F04C Date: Mon, 15 Sep 2025 10:47:01 +0800 Subject: [PATCH] added PSO calling --- automate.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/automate.py b/automate.py index bc7f452..2533e66 100644 --- a/automate.py +++ b/automate.py @@ -46,12 +46,37 @@ def make_requests() -> None: except requests.exceptions.RequestException as e: logging.error(f"Error while requesting {full_url}: {e}") +def make_pso_request() -> None: + base_url, token = load_env_vars() + + if not base_url or not token: + return + + headers = {"Authorization": f"Bearer {token}"} + full_url = f"{base_url}/api/pso/" + + with requests.Session() as session: + try: + response = session.get(full_url, headers=headers, timeout=10) + response.raise_for_status() + + if response.status_code == 200: + logging.info("PSO Updated!") + logging.info(f"Success: {full_url} - Response: {response.json()}") + except requests.exceptions.Timeout: + logging.error(f"Request to {full_url} timed out") + except requests.exceptions.HTTPError as e: + logging.error(f"HTTP error occurred: {e}") + except requests.exceptions.RequestException as e: + logging.error(f"Error while requesting {full_url}: {e}") + def run_threaded(job_func): job_thread = threading.Thread(target=job_func) job_thread.start() def schedule_daily_request(): schedule.every().day.at("00:00:00").do(run_threaded, make_requests) + schedule.every().day.at("12:00:00").do(run_threaded, make_pso_request) while True: schedule.run_pending()