import csv import os script_dir = os.path.dirname(__file__) def parsecsv(): libcsv = open(os.path.join(script_dir, "varlib.csv"), "r") with libcsv: csv_as_dict = csv.DictReader(libcsv) return csv_as_dict def getpublications(): libcsv = open(os.path.join(script_dir, "varlib.csv"), "r") with libcsv: csv_as_dict = csv.DictReader(libcsv) publications = {} for row in csv_as_dict: pubinfo = { "Title": row["Publication"], "Author": row["Author"], "Type": row["Type"], } publications[row["Id"]] = pubinfo return publications def gettypes(): libcsv = open(os.path.join(script_dir, "varlib.csv"), "r") with libcsv: csv_as_dict = csv.DictReader(libcsv) listoftypes = [] for row in csv_as_dict: lowertype = row["Type"].lower() if lowertype not in listoftypes: listoftypes.append(lowertype) return listoftypes def getyears(): libcsv = open(os.path.join(script_dir, "varlib.csv"), "r") with libcsv: csv_as_dict = csv.DictReader(libcsv) listofyears = [] for row in csv_as_dict: if row["Year"] not in listofyears: listofyears.append(row["Year"]) return listofyears # test = getpublications() # for ids, pubinfo in test.items(): # print(pubinfo["Title"])