forked from LengKundee/MQL5-Google-Onedrive
6a038d53e0
- Updated `docs/NOTEBOOK_LM_CONTEXT.txt` with user-provided NotebookLM URL and context directives.
- Enhanced `scripts/market_research.py` and `scripts/upgrade_repo.py`:
- Added proper Jules integration using the REST API structure `{"model": ..., "prompt": ...}` to match MQL5 implementation.
- Added warning suppression for deprecated `google.generativeai` warnings.
- Added robustness checks for API keys.
- Enhanced `scripts/schedule_research.py` with file logging (`logs/scheduler.log`) and next-run prediction.
83 lines
2.2 KiB
Python
Executable file
83 lines
2.2 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()
|
|
time.sleep(60)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|