Spaces:
Sleeping
Sleeping
File size: 2,974 Bytes
e06a21d 1dd0906 e06a21d |
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 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 |
#!/usr/bin/env python3
"""
Production startup script for CVE Fact Checker.
This script handles environment setup and graceful startup.
"""
import os
import sys
import time
import signal
import subprocess
from pathlib import Path
def setup_signal_handlers():
"""Setup graceful shutdown handlers."""
def signal_handler(signum, frame):
print(f"\n🛑 Received signal {signum}, shutting down gracefully...")
sys.exit(0)
signal.signal(signal.SIGTERM, signal_handler)
signal.signal(signal.SIGINT, signal_handler)
def run_health_check():
"""Run basic health check before starting."""
print("🏥 Running pre-startup health check...")
try:
result = subprocess.run([
sys.executable, "health_check.py"
], capture_output=True, text=True, timeout=30)
if result.returncode == 0:
print("✅ Health check passed")
return True
else:
print("⚠️ Health check warnings:")
print(result.stdout)
print(result.stderr)
return True # Continue anyway
except Exception as e:
print(f"⚠️ Health check failed: {e}")
return True # Continue anyway
def start_application():
"""Start the application with gunicorn."""
print("🚀 Starting CVE Fact Checker...")
# Environment setup
env = os.environ.copy()
env.update({
'PYTHONUNBUFFERED': '1',
'PYTHONDONTWRITEBYTECODE': '1',
})
# Gunicorn command
cmd = [
"gunicorn",
"-w", "1", # Single worker to avoid race conditions
"-k", "gthread",
"--threads", "4",
"-b", f"0.0.0.0:{env.get('PORT', '7860')}",
"--timeout", "180",
"--access-logfile", "-", # Log to stdout
"--error-logfile", "-", # Log to stderr
"cve_factchecker.wsgi:application"
]
print(f"📋 Command: {' '.join(cmd)}")
print(f"🌐 Will bind to port {env.get('PORT', '7860')}")
try:
# Run gunicorn
subprocess.run(cmd, env=env, check=True)
except KeyboardInterrupt:
print("\n🛑 Interrupted by user")
except subprocess.CalledProcessError as e:
print(f"❌ Application failed with exit code {e.returncode}")
sys.exit(e.returncode)
except Exception as e:
print(f"❌ Unexpected error: {e}")
sys.exit(1)
def main():
"""Main startup routine."""
print("=" * 60)
print("🤖 CVE Fact Checker - Production Startup")
print("=" * 60)
print(f"⏰ Starting at: {time.strftime('%Y-%m-%d %H:%M:%S')}")
print(f"🐍 Python: {sys.version}")
print(f"📍 Working directory: {os.getcwd()}")
print()
setup_signal_handlers()
# Optional health check
if os.path.exists("health_check.py"):
run_health_check()
print()
start_application()
if __name__ == "__main__":
main()
|