-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathrun_scheduler.py
More file actions
52 lines (43 loc) · 1.3 KB
/
run_scheduler.py
File metadata and controls
52 lines (43 loc) · 1.3 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import asyncio
import os
import sys
import logging
from dotenv import load_dotenv
# Configure logging
logging.basicConfig(
level=logging.INFO,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)
# Add current directory to path so we can import from services
sys.path.append(os.getcwd())
# Load environment variables
load_dotenv()
try:
from services.db import connect_db, disconnect_db
from services.scheduler import scheduler_loop
except ImportError as e:
logger.error(f"Failed to import services: {e}")
sys.exit(1)
async def main():
"""Run the scheduler worker."""
logger.info("🚀 Starting Scheduler Worker...")
# Connect to database
await connect_db()
try:
# Run the scheduler loop
await scheduler_loop()
except KeyboardInterrupt:
logger.info("🛑 Scheduler stopped by user")
except Exception as e:
logger.error(f"💥 Scheduler crashed: {e}", exc_info=True)
finally:
await disconnect_db()
logger.info("👋 Scheduler shutdown complete")
if __name__ == "__main__":
if sys.platform == 'win32':
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
try:
asyncio.run(main())
except KeyboardInterrupt:
pass