Closes issue #23

This commit is contained in:
Thomas Williams 2024-07-07 18:17:02 +01:00
parent 48abfbba8c
commit 7086eeac1b
Signed by: thomas
GPG key ID: EB8F975CF60BCBFF

10
main.py
View file

@ -31,6 +31,7 @@ import requests
import threading
import signal
import socket
import sys
from functools import partial
from concurrent.futures import ThreadPoolExecutor, as_completed
from bs4 import BeautifulSoup
@ -199,6 +200,15 @@ def logURLLog(hostname, logTime, url, responseTime):
def main():
if sys.platform.startswith('win'):
import win32api
win32api.SetConsoleCtrlHandler(lambda sig, frame:
signal_handler(signal.CTRL_C_EVENT, frame), True)
else:
signal.signal(signal.SIGTERM, signal_handler)
signal.signal(signal.SIGINT, signal_handler)