fix up to get running and some feature add

src/flask/beer_data.py

@@ -1,11 +1,11 @@
 from bokeh.core.enums import SpatialUnitsType
-import mongo.build_db as pool_database
-import mongo.query_db as pool_database_query
+import mongo.build_db as beer_database
+import mongo.query_db as beer_database_query
 
 from flask import Flask, render_template, request, jsonify, redirect, session
 from flask_wtf import FlaskForm, CSRFProtect
 from flask_bootstrap import Bootstrap5
-from wtforms import StringField, SubmitField, DateField, IntegerField, PasswordField, DecimalField, RadioField, TextAreaField
+from wtforms import StringField, SubmitField, DateField, IntegerField, PasswordField, DecimalField, RadioField, TextAreaField, BooleanField
 from wtforms.validators import DataRequired, Length, Optional
 from waitress import serve
 from bokeh.models.layouts import HBox
@@ -26,7 +26,7 @@ output_file("/beer_data/src/flask/static/data_plot.html")
 def create_graphs():
     chart = BeerCharts.BeerCharts()
     sg = chart.line_chart("SG", "sg", 30)
-    show(row(column(sg)))
+    show(column(sg))
 
 
 class userForm(FlaskForm):
@@ -39,6 +39,7 @@ class dataForm(FlaskForm):
     beer_run_type = TextAreaField("Beer Type")
     Date = DateField("Date:")
     sg = IntegerField("SG Reading")
+    final_run = BooleanField("Final Reading?")
     comment = TextAreaField("Any Comments?", validators=[Optional()])
     submit = SubmitField("Write it, Write it REAAAAAAL GOOOD")
 
@@ -48,7 +49,7 @@ def index():
     if form.validate_on_submit():
         username = form.username.data
         password = form.password.data
-        db = pool_database_query.pool_query()
+        db = beer_database_query.pool_query()
         if db.user_check(username, password):
             session['logged_in'] = True
             return redirect("/updater")
@@ -62,17 +63,18 @@ def updater():
     if 'logged_in' not in session:
         return redirect("/")
     create_graphs()
-    query_db = pool_database_query.pool_query()
+    query_db = beer_database_query.pool_query()
     query = query_db.get_top(10, "sg")
     form = dataForm()
 
     if form.validate_on_submit():
-        database = pool_database.pool_data()
+        database = beer_database.beer_data()
         new_record = {
                 "date": f'{form.Date.data}',
                 "beer_run_id": f'{form.beer_run_id.data}',
                 "beer_type": f'{form.beer_run_type.data}',
                 "sg" : f'{form.sg.data}',
+                "final_reading" : F'{form.final_run.data}',
                 "comment": f'{form.comment.data}'
             }
         if database.record_exists(new_record["date"], new_record["beer_run_id"]):
@@ -82,34 +84,38 @@ def updater():
                         if database.existing_record[field] != new_record[field]:
                             database.update_re_record(database.existing_record["_id"], field, new_record[field])
         else:
+            if new_record["final_reading"] == "True":
+                final_run_value = True
+            else:
+                final_run_value = False
             database.create_re_record(new_record["beer_run_id"], new_record["beer_type"], new_record["sg"],
-                                       new_record["date"], new_record["comment"])
+                                       new_record["date"], final_run_value, new_record["comment"])
 
-        return render_template("updater.html", list=query, form=form, success=True, updater_name = new_record["test_user"])
+        return render_template("updater.html", list=query, form=form, success=True, updater_name = "Saucy Beer Maker")
     else:
         return render_template("updater.html", list=query, form=form, sucess=False)
 
-@app.route("/update_db", methods=["POST"])
-def pool_data_update():
-    database = pool_database.pool_data()
-    new_record = request.json
-    if database.record_exists(new_record["date"], new_record["test_user"]):
-        for field in database.existing_record:
-            for new_field in new_record:
-                if field == new_field:
-                    if database.existing_record[field] != new_record[field]:
-                        database.update_re_record(database.existing_record["_id"], field, new_record[field])
-            else:
-                database.create_re_record(new_record["beer_run_id"], new_record["beer_type"], new_record["sg"],
-                                       new_record["date"], new_record["comment"])
+# @app.route("/update_db", methods=["POST"])
+# def beer_data_update():
+#     database = beer_database.beer_data()
+#     new_record = request.json
+#     if database.record_exists(new_record["date"], new_record["test_user"]):
+#         for field in database.existing_record:
+#             for new_field in new_record:
+#                 if field == new_field:
+#                     if database.existing_record[field] != new_record[field]:
+#                         database.update_re_record(database.existing_record["_id"], field, new_record[field])
+#             else:
+#                 database.create_re_record(new_record["beer_run_id"], new_record["beer_type"], new_record["sg"],
+#                                        new_record["date"], new_record["comment"])
 
  
 
-@app.route("/pool_top/<int:return_number>/<string:field>")
-def user_detail(id):
-    query_db = pool_database_query.pool_query()
-    query = query_db.get_top(return_number, field)
-    return jsonify([row.to_json() for row in query])
+# @app.route("/pool_top/<int:return_number>/<string:field>")
+# def user_detail(id):
+#     query_db = beer_database_query.pool_query()
+#     query = query_db.get_top(return_number, field)
+#     return jsonify([row.to_json() for row in query])
 
 if __name__ == '__main__':
     #app.run(host='0.0.0.0')