mirror of
https://github.com/koodiklinikka/palkkakysely.git
synced 2026-02-09 07:50:42 +00:00
66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
import datetime
|
|
import glob
|
|
import io
|
|
import os
|
|
|
|
import jinja2
|
|
import numpy
|
|
import pandas
|
|
|
|
from pulkka.config import OUT_DIR, YEAR
|
|
from pulkka.data_ingest import read_data
|
|
|
|
TEMPLATE_DIR = os.path.realpath("./template")
|
|
|
|
|
|
def write_massaged_files(env, df):
|
|
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"))
|
|
df.to_csv(OUT_DIR / "data.csv", index=False)
|
|
df.to_excel(OUT_DIR / "data.xlsx", index=False)
|
|
df.to_json(
|
|
OUT_DIR / "data.json",
|
|
orient="records",
|
|
date_format="iso",
|
|
force_ascii=False,
|
|
)
|
|
print("Wrote data files")
|
|
|
|
|
|
def render_statics(env):
|
|
for filename in glob.glob(os.path.join(TEMPLATE_DIR, "*.html")):
|
|
basename = os.path.relpath(filename, TEMPLATE_DIR)
|
|
if basename.startswith("_"):
|
|
continue
|
|
out_filename = OUT_DIR / basename
|
|
with open(filename, "r") as inf:
|
|
tpl: jinja2.Template = env.from_string(inf.read())
|
|
content = tpl.render(body_class="static-body")
|
|
with open(out_filename, "w") as outf:
|
|
outf.write(content)
|
|
print(filename, "=>", out_filename)
|
|
|
|
|
|
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,
|
|
})
|
|
render_statics(env)
|
|
write_massaged_files(env, df)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|