diff --git a/sshalert.py b/sshalert.py index 45d677d..4cf6d66 100755 --- a/sshalert.py +++ b/sshalert.py @@ -15,7 +15,8 @@ nexmo_key = os.getenv("NEXMO_KEY") nexmo_secret = os.getenv("NEXMO_SECRET") except: - logging.critical("ERROR: Have you exported all required environment variables? (TARGET_PHONE_NUMBER, NEXMO_KEY, NEXMO_SECRET)") + logging.critical( + "ERROR: Have you exported all required environment variables? (TARGET_PHONE_NUMBER, NEXMO_KEY, NEXMO_SECRET)") exit(1) # Initialize the nexmo client @@ -26,7 +27,8 @@ def poll_logfile(filename): """ Polls a logfile for sudo commands or ssh logins. """ - f = subprocess.Popen(["tail", "-F", "-n", "0", filename], encoding="utf8", stdout=subprocess.PIPE, stderr=subprocess.PIPE) + f = subprocess.Popen(["tail", "-F", "-n", "0", filename], encoding="utf8", stdout=subprocess.PIPE, + stderr=subprocess.PIPE) p = select.poll() p.register(f.stdout) @@ -41,11 +43,11 @@ def process_log_entry(logline): Check a logline and see if it matches the content we care about. """ # If it's a local sudo exec - if all(x in logline for x in ["sudo", "COMMAND"]: + if all(x in logline for x in ["sudo", "COMMAND"]): send_sms(logline) - + # If it's an SSH login - elif all(x in logline for x in ["ssh", "Accepted"]: + elif all(x in logline for x in ["ssh", "Accepted"]): send_sms(logline) return