kneezle's newuseralertv3.py

1 files changed, 157 insertions, 0 deletions

diff --git a/bin/newuseralertv3.py b/bin/newuseralertv3.py
new file mode 100755
index 0000000..b62a4bc
--- /dev/null
+++ b/bin/newuseralertv3.py
@@ -0,0 +1,157 @@
+#!/usr/local/bin/python3
+import os
+import argparse
+import http.client
+import sys
+import time
+
+aboutme = """
+# New User Monitor
+# Created for tilde.institute and Ahriman by Kneezle
+# BSD Rocks!
+
+# Version 3.0
+# This version supports hiding the gotify keys by passing them in via the CLI
+# or using the gotifynotify ENV variable to hide from a PS lookup
+"""
+
+defaulttitle = "Alert: New Users"
+
+defaultnewuserpath = "/admin/newusers.dat"
+
+runcmdtemplate = """curl -X POST "https://gotify.tildeverse.org/message?token={}" -F "title=Alert: New Users" -F "message={}" -F "priority=5" """
+
+ap = argparse.ArgumentParser()
+ap.add_argument('--newuserpath', default=defaultnewuserpath,
+                    help='sets the path for the new user txt file')
+
+ap.add_argument('--title', default=defaulttitle,
+                    help='sets the title')
+
+ap.add_argument('--usecurl', default=False,
+                    help='use curl or build in sockets')
+
+ap.add_argument("-a", "--about", dest='about', action='store_true', help="about me")
+
+ap.add_argument("--daemonize", dest='daemonize', action='store_true', help="run in a loop vs only run once")
+
+ap.add_argument('--gotifykeys', nargs='+', default=[],help="the gotify keys of the people to send to")
+
+ap.add_argument('--gotifykeypath', default=None,
+                    help='Load CSV gotify keys from file')
+
+ap.add_argument('--loopinterval', default=60,
+                    help='sets loop interval (seconds) in daemon mode')
+
+args = vars(ap.parse_args())
+
+if args['about']:
+    print(aboutme)
+    exit()
+
+
+title = args['title']
+newuserpath = args['newuserpath']
+usecurl = args['usecurl']
+gotifytokens = args['gotifykeys']
+daemonize = args['daemonize']
+gotifykeypath = args['gotifykeypath']
+loopinterval = int(args['loopinterval'])
+
+
+if gotifykeypath is not None:
+    with open(gotifykeypath) as f:
+        first_line = f.readline()
+        gotifytokens = first_line.rstrip("\n").split(',')
+
+
+if len(gotifytokens) < 1:
+    #we didn't get passed a token. lets try to get from ENV before giving up.
+    try:
+        envtokens = os.environ["gotifynotify"]
+    except KeyError:
+        print("You didn't provide a gotify token!")
+        sys.exit(1)
+    gotifytokens = envtokens.split(',')
+    if len(gotifytokens) < 1:
+        print("You didn't provide a gotify token!")
+        sys.exit(1)
+
+
+boilerplate = """New pending user found!
+
+Username: {}
+Email: {}
+
+"""
+
+#we have no do while in python so this will have to do
+firstloop = True
+
+previouslinecount = 0
+
+while firstloop or daemonize:
+    if not firstloop:
+        time.sleep(loopinterval) #so we don't hammer the CPU in daemon mode
+
+    with open(newuserpath) as fp:
+        line = fp.readline()
+        linecount = 1
+        numnonblanklines = 0
+        push_notification = ""
+        while line:
+            if line.strip() != "": #not a blank line
+                numnonblanklines = numnonblanklines + 1
+                print("New User Found: {} : {}".format(linecount, line.strip()))
+
+                #lets extract the useful information
+                x = line.split(" ")
+                sshkey = ""
+                itemno = 0
+                for item in x:
+                    if itemno > 1:
+                        sshkey = sshkey + " " + item
+                    itemno = itemno + 1
+                sshkey = sshkey.replace('"', "")
+
+                #Lets build the single push notification with all the details
+
+                push_notification = push_notification + boilerplate.format(x[0], x[1])
+            linecount = linecount + 1
+            line = fp.readline()
+
+    #notify is when handled
+    if previouslinecount > numnonblanklines:
+        for key in gotifytokens:
+            if not usecurl:
+                conn = http.client.HTTPSConnection("gotify.tildeverse.org")
+                payload = """title={}&message={}&priority=5""".format("New users handled", "Someone on the admin team got it.")
+                headers = {'content-type': "application/x-www-form-urlencoded"}
+                conn.request("POST", "/message?token={}".format(key), payload, headers)
+                res = conn.getresponse()
+                data = res.read()
+                print(data.decode("utf-8"))
+            else:
+                torun = runcmdtemplate.format(key, push_notification)
+                os.system(torun)
+
+    if numnonblanklines > 0 and numnonblanklines > previouslinecount:
+        if numnonblanklines > 4:
+            push_notification = "You have " + str(numnonblanklines) + " pending new users!"
+        for key in gotifytokens:
+            if not usecurl:
+                conn = http.client.HTTPSConnection("gotify.tildeverse.org")
+                payload = """title={}&message={}&priority=5""".format(title, push_notification)
+                headers = {'content-type': "application/x-www-form-urlencoded"}
+                conn.request("POST", "/message?token={}".format(key), payload, headers)
+                res = conn.getresponse()
+                data = res.read()
+                print(data.decode("utf-8"))
+            else:
+                torun = runcmdtemplate.format(key, push_notification)
+                os.system(torun)
+    else:
+        print("No new users pending")
+    previouslinecount = numnonblanklines
+    firstloop = False #no longer the first loop
+    print("prev: " + str(previouslinecount))