maehr/typademic

import os
import uuid

from flask import session, render_template, request, send_file, redirect, \
    url_for, current_app

from typademic.uploads import blueprint
from typademic.utils import remove_all_files_recursively, sh_pandoc


@blueprint.route('/', methods=['GET'])
def index():
    if 'uid' not in session:
        return render_template('index.html', files=None, error='')
    session_path = os.path.join(current_app.config['UPLOADED_PATH'],
                                session['uid'])
    files = os.listdir(session_path)
    return render_template('index.html', files=files, error='')


@blueprint.route('/', methods=['POST'])
def upload():
    if 'uid' not in session:
        try:
            uid = uuid.uuid4().hex
            session['uid'] = uid
            session_path = os.path.join(current_app.config['UPLOADED_PATH'],
                                        session['uid'])
            # ensure the upload folder exists
            os.mkdir(session_path)
        except Exception as e:
            return render_template('index.html', files=None, error=str(e))
    session_path = os.path.join(current_app.config['UPLOADED_PATH'],
                                session['uid'])
    try:
        f = request.files.get('file')
        f.save(os.path.join(session_path, f.filename))
        files = os.listdir(session_path)
        return render_template('index.html', files=files, error='')
    except Exception as e:
        files = os.listdir(session_path)
        return render_template('index.html', files=files, error=str(e))


@blueprint.route('/clear', methods=['GET'])
def clear():
    if 'uid' not in session:
        return render_template('index.html',
                               files=None,
                               error=None,
                               info='Nothing to remove.')
    session_path = os.path.join(current_app.config['UPLOADED_PATH'],
                                session['uid'])
    try:
        remove_all_files_recursively(session_path)
        return render_template('index.html',
                               files=None,
                               error=None,
                               info='All files are successfully removed.')
    except Exception as e:
        files = os.listdir(session_path)
        return render_template('index.html', files=files, error=str(e))


@blueprint.route('/pdf', methods=['GET'])
def pdf():
    return render_markdown(output_format='pdf')


@blueprint.route('/docx', methods=['GET'])
def docx():
    return render_markdown(output_format='docx')


def render_markdown(output_format):
    # No files uploaded
    if 'uid' not in session:
        return redirect(url_for('uploads.index'))
    session_path = os.path.join(current_app.config['UPLOADED_PATH'],
                                session['uid'])
    output_filename = 'typademic.' + output_format
    files = os.listdir(session_path)
    md_files = [file for file in files if file.endswith('.md')]
    if not md_files:
        return render_template(
            'index.html',
            files=files,
            error='No Markdown file was uploaded. Please reset and try again.')
    try:
        sh_pandoc(md_files, output_filename, session_path)
        return send_file(os.path.join(session_path, output_filename),
                         attachment_filename=output_filename)
    except Exception as e:
        return render_template('index.html', files=files, error=str(e))