[CSV] parsing csv with DictReader instead
This commit is contained in:
21
prep.py
21
prep.py
@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user