app.py

from flask import Flask, render_template, send_from_directory
from os import sep
from os.path import isdir, dirname, basename, relpath, join, exists

from config import models_dir, server_port
from filesystem import Scaner


app = Flask(__name__)

# https://flask.palletsprojects.com/en/1.1.x/config/#SEND_FILE_MAX_AGE_DEFAULT
app.config['SEND_FILE_MAX_AGE_DEFAULT'] = 0

@app.route('/', methods=['GET'])
def index():
    return list_dir(models_dir)
 
@app.route('/<path:path>', methods=['GET'])
def serve(path):
    
    fetch_latest = '@latest' in path
    real_path = join(models_dir, path.replace('@latest', ''))
    
    if not exists(real_path):
        return 'Not Found', 404
    
    if ".." in real_path:
        return 'Not Found', 404

    if isdir(real_path):

        if fetch_latest:
            latest_entry = Scaner(real_path).latest_entry
            
            if not latest_entry:
                return 'No Models Found', 404
            
            return download_file(latest_entry.path)
        
        else:
            return list_dir(real_path)
    else:
        return download_file(real_path)


def list_dir(path):
    rel_path = relpath(path, models_dir)
    parent_path = dirname(rel_path)
    entries = Scaner(path).entries
    return render_template('index.html', sep=sep, parent_path=parent_path, path=rel_path, entries=entries)

def download_file(path):
    return send_from_directory(
        dirname(path),
        basename(path),
        as_attachment=True,
        attachment_filename=basename(path),
        cache_timeout=0
    )


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=server_port)