attendance.py

import argparse
import asyncio
import csv
import os
from datetime import date

from config import password, username
from spond import spond

parser = argparse.ArgumentParser(
    description="Creates an attendance.csv for organizers of events."
)
parser.add_argument(
    "-f",
    "--from",
    help="First date to query for. Date is included in results (format YYYY-MM-DD)",
    type=date.fromisoformat,
    dest="f",
)
parser.add_argument(
    "-t",
    "--to",
    help="Last date to query for. Date is excluded from results (format YYYY-MM-DD)",
    type=date.fromisoformat,
    dest="t",
)
parser.add_argument(
    "-a", help="Also include all members", default=False, action="store_true"
)
args = parser.parse_args()


async def main():
    s = spond.Spond(username=username, password=password)
    events = await s.get_events(args.f, args.t)

    if not os.path.exists("./exports"):
        os.makedirs("./exports")

    for e in events:
        filename = os.path.join(
            "./exports", f"{e['startTimestamp']}-{e['heading']}.csv"
        )
        with open(filename, "w", newline="") as csvfile:
            spamwriter = csv.writer(
                csvfile, delimiter=",", quotechar='"', quoting=csv.QUOTE_MINIMAL
            )

            spamwriter.writerow(
                ["Start", "End", "Description", "Name", "Answer", "Organizer"]
            )
            for o in e["owners"]:
                person = await s.get_person(o["id"])
                full_name = person["firstName"] + " " + person["lastName"]
                spamwriter.writerow(
                    [
                        e["startTimestamp"],
                        e["endTimestamp"],
                        e["heading"],
                        full_name,
                        o["response"],
                        "X",
                    ]
                )
            if args.a is True:
                for r in e["responses"]["acceptedIds"]:
                    person = await s.get_person(r)
                    full_name = person["firstName"] + " " + person["lastName"]
                    spamwriter.writerow(
                        [
                            e["startTimestamp"],
                            e["endTimestamp"],
                            e["heading"],
                            full_name,
                            "accepted",
                        ]
                    )
                for r in e["responses"]["declinedIds"]:
                    person = await s.get_person(r)
                    full_name = person["firstName"] + " " + person["lastName"]
                    spamwriter.writerow(
                        [
                            e["startTimestamp"],
                            e["endTimestamp"],
                            e["heading"],
                            full_name,
                            "declined",
                        ]
                    )
                for r in e["responses"]["unansweredIds"]:
                    person = await s.get_person(r)
                    full_name = person["firstName"] + " " + person["lastName"]
                    spamwriter.writerow(
                        [
                            e["startTimestamp"],
                            e["endTimestamp"],
                            e["heading"],
                            full_name,
                            "unanswered",
                        ]
                    )
                for r in e["responses"]["unconfirmedIds"]:
                    person = await s.get_person(r)
                    full_name = person["firstName"] + " " + person["lastName"]
                    spamwriter.writerow(
                        [
                            e["startTimestamp"],
                            e["endTimestamp"],
                            e["heading"],
                            full_name,
                            "unconfirmed",
                        ]
                    )
                for r in e["responses"]["waitinglistIds"]:
                    person = await s.get_person(r)
                    full_name = person["firstName"] + " " + person["lastName"]
                    spamwriter.writerow(
                        [
                            e["startTimestamp"],
                            e["endTimestamp"],
                            e["heading"],
                            full_name,
                            "waitinglist",
                        ]
                    )

    await s.clientsession.close()


loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
asyncio.run(main())