diff --git a/pulkka/massage_outputs.py b/pulkka/massage_outputs.py
index 37d17cc..ff247b7 100644
--- a/pulkka/massage_outputs.py
+++ b/pulkka/massage_outputs.py
@@ -1,3 +1,4 @@
+import base64
import datetime
import glob
import io
@@ -14,11 +15,15 @@ TEMPLATE_DIR = os.path.realpath("./template")
def write_massaged_files(env, df):
- with open(OUT_DIR / 'data.html', 'w') as f:
+ with open(OUT_DIR / "data.html", "w") as f:
with io.StringIO() as s:
df.to_html(s, index=False, na_rep="", border=0)
table_html = s.getvalue()
- f.write(env.get_template('_table.html').render(table_html=table_html, body_class="table-body"))
+ f.write(
+ env.get_template("_table.html").render(
+ table_html=table_html, body_class="table-body"
+ )
+ )
df.to_csv(OUT_DIR / "data.csv", index=False)
df.to_excel(OUT_DIR / "data.xlsx", index=False)
df.to_json(
@@ -44,19 +49,32 @@ def render_statics(env):
print(filename, "=>", out_filename)
+def read_asset_to_data_uri(filename, content_type):
+ with open(filename, "rb") as f:
+ return (
+ f"data:{content_type};base64,{base64.b64encode(f.read()).decode('ascii')}"
+ )
+
+
def main():
df = read_data()
env = jinja2.Environment(
autoescape=True,
loader=jinja2.FileSystemLoader(TEMPLATE_DIR),
)
- env.globals.update({
- "date": datetime.datetime.utcnow(),
- "pd": pandas,
- "np": numpy,
- "df": df,
- "year": YEAR,
- })
+ env.globals.update(
+ {
+ "date": datetime.datetime.utcnow(),
+ "pd": pandas,
+ "np": numpy,
+ "df": df,
+ "year": YEAR,
+ "logo_svg": read_asset_to_data_uri(
+ os.path.join(TEMPLATE_DIR, "logo.svg"), "image/svg+xml"
+ ),
+ "site_url": f"https://koodiklinikka.github.io/palkkakysely/{YEAR}/",
+ }
+ )
render_statics(env)
write_massaged_files(env, df)
diff --git a/template/index.html b/template/index.html
index c1fb5e6..2af91cf 100644
--- a/template/index.html
+++ b/template/index.html
@@ -1,6 +1,10 @@
{% extends "_base.html" %}
-{% block title %}Koodiklinikan palkkakysely, {{ year }}{% endblock %}
{% block body %}
+
Koodiklinikan palkkakysely, {{ year }}
Tunnusluvut
+ Viittaathan dataa käyttäessäsi tähän sivustoon: {{ site_url }}
{% endblock %}
{% block footer %}