Coverage for application/web.py : 44%

import application 

import flask 

import shutil 

import tempfile 

import pathlib 

import werkzeug.exceptions 

app, dumpdir, archivedir = application.config.create_app() 

@app.route("/") 

def index(): 

""" 

Renders the main page, and unlinks the content (if any) in the temporary 

folders. 

""" 

application.utils.unlink_content(dumpdir) 

application.utils.unlink_content(archivedir) 

return flask.render_template("index.html") 

@app.route("/help") 

def help(): 

""" 

Renders the help page. 

""" 

return flask.render_template("help.html") 

@app.route("/modeling", methods=["POST"]) 

def modeling(): 

""" 

Streams the modeling page, printing useful information to the UI. 

The generated data will be dumped into a tempdir (specified above). 

""" 

def stream_template(template_name, **context): 

app.update_template_context(context) 

t = app.jinja_env.get_template(template_name) 

return t.stream(context) 

internet = application.utils.is_connected() 

stream = flask.stream_with_context(application.modeling.workflow(dumpdir, archivedir)) 

return flask.Response(stream_template("modeling.html", stream=stream, internet=internet)) 

@app.route("/model") 

def model(): 

""" 

Loads the dumped data, deletes the temporary data, and renders the model page. 

""" 

data = application.utils.load_data(dumpdir) 

return flask.render_template("model.html", **data) 

@app.route("/download") 

def download(): 

""" 

Sends the ZIP archive. 

""" 

return flask.send_file(str(pathlib.Path(archivedir, "topicmodeling.zip"))) 

@app.errorhandler(werkzeug.exceptions.HTTPException) 

def handle_http_exception(e): 

""" 

Handles any HTTP Exception, shows a Bootstrap modal with the error 

message and renders the index.html again. 

""" 

return flask.render_template("index.html", http="error", error_message=e) 

for code in werkzeug.exceptions.default_exceptions: 

app.errorhandler(code)(handle_http_exception) 

@app.after_request 

def add_header(r): 

""" 

Clears the cache after a request. 

""" 

r.headers["Cache-Control"] = "no-cache, no-store, must-revalidate" 

r.headers["Pragma"] = "no-cache" 

r.headers["Expires"] = "0" 

r.headers["Cache-Control"] = "public, max-age=0" 

return r