Files
POTE/scripts/monitor_market.py
T
ilia 5df21d82f4
CI / skip-ci-check (pull_request) Successful in 19s
CI / secret-scan (pull_request) Successful in 19s
CI / python-ci (pull_request) Successful in 2m38s
Fix lint debt and make CI gates honest
- ruff: fix all 328 errors (autofix + manual); move config to [tool.ruff.lint]
- mypy: fix all errors (annotations, nullable-column guards, stale kwargs
  in weekly report caller)
- black: reformat src/tests so black --check passes
- CI: remove `|| true` from ruff/black/mypy/pytest steps in both
  .gitea and .github workflows; install project deps and add mypy
  step in gitea python-ci so gates actually run
- docs: move 14 root status/guide markdown files into docs/, update links
2026-07-26 15:41:07 -04:00

118 lines
3.9 KiB
Python
Executable File

#!/usr/bin/env python
"""
Real-time market monitoring for congressional tickers.
Run this continuously or on a schedule to detect unusual activity.
"""
import logging
import time
from datetime import datetime
from pathlib import Path
import click
from pote.db import get_session
from pote.monitoring.alert_manager import AlertManager
from pote.monitoring.market_monitor import MarketMonitor
logging.basicConfig(
level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)
logger = logging.getLogger(__name__)
@click.command()
@click.option("--tickers", help="Comma-separated list of tickers (default: congressional watchlist)")
@click.option("--interval", default=300, help="Scan interval in seconds (default: 300 = 5 minutes)")
@click.option("--once", is_flag=True, help="Run once and exit (no continuous monitoring)")
@click.option("--min-severity", default=5, help="Minimum severity to report (1-10)")
@click.option("--save-report", help="Save report to file")
@click.option("--lookback", default=5, help="Days of history to analyze (default: 5)")
def main(tickers, interval, once, min_severity, save_report, lookback):
"""Monitor market for unusual activity in congressional tickers."""
session = next(get_session())
monitor = MarketMonitor(session)
alert_mgr = AlertManager(session)
# Parse tickers if provided
ticker_list = None
if tickers:
ticker_list = [t.strip().upper() for t in tickers.split(",")]
logger.info(f"Monitoring {len(ticker_list)} specified tickers")
else:
logger.info("Monitoring congressional watchlist")
def run_scan():
"""Run a single scan."""
logger.info("=" * 80)
logger.info(f"Starting market scan at {datetime.now()}")
logger.info("=" * 80)
try:
# Scan for unusual activity
alerts = monitor.scan_watchlist(tickers=ticker_list, lookback_days=lookback)
if alerts:
logger.info(f"\n🔔 Found {len(alerts)} alerts!")
# Save to database
monitor.save_alerts(alerts)
# Get MarketAlert objects for reporting
from pote.db.models import MarketAlert
alert_objects = (
session.query(MarketAlert)
.order_by(MarketAlert.timestamp.desc())
.limit(len(alerts))
.all()
)
# Filter by severity
filtered = alert_mgr.filter_alerts(alert_objects, min_severity=min_severity)
if filtered:
# Generate report
report = alert_mgr.generate_summary_report(filtered, output_format="text")
print("\n" + report)
# Save report if requested
if save_report:
Path(save_report).write_text(report)
logger.info(f"Report saved to {save_report}")
else:
logger.info(f"No alerts above severity {min_severity}")
else:
logger.info("✅ No unusual activity detected")
except Exception as e:
logger.error(f"Error during scan: {e}", exc_info=True)
logger.info("=" * 80)
logger.info(f"Scan complete at {datetime.now()}")
logger.info("=" * 80)
# Run scan
run_scan()
# Continuous monitoring mode
if not once:
logger.info(f"\n🔄 Continuous monitoring enabled (interval: {interval}s)")
logger.info("Press Ctrl+C to stop\n")
try:
while True:
time.sleep(interval)
run_scan()
except KeyboardInterrupt:
logger.info("\n\n⏹️ Monitoring stopped by user")
else:
logger.info("\n✅ Single scan complete (use --interval for continuous monitoring)")
if __name__ == "__main__":
main()