file browser writen in Python with Flask Blueprint

file_browser.py

import os
from flask import Flask, Blueprint, send_file, render_template, abort, jsonify
from time import ctime


SIZES = ['B', 'K', 'M', 'G']
CURRENT_PATH = os.path.abspath(os.path.curdir)
bp = Blueprint(
    'file_browser',
    __name__,
    template_folder=CURRENT_PATH,
)


def size4human(size):
    index = 0
    while size > 1024:
        size /= 1024
        index += 1

    return '{:.2f}{}'.format(size, SIZES[index])

def get_file_info(files, path):
    info = [{
        'name': f,
        'size': size4human(os.path.getsize(os.path.join(path, f))),
        'update_time': ctime(os.path.getmtime(os.path.join(path, f))),
        'type': 'file' if os.path.isfile(os.path.join(path, f)) else 'folder',
    } for f in files]
    return info


@bp.route('/')
def index():
    return file_browser('.')

@bp.route('/<path:filepath>')
def file_browser(filepath=None):
    path = os.path.join(CURRENT_PATH, filepath)

    if os.path.isdir(path):
        files = get_file_info(os.listdir(path), path)
        return jsonify(files=files)
    
    elif os.path.isfile(path):
        if filepath.endswith('.html'):
            return render_template(filepath)
        return send_file(path, as_attachment=True)

    abort(404)


app = Flask(__name__)
app.register_blueprint(bp)

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