python_pinger.git

1) #!/usr/bin/python3
2) #
3) # Since this is meant to be run as a daemon, do not use #!/usr/bin/env python3,
4) # because the process name would then be python3, not based on __FILE__.
5) #
6) # [filename] -d 3 192.168.1.1 192.168.1.12 &
7) # disown [jobid]
8) import os
9) import sys
10) import subprocess
11) import time
12) from argparse import ArgumentParser
13) import threading
14) import queue

15) import signal

16) from typing import Dict, Optional

17) 

18) pinger_lock = threading.Lock()
19) 

20) 

21) def log_exit(sig: int, frame) -> None:

22)     _print_queue.put([f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} '
23)                       f'PID={os.getpid()} signal={signal.Signals(sig).name} Exiting.'])

24)     _print_queue.join()
25)     sys.exit(0)
26) 
27) 

28) def pinger(host: str) -> None:

29)     """Executes one ping, prints if there was a change, exits thread."""
30)     now = time.localtime()
31)     success = ping(host)

32)     with pinger_lock:
33)         # Access to _results needs to be serialized
34)         need_update = success != _results[host]
35)         if need_update:
36)             _results[host] = success
37)     if need_update:

38)         status = "UP" if success else "DOWN"
39)         _print_queue.put([f'{time.strftime("%Y-%m-%d %H:%M:%S", now)} {host} {status}'])
40) 
41) 

42) def ping(host: str) -> int:

43)     """Returns True if host (str) responds to a ping request."""
44)     return subprocess.call([*_ping_cmd, host], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) == 0
45) 
46) 

47) def print_manager() -> None:

48)     """ The only thread allowed to write output. """
49)     while True:
50)         job = _print_queue.get()
51)         for line in job:
52)             log(line)
53)         _print_queue.task_done()
54) 
55) 

56) def log(*args, **kwargs) -> None:

57)     """Opens the logfile, writes the log, and closes the logfile."""
58)     # I don't leave the log file open for writing because another process needs access too.
59)     with open(_logname, 'a') as fp:

60)         print(*args, **kwargs, file=fp)

61) 
62) 
63) if __name__ == '__main__':
64)     parser = ArgumentParser(description="Repeatedly ping sites. Ex: %(prog)s -d 3 192.168.1.1 192.168.1.12")

65)     parser.add_argument('-d', '--delay', type=float, default=3.0, help='Delay between pings')

66)     parser.add_argument('addresses', nargs='+', help='Addresses to ping')
67)     parser_args = parser.parse_args()
68) 

69)     _results: Dict[str, Optional[int]] = dict()

70)     for addr in parser_args.addresses:
71)         _results[addr] = None
72)     delay = parser_args.delay
73)     _logname = os.path.join(os.path.expanduser('~'), 'log', os.path.basename(sys.argv[0]).replace('py', 'txt'))
74) 
75)     # Choose the ping parameters appropriate for the platform
76)     if sys.platform == 'cygwin' or sys.platform == 'win32':
77)         _ping_cmd = ('ping', '-n', '1', '-w', '2000')
78)     else:
79)         _ping_cmd = ('ping', '-c', '1', '-W', '2', '-q')
80) 
81)     _print_queue = queue.Queue()
82)     t = threading.Thread(target=print_manager, daemon=True)
83)     t.start()
84)     del t
85) 

86)     signal.signal(signal.SIGINT, log_exit)
87)     signal.signal(signal.SIGTERM, log_exit)
88) 

89)     _print_queue.put([f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}'

90)         f' PID={os.getpid()} repeat={delay}s addresses={",".join(parser_args.addresses)} Starting'])