application.py

from flask import Flask, request, redirect, url_for, flash, \
    render_template, send_from_directory, jsonify, abort
from urlparse import urljoin
from werkzeug.utils import secure_filename
from astropy import log
from flask_bootstrap import Bootstrap
from flask_login import (login_required, current_user, login_user,
                         LoginManager, logout_user)

from aws_controller.upload_download_s3 import upload_to_s3, return_s3_connection

from app import InputForm, ResultInfo

from app import db
from app.models import Data
from app.forms import EnterDBInfo, RetrieveDBInfo
from app.user import User

from flask.ext.sqlalchemy import SQLAlchemy

UPLOAD_FOLDER = 'uploads/'

app = Flask(__name__)
Bootstrap(app)
app.config.from_object('config.AWSConfig')
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

key = app.config['AWS_KEY']
secret = app.config['AWS_SECRET']

login_manager = LoginManager(app=app)


@app.route("/")
def default():
    # display welcome page
    return render_template('index.html')


@app.route('/submit', methods=['GET', 'POST'])
@login_required
def submit():
    form = InputForm()
    form.filename(multiple="")
    if request.method == "POST":
        if form.validate_on_submit():
            return render_template('job_submission.html')
        else:
            return render_template('submit.html', form=form)
    return render_template('submit.html', form=form)


#@app.route('/upload', methods=['POST'])
#def upload(timestamp):
#    if request.method == 'POST':
#        data_file = request.files('file')
#        file_name = secure_filename(data_file.filename)
#        upload_to_s3(timestamp, file_name, create_bucket=True,
#                     aws_access={"aws_access_key_id": key,
#                                 "aws_secret_access_key": secret})

#        return jsonify(name=file_name)


@login_manager.user_loader
def load_user(username):

    user = Data.user_query(username)

    return user


@app.route("/login", methods=['GET', 'POST'])
def login():
    '''
    Login form for users
    '''
    form1 = EnterDBInfo(request.form)

    if request.method == 'POST' and form1.validate():
        user_name=form1.username.data
        query_test=Data.query.filter_by(nickname=user_name.encode('ascii','ignore')).first()
        if query_test!=None:
            string_response='You already have an account. Please sign in.'
            return render_template('thanks2.html', output=string_response)
        else:
            data_entered = Data(nickname=form1.username.data,password=form1.passwd.data,inst=form1.institute.data,emailaddress=form1.email_ad.data)
            try:
                db.session.add(data_entered)
                db.session.commit()
                db.session.close()
            except:
                db.session.rollback()
            return render_template('thanks.html', nickname=form1.username.data,password=form1.passwd.data)

    return render_template('login.html', form=form1)


@app.route("/login2", methods=['GET', 'POST'])
def login2():
    form2 = RetrieveDBInfo(request.form)

    if request.method == 'POST' and form2.validate():
        user_return = form2.username_ret.data
        passwd_return = form2.passwd_ret.data

        user = Data.query_by_name(user_return.encode('ascii', 'ignore'),
                                  passwd_return.encode('ascii', 'ignore'))

        db.session.close()

        if not isinstance(user, str):
            login_user(user)
            return redirect(url_for('submit'))
        else:
            if user == "no_user":
                string_response = 'The username {} does not exist in the ' \
                    'database. Please sign up.'.format(user_return)
            elif user == "no_password":
                string_response = \
                    'The password entered for username {} is incorrect.' \
                    ' Please try again.'.format(user_return)
            else:
                string_response = "Unknown error occurred."

            return render_template('thanks2.html', output=string_response)

    return render_template('login2.html', form=form2)


@app.route("/logout")
@login_required
def logout():
    logout_user()
    return redirect(url_for('login2'))


@app.route("/summary/<int:job_name>", methods=['POST'])
@login_required
def summary(job_name, params):
    output = ResultInfo(params)

    return render_template('summary.html',
                           title='Results for {}'.format(job_name),
                           output=output)


@app.route('/download/<int:filename>/', methods=['GET', 'POST'])
def download(filename, bucket_name):
    # Connect to S3
    s3conn = return_s3_connection({"aws_access_key_id": key,
                                   "aws_secret_access_key": secret})
    try:
        bucket = s3conn.get_bucket(bucket_name)
    except Exception:
        abort(404)

    if bucket.get_key(filename) is None:
        raise ValueError("The filename ({0}) does not exist in the given "
                         "bucket ({1})".format(filename, bucket_name))

    # redirect to the url of the file hosted on S3
    return redirect(urljoin(bucket_name, filename))


if __name__ == '__main__':
    log.setLevel(10)
    app.run(debug=True)