run.py

import os
import signal
import socket
import time
import logging
import uvicorn
from app.config import settings

PID_FILE = os.path.join(os.path.dirname(__file__), ".server.pid")


def kill_old_server():
    """Kill previous server process using PID file."""
    if not os.path.exists(PID_FILE):
        return
    try:
        with open(PID_FILE) as f:
            old_pid = int(f.read().strip())
        os.kill(old_pid, signal.SIGKILL)
        print(f"  Killed old server (PID {old_pid})")
        time.sleep(1)
    except (ProcessLookupError, ValueError, PermissionError, FileNotFoundError):
        pass
    try:
        os.remove(PID_FILE)
    except OSError:
        pass


def make_server_socket(port: int) -> socket.socket:
    """Create a TCP socket with SO_REUSEADDR to handle TIME_WAIT."""
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    sock.bind(("0.0.0.0", port))
    sock.listen(1024)
    return sock


if __name__ == "__main__":
    port = settings.port

    kill_old_server()

    # Create socket with SO_REUSEADDR (handles TIME_WAIT from killed processes)
    for attempt in range(10):
        try:
            sock = make_server_socket(port)
            print(f"  Listening on port {port}")
            break
        except OSError:
            if attempt < 9:
                time.sleep(0.5)
            else:
                print(f"  Failed to bind port {port}")
                exit(1)

    with open(PID_FILE, "w") as f:
        f.write(str(os.getpid()))

    config = uvicorn.Config("app.main:app", host=settings.host, port=port)
    config.log_config = None  # let app/logging_config.py drive output
    server = uvicorn.Server(config)

    _log = logging.getLogger("uvicorn.error")
    _display_host = "localhost" if settings.host in ("0.0.0.0", "::") else settings.host
    _log.info(f"Web Media Scraper running on http://{_display_host}:{settings.port}")
    _log.info("Press Ctrl+C to stop")

    try:
        server.run(sockets=[sock])
    except KeyboardInterrupt:
        _log.info("Shutting down...")
        server.should_exit = True
    finally:
        sock.close()
        try:
            os.remove(PID_FILE)
        except OSError:
            pass