manage.py

#!/usr/bin/env python
from config import Config
import subprocess
import argparse
import sys
import os

# Ensure the root directory is in sys.path
sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))

def run(workers="1", host="localhost", port=8080, reload=False):
    """
    Runs the application.
    """
    # Validate.
    Config.validate()

    if reload:
        from app import create_app
        app = create_app()
        app.run(
            host=host,
            port=port,
            debug=True,
            extra_files=[
                'templates/**/*.html',
                'static/**/*.scss'
            ]
        )
    else:
        subprocess.run([
            "gunicorn", 
            "--bind", f"{host}:{port}", 
            "--workers", workers,
            "--worker-class", "gevent",
            "--timeout", "300", # Increase timeout to 5 minutes.
            "--keep-alive", "5", # Keep connections alive for 5 seconds.
            "--max-requests", "1000", # Restart workers after 1000 requests.
            "--max-requests-jitter", "50", # Add some randomness to prevent all workers from restarting at once.
            "--worker-connections", "1000", # Maximum number of simultaneous connections per worker.
            "--graceful-timeout", "120", # Give workers 2 minutes to finish their work.
            "--log-level", "info", # Log level for Gunicorn.
            "--threads", "4", # Threads per worker.
            "app:app"
        ])

def main():
    """
    Main function to parse arguments and run the appropriate command.
    """

    parser = argparse.ArgumentParser(description='Flask Management Script')
    subparsers = parser.add_subparsers(dest='command', help='Available commands')

    # Run command
    run_parser = subparsers.add_parser('run', help='Run the application')
    run_parser.add_argument('--workers', default='1', help='Number of workers to run with Gunicorn')
    run_parser.add_argument('--host', default='localhost', help='Host to run the application on')
    run_parser.add_argument('--port', type=int, default=8080, help='Port to run the application on')
    run_parser.add_argument('--reload', action='store_true', help='Run with development server')

    args = parser.parse_args()

    if args.command == 'run':
        print("Starting server...")
        print(f"Starting with reload: {args.reload}")
        run(args.workers, args.host, args.port, args.reload)
    else:
        parser.print_help()

if __name__ == "__main__":
    main()