progress count

userdb/db.py

@@ -18,9 +18,12 @@ 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'))
+        cat = str(request.args.get('cat'))
         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())
@@ -35,7 +38,7 @@ def track():
         visits = list(set(visits))
         db.update({
             hash : { 
-                'visits' : visits,
+                cat : visits,
                 'agent' : ua
             }
         })
@@ -45,7 +48,7 @@ def track():
         with open('db.json', 'w+') as dbw:
             dbw.write(json.dumps(db, indent=4))
 
-        return str(len(db[hash]))
+        return str(len(db[hash][cat]))
     else:
         return 'hash missing'