initial

src/flask/beer_data.py

@@ -0,0 +1,116 @@
+from bokeh.core.enums import SpatialUnitsType
+import mongo.build_db as pool_database
+import mongo.query_db as pool_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.validators import DataRequired, Length, Optional
+from waitress import serve
+from bokeh.models.layouts import HBox
+from bokeh.plotting import column
+from charts import BeerCharts
+from bokeh.io import output_file, show
+from bokeh.layouts import row
+
+app = Flask(__name__)
+app.secret_key = 'testsecret' #this value will change
+
+bootstrap = Bootstrap5(app)
+
+csrf = CSRFProtect(app)
+# used to configure the bokeh plot for graphs
+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)))
+
+
+class userForm(FlaskForm):
+    username = StringField("User Name?", validators=[DataRequired()])
+    password = PasswordField("Password?")
+    submit = SubmitField("Letsa GO!")
+
+class dataForm(FlaskForm):
+    beer_run_id = IntegerField("Beer Run ID")
+    beer_run_type = TextAreaField("Beer Type")
+    Date = DateField("Date:")
+    sg = IntegerField("SG Reading")
+    comment = TextAreaField("Any Comments?", validators=[Optional()])
+    submit = SubmitField("Write it, Write it REAAAAAAL GOOOD")
+
+@app.route("/", methods=["GET","POST"])
+def index():
+    form = userForm()
+    if form.validate_on_submit():
+        username = form.username.data
+        password = form.password.data
+        db = pool_database_query.pool_query()
+        if db.user_check(username, password):
+            session['logged_in'] = True
+            return redirect("/updater")
+        else:
+            return render_template("index.html", try_again=True, form=form)
+    else:
+        return render_template("index.html", try_again=False, form=form)
+
+@app.route("/updater", methods=["GET", "POST"])
+def updater():
+    if 'logged_in' not in session:
+        return redirect("/")
+    create_graphs()
+    query_db = pool_database_query.pool_query()
+    query = query_db.get_top(10, "sg")
+    form = dataForm()
+
+    if form.validate_on_submit():
+        database = pool_database.pool_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}',
+                "comment": f'{form.comment.data}'
+            }
+        if database.record_exists(new_record["date"], new_record["beer_run_id"]):
+            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"])
+
+        return render_template("updater.html", list=query, form=form, success=True, updater_name = new_record["test_user"])
+    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("/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])
+
+if __name__ == '__main__':
+    #app.run(host='0.0.0.0')
+    serve(app, host='0.0.0.0', port=5000, url_scheme='https')