forked from LengKundee/MQL5-Google-Onedrive
93 lines
2.6 KiB
Python
Executable file
93 lines
2.6 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
"""
|
|
Schedule Research Script
|
|
Runs market research and upgrade suggestions on a schedule.
|
|
"""
|
|
|
|
import schedule
|
|
import time
|
|
import subprocess
|
|
import logging
|
|
import sys
|
|
import os
|
|
from pathlib import Path
|
|
from datetime import datetime, timedelta
|
|
from dotenv import load_dotenv
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[1]
|
|
LOGS_DIR = REPO_ROOT / "logs"
|
|
|
|
# Setup logging
|
|
LOGS_DIR.mkdir(exist_ok=True)
|
|
log_file = LOGS_DIR / "scheduler.log"
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(levelname)s - %(message)s',
|
|
handlers=[
|
|
logging.FileHandler(log_file),
|
|
logging.StreamHandler(sys.stdout)
|
|
]
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def job():
|
|
logger.info("Running scheduled research task...")
|
|
|
|
# Run market research
|
|
research_script = REPO_ROOT / "scripts" / "market_research.py"
|
|
try:
|
|
subprocess.run([sys.executable, str(research_script)], check=True)
|
|
except subprocess.CalledProcessError as e:
|
|
logger.error(f"Market research failed: {e}")
|
|
return
|
|
|
|
# Run upgrade repo
|
|
upgrade_script = REPO_ROOT / "scripts" / "upgrade_repo.py"
|
|
try:
|
|
subprocess.run([sys.executable, str(upgrade_script)], check=True)
|
|
except subprocess.CalledProcessError as e:
|
|
logger.error(f"Upgrade repo failed: {e}")
|
|
return
|
|
|
|
logger.info("Scheduled task completed successfully.")
|
|
|
|
# Calculate next run
|
|
next_run = datetime.now() + timedelta(hours=4)
|
|
logger.info(f"Next scheduled run at: {next_run.strftime('%Y-%m-%d %H:%M:%S')}")
|
|
|
|
def main():
|
|
# Load env vars
|
|
load_dotenv()
|
|
|
|
if not os.environ.get("GEMINI_API_KEY") and not os.environ.get("GOOGLE_API_KEY"):
|
|
logger.warning("Missing GEMINI_API_KEY or GOOGLE_API_KEY in environment.")
|
|
|
|
if not os.environ.get("JULES_API_KEY"):
|
|
logger.warning("Missing JULES_API_KEY in environment.")
|
|
|
|
logger.info("Starting Schedule Research Service...")
|
|
logger.info(f"Logs will be written to {log_file}")
|
|
|
|
# Run immediately on start
|
|
job()
|
|
|
|
# Schedule every 4 hours
|
|
schedule.every(4).hours.do(job)
|
|
|
|
while True:
|
|
schedule.run_pending()
|
|
# Use dynamic sleep based on time until next job
|
|
sleep_time = schedule.idle_seconds()
|
|
if sleep_time is None:
|
|
# No jobs scheduled, wait default interval
|
|
time.sleep(60)
|
|
elif sleep_time > 0:
|
|
# Sleep only as long as needed (max 60 seconds to remain responsive)
|
|
time.sleep(min(sleep_time, 60))
|
|
else:
|
|
# Job ready now, minimal sleep to avoid busy loop
|
|
time.sleep(1)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|