Exit gracefully when receiving SIGTERM.

Author: xybu

README.md

@@ -94,7 +94,7 @@ optional arguments:
                         "psstat.csv".
 ```
 
-To terminate the monitor, send `SIGINT` signal to the process.
+To terminate the monitor, send `SIGINT` or `SIGTERM` signal to the process.
 
 # Example
 

resmon/resmon.py

@@ -18,8 +18,8 @@
 import argparse
 import os
 import sched
+import signal
 import sys
-import threading
 import time
 import psutil
 
@@ -246,6 +246,10 @@ def chprio(prio):
         print('Warning: failed to elevate priority!', file=sys.stderr)
 
 
+def sigterm(signum, frame):
+    raise KeyboardInterrupt()
+
+
 def main():
     parser = argparse.ArgumentParser(
         description='Monitor system-wide resource availability. Optionally monitor processes that match the specified criteria and their children.')
@@ -281,6 +285,8 @@ def main():
         # Convert to lowercase to achieve case IN-sensitiveness.
         args.ps_keywords = [k.lower() for k in args.ps_keywords]
 
+    signal.signal(signal.SIGTERM, sigterm)
+
     try:
         chprio(-20)
         scheduler = sched.scheduler(time.time, time.sleep)