Working on Pulling info about client

utils/clientfunctions.py

@@ -0,0 +1,60 @@
+import json
+import requests
+import os
+import pandas as pd
+import utils.pretty as ct
+
+
+def getClientsPolicy(url, clientid="0dcddaf3-d017-4ced-9227-ceeca3117286"):
+
+    endpoint = url + '/v1/agent/find'
+    payload = {
+            "agentid" : f"{clientid}"
+    }
+
+    headers = {"X-APIKey": os.getenv('APIKEY')}
+    payload = json.dumps(payload)
+
+
+    response = requests.post(endpoint, headers=headers, data=payload, verify=False)
+    result = json.loads(response.text)
+    data = pd.DataFrame(result["response"]["agents"])
+
+    policyname = getPolicyName(url,data.loc[0, "groupid"])
+    allowlists = getPolicyAllowlists(url,data.loc[0, "groupid"]) 
+    allowlists = allowlists['name'].tolist()
+    print(policyname)
+    print(allowlists)      
+    
+
+
+def getPolicyAllowlists(url, groupid):
+
+    endpoint = url + '/v1/group/policies'
+    payload = {
+            "groupid" : f"{groupid}"
+    }
+
+    headers = {"X-APIKey": os.getenv('APIKEY')}
+    payload = json.dumps(payload)
+
+    response = requests.post(endpoint, headers=headers, data=payload, verify=False)
+    result = json.loads(response.text)
+    data = pd.DataFrame(result["response"]["applications"])
+    return data
+
+def getPolicyName(url, groupid):
+
+    endpoint = url + '/v1/group/'
+    payload = {
+            
+    }
+
+    headers = {"X-APIKey": os.getenv('APIKEY')}
+    payload = json.dumps(payload)
+
+    response = requests.post(endpoint, headers=headers, data=payload, verify=False)
+    result = json.loads(response.text)
+    data = pd.DataFrame(result["response"]["groups"])
+    name = data.loc[data['groupid'] == f"{groupid}", 'name'].values[0]
+    return name