Initial commit.

JASchilz · JASchilz · commit f0abe39403a8 · 2018-04-03T20:44:31.000-07:00
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+__pycache__/
diff --git a/main.py b/main.py
@@ -0,0 +1,52 @@
+import os
+import base64
+
+from flask import Flask, render_template, request, redirect, url_for, session
+
+from model import SavedTotal
+
+app = Flask(__name__)
+app.secret_key = b'\xb6x(\xd67\x1f\xa7\x15\x92\xf1VqU\xe9|\xbcqu\xac\xf6\x16\xa8\x8f\xe5'
+
+@app.route('/retrieve')
+def retrieve():
+	code = request.args.get('code', None)
+
+	if code is None:
+		return render_template("retrieve.jinja2") 
+	try:
+		saved_total = SavedTotal.get(SavedTotal.code == code)
+	except SavedTotal.DoesNotExist:
+		return render_template("retrieve.jinja2")
+	
+	session['total'] = saved_total.value
+
+	return redirect(url_for('add'))
+	
+
+@app.route('/add', methods=['GET', 'POST'])
+def add():
+	if 'total' not in session:
+		session['total'] = 0
+	if request.method == 'POST':
+		# Handle the form submission
+		number = int(request.form['number'])
+		session['total'] += number
+
+	return render_template('add.jinja2', session=session)
+
+@app.route('/save', methods=['POST'])
+def save():
+	total = session.get('total', 0)
+	code = base64.b32encode(os.urandom(8)).decode().strip('=')
+
+	saved_total = SavedTotal(value=total, code=code)
+	saved_total.save()
+
+	return render_template('save.jinja2', code=code)
+
+
+if __name__ == "__main__":
+	port = int(os.environ.get("PORT", 6738))
+	app.run(host='0.0.0.0', port=port)
+
diff --git a/model.py b/model.py
@@ -0,0 +1,13 @@
+import os
+
+from peewee import Model, CharField, IntegerField
+from playhouse.db_url import connect
+
+db = connect(os.environ.get('DATABASE_URL', 'sqlite:///my_database.db'))
+
+class SavedTotal(Model):
+	code = CharField(max_length=255, unique=True)
+	value = IntegerField()
+
+	class Meta:
+		database = db
diff --git a/my_database.db b/my_database.db
diff --git a/setup.py b/setup.py
@@ -0,0 +1,4 @@
+from model import db, SavedTotal
+
+db.connect()
+db.create_tables([SavedTotal])
diff --git a/templates/add.jinja2 b/templates/add.jinja2
@@ -0,0 +1,14 @@
+{% extends 'base.jinja2' %}
+
+{% block subtitle %}Add them up! {% endblock subtitle %}
+
+{% block content %}
+	<p>Total: <span id="total">{{ session.total }}</span></p>
+	<form method="POST">
+		<label for="number">Number: </label><input type="number" id="number" name="number">
+		<input type="submit" value="Add It!">
+	</form>
+	<form method="POST" action="{{ url_for('save') }}">
+		<input type="submit" value="Save It!">
+	</form>
+{% endblock content %}
diff --git a/templates/base.jinja2 b/templates/base.jinja2
@@ -0,0 +1,10 @@
+<html>
+<head>
+<title>Calculator</title>
+</head>
+<body>
+	<h1>Calculator</h1>
+	<h2>{% block subtitle %} {% endblock subtitle %}</h2>
+	{% block content %} {% endblock content %}
+</body>
+</html>
diff --git a/templates/retrieve.jinja2 b/templates/retrieve.jinja2
@@ -0,0 +1,10 @@
+{% extends 'base.jinja2' %}
+
+{% block subtitle %} Retrieve a Total {% endblock subtitle %}
+
+{% block content %}
+<form method="GET">
+	<label for="code">Code: </label><input type="text" id="code" name="code">
+	<input type="submit" value="Retrieve It!">
+</form>
+{% endblock content %}
diff --git a/templates/save.jinja2 b/templates/save.jinja2
@@ -0,0 +1,7 @@
+{% extends "base.jinja2" %}
+
+{% block subtitle %} Save Your Progress {% endblock subtitle %}
+
+{% block content %}
+Your code is: <span id="code">{{ code }}</span>
+{% endblock content %}
