27 lines
810 B
Python
27 lines
810 B
Python
import time
|
|
import logging
|
|
import signal
|
|
import sys
|
|
from gpu_checker import check_rtx_50_founders
|
|
from env_config import REFRESH_TIME
|
|
|
|
# Signal handler function
|
|
def handle_exit(signum, frame):
|
|
logging.info(f"🛑 Received signal {signum}. Exiting gracefully...")
|
|
sys.exit(0)
|
|
|
|
# Register signal handlers
|
|
signal.signal(signal.SIGINT, handle_exit) # Ctrl+C
|
|
signal.signal(signal.SIGTERM, handle_exit) # docker stop / kill -15
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
while True:
|
|
start = time.time()
|
|
check_rtx_50_founders()
|
|
elapsed = time.time() - start
|
|
time.sleep(max(0, REFRESH_TIME - elapsed))
|
|
except KeyboardInterrupt:
|
|
logging.info("🛑 Script interrupted by user (KeyboardInterrupt). Exiting gracefully.")
|
|
sys.exit(0)
|
|
|