Finishing pdf export, yet to test it.HEADmaster

1 files changed, 6 insertions, 5 deletions

diff --git a/main.py b/main.py
index 2fdb0f9..05f7564 100644
--- a/main.py
+++ b/main.py
@@ -9,6 +9,7 @@ import dates
 
 from flask import Flask, render_template, redirect, url_for
 from threading import Thread
+import os
 
 app = Flask(__name__)
 
@@ -106,13 +107,13 @@ def export(opt):
         for i in data:
             if data[1] == 1:
                 days.inc(data[0], 1 / STEPS * 100)
-        days.sort_by_value()
         con.close()
+        days.sort_by_value()
         # writing data to a file
-        # with open("static/export.txt", "w") as file:
-        #    file.write(o)
-        # return redirect("/static/export.txt")
-        return True
+        with open("static/table.ms", "w") as file:
+            file.write(days.roff_tbl())
+        os.system("groff -Tpdf -P-pa4 -tk static/table.ms > static/table.pdf")
+        return redirect("/static/table.pdf")
     else:
         return render_template(
             "error.html", error="Invalid export type"