MakerAdmin-Tools/get_lab_enddate.py at master · makerspace/MakerAdmin-Tools · 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
68
69
70
71
72
73
74
75
76
#!/usr/bin/python3
import requests;
import argparse
import csv

# Read the .env file
env = {s[0]: (s[1] if len(s) > 1 else "") for s in (s.split("=") for s in open(".env").read().split('\n'))}

class APIGateway:
    def __init__(self, host, key):
        self.host = APIGateway._ensure_protocol(host)
        self.key = key
        print(self.host)

    def get(self, path, payload=None):
        headers = {"Authorization": "Bearer " + self.key}
        return requests.get(self.host + "/" + path, params=payload, headers=headers)

    def _ensure_protocol(host):
        if not host.startswith("http://") and not host.startswith("https://"):
            host = "https://" +  host
        return host


gateway = APIGateway(env["API_HOST"], env["API_BEARER"])

page = 1
abort = False

member_data_fields = ['member_number', 'firstname', 'lastname']
all_members = {}
while True:
    r = gateway.get("membership/member?sort_order=asc&sort_by=member_id&per_page=1000&page=" + str(page))
    assert r.ok, r.text

    members_resp = r.json()
    members = members_resp["data"]
    page_count = members_resp["last_page"]
    print("Page "+str(page)+" contains "+str(len(members))+" members")
    for member in members:
        current_member = {field: member[field] for field in member_data_fields}
        all_members[str(member['member_number'])] = current_member
    page += 1;
    if page > page_count or abort:
        break

page = 1
active_member_data = [];

while True:
    r = gateway.get("keys?sort_order=asc&sort_by=key_id&per_page=1000&page=" + str(page))
    assert r.ok, r.text

    key_resp = r.json()
    keys = key_resp["data"]
    page_count = key_resp["last_page"]
    print("Page "+str(page)+" contains "+str(len(keys))+" keys")
    for key in keys:
        member_data = all_members[key["title"]]
        active_member = {field: member_data[field] for field in member_data_fields}
        active_member['enddate'] = key['enddate']

        active_member_data.append(active_member)
    page += 1;
    if page > page_count or abort:
        break

if not abort:
    export_data_fields = ['member_number', 'firstname', 'lastname', 'enddate']
    with open('member_lab_enddate.csv', 'w') as csvfile:
        writer = csv.DictWriter(csvfile, fieldnames=export_data_fields)
        writer.writeheader()
        for member in active_member_data:
            writer.writerow({field: member[field] for field in export_data_fields})

print("Done")