__init__.py

# app/__init__.py

# local imports
from __future__ import print_function

import os
import os.path
import sys
import time

import datetime
import subprocess
# third-party imports
from flask import Flask, redirect, url_for, request, render_template, flash
from flask_wtf import FlaskForm
from flask_bootstrap import Bootstrap
from wtforms import StringField, SubmitField, validators


def create_app(config_name):

    app = Flask(__name__)
    Bootstrap(app)

    global return_url
    return_url = ''

    class MainForm(FlaskForm):
        fullname = StringField('Full Name: ', [validators.DataRequired(), ])
        submit = SubmitField('Submit')

    @app.route('/', methods=['GET', 'POST'])
    def index():
        global return_url
        user = request.remote_user

        if "redir" in request.args and return_url == "":
            return_url = request.args.get("redir") or "/pun/sys/dashboard"

        username = False
        form = MainForm()
        if form.is_submitted():
            username = form.fullname.data
            form.fullname.data = ''

            return redirect(url_for('success', username=str(user), fullname=username))

        return render_template('auth/SignUp.html', form=form, user=user)

    @app.route('/success/<username>/<fullname>')
    def success(username, fullname):

        global return_url
        print(username, fullname, return_url, file=sys.stdout)

        # Deliver arguments to script.
        tempString = 'ssh ohpc "sudo /opt/ohpc_user_create/user_create ' + username + " " + fullname + '"'
        print(tempString, file=sys.stdout)

        try:
            time_stamp = time.strftime("%Y%m%d-%H%M%S")
            blazer_path = "/var/www/ood/register/flask_user_reg"
            complete_file_name = os.path.join(blazer_path, time_stamp+".txt")
            file = open(complete_file_name, "w")
            file.write(fullname)
            file.close()
            return redirect(return_url, 302)

        except:
            flash("Registration Failed. Please try again.")
            return redirect(url_for('index'))

    @app.errorhandler(403)
    def forbidden(error):
        return render_template('errors/403.html', title='Forbidden'), 403

    @app.errorhandler(404)
    def page_not_found(error):
        return render_template('errors/404.html', title='Page Not Found'), 404

    @app.errorhandler(500)
    def internal_server_error(error):
        return render_template('errors/500.html', title='Server Error'), 500

    return app