mildlyfunctionalgays/typstcard
2
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Add type to CSV row

Browse source
This commit is contained in:
Skye 2023-12-07 18:20:56 -05:00
parent f8887eb7ac
commit a8e89ca579
1 changed files with 25 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
format.py
View file

@ -273,6 +273,19 @@ elif args.address_file.suffix == ".ods":
else: else:
raise Exception("Unknown file type for --address-file") raise Exception("Unknown file type for --address-file")
class CSVRow(TypedDict):
Name: str
Address: str
DPC: str
Country: str
Design: str
Type: str
Personalization: str
Avatar: str | None
US: str
csvfile = open(csv_filename) csvfile = open(csv_filename)
rows = csv.DictReader(csvfile) rows = csv.DictReader(csvfile)
@ -286,12 +299,13 @@ mid = secrets.get("mailer_id")
class Card(TypedDict): class Card(TypedDict):
address: str address: str
avatar: Path | None avatar: Path | None
row: dict[str, str] row: CSVRow
imb: str imb: str
cards: list[Card] = [] cards: list[Card] = []
for row in rows: for row in rows:
row = CSVRow(**row)
if row["Address"] == "": if row["Address"] == "":
continue continue
@ -304,18 +318,19 @@ for row in rows:
address = row["Address"].split("\n") + country address = row["Address"].split("\n") + country
if row.get("Avatar", "") != "": if (avatar_url := row.get("Avatar", "")) != "":
avatar = get_avatar(row["Avatar"], secrets) avatar = get_avatar(avatar_url, secrets) # type: ignore
else: else:
avatar = None avatar = None
card: Card = { cards.append(
"address": "\n".join(address), {
"avatar": avatar, "address": "\n".join(address),
"row": row, "avatar": avatar,
"imb": "", "row": row,
} "imb": "",
cards.append(card) }
)
# Typst can't access files outside the project root, except through a symlink # Typst can't access files outside the project root, except through a symlink
# Create one in cache to use here # Create one in cache to use here