#!/usr/bin/env python3

# coding: utf-8

import os

import errno

import re

import json

import tempfile

import logging

import threading

import sys

import report

from flask import Flask, Blueprint, Response

from flask import escape, redirect, render_template, request, send_from_directory, session, url_for

from flask.ext.session import Session

from werkzeug import secure_filename

from redis import Redis

# Set up logger

log = logging.getLogger(__name__)

# Create flask app

app = Flask(__name__)

# Get appropriate directory for templates

directory = os.path.dirname(os.path.abspath(__file__))

# Set up blueprint

APPLICATION_ROOT = os.environ.get('PROXY_PATH', '/').strip() or '/'

bax = Blueprint('bax', __name__, template_folder=os.path.join(directory, 'templates'))

# Look at end of file for where this blueprint is actually registered to the app

# App options (loaded from_object)

UPLOAD_FOLDER = '/tmp/uploads'

MAX_CONTENT_LENGTH = 512 * 1024 * 1024

DEBUG = True

ALLOWED_EXTENSIONS = {'txt', 'csv', 'bin', 'bax'}

IMAGE_EXTENSIONS = {'jpg', 'png', 'svg', 'gif'}

host = os.environ.get('REDIS_URL', os.environ.get('REDIS_HOSTNAME', 'localhost'))

port = int(host.split(':')[1:][0]) if len(host.split(':')[1:]) else 6379

redis = Redis(host=host.split(':')[0], port=port)

# Flask-Session module setup

SESSION_TYPE = 'redis'

SESSION_REDIS = redis

app.config.from_object(__name__)

Session(app)

flask_options = {

'host': '0.0.0.0',

'threaded': True

}

#

# Auxiliary functions

#

def mkdir_p(path):

raise

def escape_filename(filename):

return re.sub('[^a-zA-Z0-9\-_\.]', '', secure_filename(filename))

def save_file(fil, extensions_list):

return None

#

# Report generation worker thread

#

def report_worker(sid):

raise e

# Store files in redis by keys prefix:sid:filename

def rkey(prefix, sid, suffix=None):

return prefix + ":" + sid + (":" + suffix if suffix else '')

def dict_utf8(j):

}

def get_template_variables():

return tpl

#

# File handling

#

def list_fkeys(sid='*'):

return fkeys

def get_files(sid):

key=lambda x: x['original_name'])

def add_file(file_info):

raise KeyError("File already exists!")

def count_files():

return len(list_fkeys())

# Atomic multi-key delete: http://stackoverflow.com/a/16974060/1681205

def delete_files():

)

#

# Job handling

#

def get_job(sid):

return dict_utf8(redis.hgetall(rkey('job', sid)))

def put_job(sid, job):

return redis.hmset(rkey('job', sid), job)

def upd_job(sid, key, value):

return redis.hset(rkey('job', sid), key, value)

def has_job(sid):

return redis.exists(rkey('job', sid))

def rem_job(sid):

return redis.delete(rkey('job', sid))

#

# Application Routes

#

@bax.route('/', methods=['GET'])

def index():

return render_template("upload.htm")

@bax.route('generate', methods=['POST'])

def generate():

return redirect(url_for('.job'))

@bax.route('job', methods=['GET', 'POST'])

def job():

return render_template("jobstatus.htm", **get_template_variables())

@bax.route('download', methods=['GET'])

def download():

return "No file", 400

#

# API Routes

#

@bax.route("clear", methods=['GET', 'POST'])

def clear():

return 'Files cleared', 200

@bax.route("upload", methods=['PUT'])

def upload():

return "File type '{}' rejected".format(file_info['file_extension']), 400

@bax.route('status', methods=['GET'])

def status():

return Response(json.dumps(get_template_variables()), mimetype='application/json')

@bax.route('cancel', methods=['GET', 'POST'])

def cancel():

mimetype='application/json')

#

# Register blueprint to the app

#

app.register_blueprint(bax, url_prefix=APPLICATION_ROOT)

# Main. Does not run when running with WSGI

def run():

app.run(**flask_options)

if __name__ == "__main__":

run()