diff --git a/dockup b/dockup index d058a4f..cca027b 100755 --- a/dockup +++ b/dockup @@ -4,15 +4,31 @@ import http.server import json import logging import os +import signal import socketserver import subprocess +import sys import time from threading import Lock, Thread import docker from dotenv import load_dotenv +from fastapi import FastAPI from packaging.version import InvalidVersion, Version, parse +# Enable quick restart in docker +app = FastAPI() + + +def shutdown_handler(signum, frame): + logging.info("Received shutdown signal") + sys.exit(0) + + +signal.signal(signal.SIGTERM, shutdown_handler) +signal.signal(signal.SIGINT, shutdown_handler) + + # Global variables data_lock = Lock() g_data = [] diff --git a/requirements.txt b/requirements.txt index 2ad54b2..3e6181d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ docker packaging dotenv +fastapi