budget_calculator.py

import argparse
import gc
import keyring
import random
import wtforms
from contextlib import closing
from flask import Flask, request, session, g, redirect, url_for, \
    abort, render_template, flash
import firebase_admin
from firebase_admin import credentials, auth, firestore
from firebase_conf import params as fb_params
from flask_wtf import Form
from google.cloud.exceptions import NotFound
import jinja2
from functools import wraps
from passlib.hash import sha256_crypt
from wtforms import Form, BooleanField, PasswordField, StringField, validators

# Handle Arguments
parser = argparse.ArgumentParser()
parser.add_argument("--prod", action="store_true", default=False,
					help="Run the app in a production environment")
parser.add_argument("--debug", action="store_true", default=False,
					help="Run the app in debug mode. **Avoid using in PROD!**")
args = parser.parse_args()

# initialize app
app = Flask(__name__)
cred = credentials.Certificate("credentials.json")
fb_app = firebase_admin.initialize_app(cred, {
        "databaseURL": fb_params["DB_URL"],
        "databaseAuthVariableOverride": {
            "uid": fb_params["DB_AUTH_OR_UID"]
        }
    })
db = firestore.client()

if args.prod:
    app.config.from_object('flask_conf.ProdConfig')
    if args.debug:
        print("**CAUTION** Running in PROD with Debug!")
    else:
        print("Running in PROD Mode")
else:
    app.config.from_object('flask_conf.TestConfig')
    if args.debug:
        print("Running in TEST debug mode")
    else:
        print("Running in TEST Mode")

app.config["DEBUG"] = args.debug
args.debug = None

#test database access
if app.config["DEBUG"]:
    doc_ref = db.document(u"users/4rxvrvjwJX35qkPgb914")
    doc = doc_ref.get()
    print("User: {} ==> {}".format(doc.id, doc.to_dict()))

#
#
#   Template navigation
#
#

def login_required(f):
    @wraps(f)
    def wrap(*args, **kwargs):
        if 'logged_in' in session:
            return f(*args, **kwargs)
        else:
            flash("You need to be logged in to do that!")
            return redirect(url_for('login'))
    return wrap

def debug_mode_only(f):
    @wraps(f)
    def wrap(*args, **kwargs):
        if app.config["DEBUG"]:
            return f(*args, **kwargs)
        else:
            flash("Not authorized - Error: 0x00001")
            return redirect(url_for("index"))
    return wrap
        
@app.route('/index', methods = ["GET", "POST"])
@app.route('/', methods = ["GET", "POST"])
def index():
    '''
        TODO: Make pretty homepage
    '''
    return render_template('homepage.html')

@app.route('/test')
@debug_mode_only
def test():
    items = []
    for i in range(1,11):
        i = str(i)
        an_item = dict(desc="Item "+i,num1=(random.randint(1,11)*100),num2=0)
        items.append(an_item)
    
    return render_template('month design.html')    

@app.route('/graph')
@debug_mode_only
def graph():
    return render_template('graph.html')
    
#display user's profile information
@app.route('/myprofile')
@login_required
def myprofile():
    return render_template('profile.html')
    
#display budget for a single month
@app.route('/monthview')
@login_required
def monthview():
    return render_template('month.html')
    
#display overview for a year
@app.route('/yearview')
@login_required
def yearview():
    return render_template('year.html')


#display login form
@app.route('/login', methods = ["GET", "POST"])
def login():
    error = ''
    try:
        if request.method == "POST":
            #data = db_execute("SELECT * FROM `users` WHERE `username` = \"{}\";".format(
            #    request.form['username']), fetchall=True)
            # TODO: See if user exists
            flash("Please try again later - Error: 0x00003")
            if data:
                #data = c.fetchone()
                if request.form['password'] == data['password'] and request.form['username'] == data['username']:
                    user = User(data['uid'])
                    login_user(user)
                    flash("You're now logged in. Hello {}".format(session['username']))
                    print("You're now logged in. Hello {}".format(session['username']))
                    return redirect(url_for('index'))
                else:
                    error = "Invalid Credentials. Try Again"
            else: #username not found in database
                error = "Invalid Credentials. Try Again"
        gc.collect
        return render_template("login.html", error=error)
        
    except Exception as e:
        flash(e) #used for debugging
        return render_template('login.html', error=e)
    
@app.route('/logout')
@login_required
def logout():
    session.clear()
    flash("You have been logged out")
    gc.collect()
    return redirect(url_for('.index'))

class RegistrationForm(Form):
    username = StringField('Username', validators = [validators.Length(min=4, max=20)])
    email = StringField('Email Address', validators = [validators.Email()])
    password = PasswordField('Password', validators = [validators.InputRequired(), 
        validators.EqualTo('confirm', message="Passwords must match.")])
    confirm = PasswordField('Repeat Password') 
    accept_tos = BooleanField('''I accept the <a href="/tos/">Terms of Service</a>
     and the <a href="/privacy/">Privacy Notice</a>.''', validators = [validators.InputRequired()])


#display registration
@app.route('/register', methods = ["GET", "POST"])
def register():
    try:
        form = RegistrationForm(request.form)
        if request.method == "POST" and form.validate():
            username = form.username.data
            email = form.email.data
            password = str(form.password.data)

            #data = db_execute("SELECT * FROM `users` WHERE `username` = \"{}\";".format(
            #    (username)))
            # TODO: check to see if username is already in database
            data = None

            if data is None:
                flash("Please try again later - Error: 0x00002")
            elif int(data) > 0:
                flash("That username is already taken")
                return render_template("register.html", form=form)
            else:
             #   db_execute("INSERT INTO `users` (`username`, `password`, `email`) VALUES (\"{}\", \"{}\", \"{}\");".format( 
             #       username, password, email))
                flash("Thank you for registering")

                return redirect(url_for('.index'))
    except Exception as e:
        flash(str(e))

    return render_template('register.html', form=form)

class User:

    def __init__(self, id):
        self.id = id
        self.name = "user" + str(id)
        self.password = self.name + "_secret"

    def __repr__(self):
        return "%d/%s/%s" % (self.id, self.name, self.password)
    
    def get_id(self):
        return str(self.id).encode("utf-8").decode("utf-8") 
                                     
#
#
#   Error handling
#
#
@app.errorhandler(404)
def page_not_found(e):
    return render_template('404.html')

@app.errorhandler(500)
def internal_server_error(e):
    try:
        return render_template('500.html', error = e)
    except Exception as e:
        return ("Something went really wrong here: " + e)

#
#
#   Main
#
#
if __name__ == '__main__':
    app.run(host='0.0.0.0')