Browse Source

cleaned up code with flake8 and black

master
crunk 3 years ago
parent
commit
230f242af5
  1. 22
      library/csvparser/csvparser.py
  2. 13
      library/page.py

22
library/csvparser/csvparser.py

@ -17,12 +17,15 @@ def getpublications():
csv_as_dict = csv.DictReader(libcsv)
publications = {}
for row in csv_as_dict:
pubinfo = {"Title" : row["Publication"],
"Author" : row["Author"],
"Type" : row["Type"]}
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:
@ -32,9 +35,20 @@ def gettypes():
lowertype = row["Type"].lower()
if lowertype not in listoftypes:
listoftypes.append(lowertype)
#listoftypes = [t.title() for t in listoftypes]
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"])

13
library/page.py

@ -4,7 +4,7 @@
import flask
from flask import render_template
from csvparser.csvparser import getpublications, gettypes
from csvparser.csvparser import getpublications, gettypes, getyears
APP = flask.Flask(__name__, static_folder="static")
@ -14,15 +14,22 @@ def index():
"""Main route"""
# parse csv, render template with a few elements from the csv
pubtypes = gettypes()
pubyears = getyears()
publicatons = getpublications()
return render_template("index.html", publications=publicatons, pubtypes=pubtypes)
template = render_template(
"index.html",
publications=publicatons,
pubtypes=pubtypes,
pubyears=pubyears
)
return template
@APP.route("/<publicationID>")
def show_book(publicationID):
"""route for a publication"""
# parse csv, render template with full list.
return render_template("publication.html", publication=publication)
return render_template("publication.html")
@APP.route("/<publication>")

Loading…
Cancel
Save