[CSV] parsing csv with DictReader instead

This commit is contained in:
coja
2026-05-02 04:00:17 +02:00
parent 749574c8dc
commit 02c2dcc9fc

21
prep.py
View File

@@ -20,20 +20,16 @@ TYPES_DICT = {
def load_events(csv_path:str) -> list[dict]: def load_events(csv_path:str) -> list[dict]:
events = [] events = []
with open(csv_path) as csv_file: with open(csv_path, encoding='utf-8') as csv_file:
csv_reader = csv.reader(csv_file, skipinitialspace=True) csv_reader = csv.DictReader(csv_file, skipinitialspace=True)
next(csv_reader, None)
for event in csv_reader: for event in csv_reader:
event_date = event[0] event_date = event["datum"]
event_date_parsed = datetime.strptime(event_date, "%d-%m-%Y").date() event_date_parsed = datetime.strptime(event_date, "%d-%m-%Y").date()
event_time = event[1] event_time = event["vreme"]
event_location = event[2] event_location = event["lokacija"]
event_title = event[3] event_title = event["tema"]
types = event[4].split() types = event["tip"].split()
try: link = event.get("link", "")
link = event[5]
except IndexError:
link = ""
current_event = {"date":event_date_parsed, current_event = {"date":event_date_parsed,
"time":event_time, "time":event_time,
"location": event_location, "location": event_location,
@@ -43,6 +39,7 @@ def load_events(csv_path:str) -> list[dict]:
events.append(current_event) events.append(current_event)
return events return events
def build_html(events: list[dict], dayNames: list[str], typesNames: dict) -> str: def build_html(events: list[dict], dayNames: list[str], typesNames: dict) -> str:
events_html = [] events_html = []
for event in events: for event in events: