delete default users

diff --git a/camunda-overlay/camunda.py b/camunda-overlay/camunda.py
index 65ad271d3e74ac9e130f4bea08093ecd54ba8305..514b508cc90086517e714de865919315065fec92 100755 (executable)
--- a/camunda-overlay/camunda.py
+++ b/camunda-overlay/camunda.py
@@ -34,6 +34,26 @@ def cleanup_old_deployments(key='sentiment-analysis'):
             except:
                 pprint(res.content)
 
+def get_users():
+    res = requests.get(CAMUNDA + "user")
+    return res.json()
+
+def cleanup_users(keep = ["demo"]):
+    for user in get_users():
+        # keep user "demo"
+        if user['id'] in keep:
+            continue
+
+        res = requests.delete(CAMUNDA + "user/" + user['id'])
+        if (res.status_code == 204):
+            print ("Cleaned up user {}".format(user['id']))
+        else:
+            print ("Error cleaning user {}: Code: {}".format(user['id'], res.status_code))
+            try:
+                pprint(res.json())
+            except:
+                pprint(res.content)
+
 def create_deployment(cleanup=False):
     parameters = [
             ("deployment-name", "sentiment-analysis"),
@@ -106,6 +126,8 @@ if __name__ == "__main__":
     if args.deploy:
         # initialize camunda process
         create_deployment(cleanup=args.cleanup)
+        if args.cleanup:
+            cleanup_users()
 
     if args.autoclick >= 1:
         # start clicking