Spaces:
Running
Running
create a function called job schedule
Browse files- api_monitor.py +11 -1
api_monitor.py
CHANGED
@@ -6,6 +6,7 @@ import psycopg2
|
|
6 |
import psycopg2.extras
|
7 |
import os
|
8 |
from dotenv import load_dotenv
|
|
|
9 |
|
10 |
# Load environment variables from .env file
|
11 |
load_dotenv(override=True)
|
@@ -36,7 +37,7 @@ def cleanup_old_configurations():
|
|
36 |
"""
|
37 |
DELETE FROM api_configurations
|
38 |
WHERE stop_at IS NOT NULL
|
39 |
-
AND stop_at < NOW() - INTERVAL '
|
40 |
""",
|
41 |
]
|
42 |
|
@@ -62,6 +63,13 @@ def cleanup_old_configurations():
|
|
62 |
conn.close()
|
63 |
|
64 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
65 |
def validate_api_configuration(
|
66 |
mcp_api_key,
|
67 |
name,
|
@@ -690,6 +698,8 @@ def retrieve_monitored_data(config_id, mcp_api_key, mode="summary"):
|
|
690 |
## testing
|
691 |
if __name__ == "__main__":
|
692 |
cleanup_old_configurations()
|
|
|
|
|
693 |
# validation_response = validate_api_configuration(
|
694 |
# mcp_api_key="your_api_key",
|
695 |
# name="Dog Facts API",
|
|
|
6 |
import psycopg2.extras
|
7 |
import os
|
8 |
from dotenv import load_dotenv
|
9 |
+
from apscheduler.schedulers.blocking import BlockingScheduler
|
10 |
|
11 |
# Load environment variables from .env file
|
12 |
load_dotenv(override=True)
|
|
|
37 |
"""
|
38 |
DELETE FROM api_configurations
|
39 |
WHERE stop_at IS NOT NULL
|
40 |
+
AND stop_at < NOW() - INTERVAL '14 days';
|
41 |
""",
|
42 |
]
|
43 |
|
|
|
63 |
conn.close()
|
64 |
|
65 |
|
66 |
+
def job_schedule():
|
67 |
+
sched = BlockingScheduler()
|
68 |
+
sched.add_job(cleanup_old_configurations, "cron", hour=0, minute=0)
|
69 |
+
print("cleanup job scheduled at 00:00 UTC")
|
70 |
+
sched.start()
|
71 |
+
|
72 |
+
|
73 |
def validate_api_configuration(
|
74 |
mcp_api_key,
|
75 |
name,
|
|
|
698 |
## testing
|
699 |
if __name__ == "__main__":
|
700 |
cleanup_old_configurations()
|
701 |
+
job_schedule()
|
702 |
+
|
703 |
# validation_response = validate_api_configuration(
|
704 |
# mcp_api_key="your_api_key",
|
705 |
# name="Dog Facts API",
|