MZScripts/getBulkAssociateDetails.py at master · MarceloCFerraz/MZScripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
from utils import files, utils, associates


def main():
    """
    Retrieves and displays associate details.

    This function prompts the user to select the environment and organization ID. It then prompts the user to select the key type index. Next, it reads a file containing search keys and retrieves associate details based on the selected key type index. The function prints the details of each associate, including their state, account type, associate ID, contact information, location ID, hub ID, fleet ID (if available), and WorldView ID. If there are more than 10 associates, a message is displayed indicating that there are too many associates, and the full response is saved in the 'RESULTS' folder.

    Parameters:
    - None

    Returns:
    - None
    """
    env = utils.select_env()
    orgId = utils.select_org(env)

    key_type_index = utils.get_associate_key_type_index()

    search_keys = files.get_data_from_file("associates")
    associatesArray = []

    allOrgAssociates = list(associates.search_associate(env, orgId, 0, orgId))

    for search_key in search_keys:
        if key_type_index == 0:
            search_key = orgId

        search_results = [
            a
            for a in allOrgAssociates
            if a[utils.ASSOCIATE_KEY_TYPES[key_type_index]] == search_key
        ]
        for associate in search_results:
            associatesArray.append(associate)

    associates_count = len(associatesArray)
    print(f"Found {associates_count} associates")

    if associates is not None:
        if associates_count <= 10:
            for associate in associatesArray:
                print(f"Associate State: {associate['state']}")
                print(f"Account Type: {associate['associateType']}")
                print(f"Associate ID: {associate['associateId']}")
                print(f"   User Name: {associate['contact']['userName']}")
                print(f"        Name: {associate['contact']['name']}")
                print(f"      E-Mail: {associate['contact']['email']}")
                print(f"       Phone: {associate['contact']['phone']}")
                print(f"Location ID: {associate['location']['locationId']}")
                print(f"HUB ID: {associate['hubId']}")
                try:
                    print(f"Fleet ID: {associate['fleetId']}")
                except Exception:
                    pass
                print(f"WorldView ID: {associate['worldViewId']}\n")
        else:
            print(
                "Too many associates, check 'RESULTS' folder to see the full response!"
            )
            files.save_json_to_file(
                filePrefix="ASSOCIATES", jsonData=files.format_json(associatesArray)
            )


main()