150 lines
5.4 KiB
Python
150 lines
5.4 KiB
Python
import requests
|
|
import pprint
|
|
import pandas as pd
|
|
import Apikeys
|
|
import time
|
|
|
|
pp = pprint.PrettyPrinter(indent=4)
|
|
APIKEY = Apikeys.CHUBB
|
|
HEADERS = {
|
|
"accept": "application/json",
|
|
"X-Api-Key": APIKEY,
|
|
}
|
|
BASEURL = "https://api.northpass.com/v2/"
|
|
IMPORTFILE = "/Users/normrasmussen/Downloads/cisa-090825.csv"
|
|
|
|
|
|
def bulk_invite_and_group():
|
|
"""
|
|
Bulk endpoint which invites new people and adds them to a group via this structure:
|
|
{
|
|
"email": "me@mac.com"
|
|
"groups": "GroupA"
|
|
}
|
|
|
|
This function looks for the group in the CSV as well.
|
|
"""
|
|
df = pd.DataFrame()
|
|
data = pd.read_csv(IMPORTFILE)
|
|
groups = data["Group"].unique()
|
|
groups = list(groups)
|
|
print(groups)
|
|
for group in groups:
|
|
payload = ""
|
|
print(group)
|
|
tmp_group = data[data.Group == group]
|
|
people = list(tmp_group["Email"])
|
|
group = str(tmp_group["Group"].unique())[2:-2]
|
|
print(f"Group --> {group} ... Amount of People --> {len(people)}")
|
|
url = f"{BASEURL}bulk/people"
|
|
if len(people) > 1500:
|
|
for chunk in range(0, len(people), 1500):
|
|
i = chunk
|
|
payload_1 = []
|
|
i_to_add = people[i : i + 1500]
|
|
for person in i_to_add:
|
|
miniload = {"email": person, "groups": group}
|
|
payload_1.append(miniload)
|
|
print(f"The long length {group} payload has {len(payload_1)}")
|
|
payload = {"data": {"attributes": {"people": payload_1}}}
|
|
response = requests.post(url, headers=HEADERS, json=payload)
|
|
print(f"Completed. Status code is {response.status_code}")
|
|
else:
|
|
payload_1 = []
|
|
for person in people:
|
|
miniload = {"email": person, "groups": group}
|
|
payload_1.append(miniload)
|
|
print(f"The {group} payload has {len(payload_1)}")
|
|
payload = {"data": {"attributes": {"people": payload_1}}}
|
|
response = requests.post(url, headers=HEADERS, json=payload)
|
|
print(f"Completed. Status code is {response.status_code}")
|
|
print(response.text)
|
|
print("Running add props from func...")
|
|
time.sleep(3)
|
|
add_props_from_func(people, data, group)
|
|
|
|
|
|
def add_props_from_func(people, data, group):
|
|
errorlist = []
|
|
for learner_email in people:
|
|
agency_name = data.loc[data["Email"] == learner_email, "AgencyName"]
|
|
agname = str(agency_name.values)[2:-2]
|
|
print(f"Learner: {learner_email} --> Agency: {agname} from Group: {group}")
|
|
ppl_search = f"{BASEURL}people?filter[email][eq]={learner_email}"
|
|
ppl_response = requests.get(ppl_search, headers=HEADERS)
|
|
try:
|
|
ppl_data = ppl_response.json()
|
|
nextlink = ppl_data["links"]
|
|
learner_uuid = ppl_data["data"][0]["id"]
|
|
|
|
# Now update the props
|
|
prop_url = f"{BASEURL}properties/people/bulk"
|
|
payload = {
|
|
"data": [
|
|
{
|
|
"attributes": {"properties": {"agency_name": agname}},
|
|
"id": learner_uuid,
|
|
"type": "person_properties",
|
|
}
|
|
]
|
|
}
|
|
propresponse = requests.post(prop_url, json=payload, headers=HEADERS)
|
|
print(propresponse.status_code)
|
|
if propresponse.status_code != 200:
|
|
error_tupe = (learner_uuid, learner_email, agname)
|
|
errorlist.append(error_tupe)
|
|
else:
|
|
print(f"Looks like {learner_email} and {agname} was successful.")
|
|
except (TypeError, IndexError) as e:
|
|
error_tupe = (0, learner_email, agency_name)
|
|
errorlist.append(error_tupe)
|
|
print(f"{e} has occurred with {learner_email}")
|
|
finally:
|
|
pass
|
|
print(f"Error list: {errorlist}")
|
|
|
|
|
|
def add_props_from_csv():
|
|
errorlist = []
|
|
df = pd.DataFrame()
|
|
data = pd.read_csv(IMPORTFILE)
|
|
for dat in data.iterrows():
|
|
agency_name = dat[1][3]
|
|
# agency_name = "EMPLOYEE"
|
|
learner_email = dat[1][2]
|
|
# print(learner_email)
|
|
ppl_search = f"{BASEURL}people?filter[email][eq]={learner_email}"
|
|
ppl_response = requests.get(ppl_search, headers=HEADERS)
|
|
try:
|
|
ppl_data = ppl_response.json()
|
|
nextlink = ppl_data["links"]
|
|
learner_uuid = ppl_data["data"][0]["id"]
|
|
|
|
# Now update the props
|
|
prop_url = f"{BASEURL}properties/people/bulk"
|
|
payload = {
|
|
"data": [
|
|
{
|
|
"attributes": {"properties": {"agency_name": agency_name}},
|
|
"id": learner_uuid,
|
|
"type": "person_properties",
|
|
}
|
|
]
|
|
}
|
|
propresponse = requests.post(prop_url, json=payload, headers=HEADERS)
|
|
print(propresponse.status_code)
|
|
if propresponse.status_code != 200:
|
|
error_tupe = (learner_uuid, learner_email, agency_name)
|
|
errorlist.append(error_tupe)
|
|
else:
|
|
print(f"Looks like {learner_email} and {agency_name} was successful.")
|
|
except TypeError as e:
|
|
print(e)
|
|
pass
|
|
finally:
|
|
print(f"Error list: {errorlist}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
bulk_invite_and_group()
|