cgi shenanigans

userdb/db.py

@@ -0,0 +1,39 @@
+import json
+import os
+from flask import Flask, request
+from functools import reduce
+
+app = Flask(__name__)
+
+@app.route("/track", methods = ['GET'])
+def track():
+    if request.method == 'GET' and request.args.get('hash'):
+        hash = str(request.args.get('hash'))
+        ua = str(request.args.get('ua'))
+        url = str(request.values.get("url") or request.headers.get("Referer"))
+        if os.path.exists('db.json'):
+            with open('db.json', 'r') as db:
+                db = json.loads(db.read())
+        else:
+            db = {}
+        if hash in db.keys():
+            visits = db[hash]['visits']
+        else:
+            visits = list()
+        
+        visits.append(url)
+        visits = list(set(visits))
+        db.update({
+            hash : { 
+                'visits' : visits,
+                'agent' : ua
+            }
+        })
+        with open('db.json', 'w+') as dbw:
+            dbw.write(json.dumps(db, indent=4))
+
+        return str(len(db[hash]))
+    else:
+        return 'hash missing'
+
+app.run()