JamesyKim commited on
Commit
693029a
·
1 Parent(s): 6ee0f62

create a function called job schedule

Browse files
Files changed (1) hide show
  1. 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 '2 secs';
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",