flask文件夹列表前文修订版

书接上文：
https://blog.csdn.net/weixin_44517278/article/details/135228805

本文的代码和HTML模板已经修改了上文的问题，目前可以做到正常的上传，返回上层目录，下载

from flask import Flask, render_template, send_file, request, redirect, url_for
import os

app = Flask(__name__)

@app.template_filter('basename')
def basename_filter(value):
    return os.path.basename(value)

@app.route('/')
def index():
    files, folder_names, folder_name = getfile()
    return render_template('index.html', files=files, folder_names=folder_names, folder_name=folder_name)

@app.route('/download/')
def download_file(file_name):
    file_name = os.path.normpath(file_name)  # 替换windows环境下出现的双斜杠 \\ 为 /
    return send_file(file_name, as_attachment=True)

@app.route('/show_folder/')
def show_folder(folder_name):
    files, folder_names, folder_name = getfile(folder_name)
    return render_template('index.html', files=files, folder_names=folder_names, folder_name=folder_name)

# 这里是返回上一级的目录
@app.route('/return_folder/')
def return_folder(folder_name):
    refolder = folder_name
    # 抓完全体的路径
    full_path = os.path.join(os.getcwd(), refolder)
    # 遍历指定（这里是当前）目录下的所有文件和文件夹
    for root, dirs, files in os.walk(os.getcwd(), topdown=True):
        # 遍历文件夹
        for dir in dirs:
            # 当此文件夹的完整路径等于传入文件夹的所有路径的时候，root就是这个文件夹的上级目录
            if os.path.join(root, dir) == full_path:
                # 用这个抓出来上级目录的相对路径，这条命令的作用就是抓相对路径，起始位置为start定义的位置
                folder_name = os.path.relpath(root, start=os.getcwd())
                # 两个目录相同的时候，相对路径就是 . , 当相对路径是这个，代表到根目录了，就不提供返回按钮了
                if folder_name == ".":
                    folder_name = ""
                files, folder_names, folder_name = getfile(folder_name)
                return render_template('index.html', files=files, folder_names=folder_names, folder_name=folder_name)

def getfile(folder_name=""):
    files = []
    folder_names = []
    fileList = os.listdir(os.path.join(os.getcwd(), folder_name))
    for file in fileList:
        file = os.path.join(folder_name, file)
        file = os.path.normpath(file)  # 替换windows环境下出现的双斜杠 \\ 为 /
        if os.path.isfile(file):
            files.append(file)
        else:
            folder_names.append(file)
    return files, folder_names, folder_name

@app.route('/upload', methods=['POST'])
def upload_file():
    # 接受HTML返回的需要下载的文件名
    file = request.files['file']
    # 接受HTML返回的folder_name的值（我这里folder_name是相对路径值）
    folder_name = request.form['folder_name']
    # 这里指定文件上传目录，实现选到那个路径下就上传文件到那个文件夹下
    UPLOAD_FOLDER = folder_name  
    app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
    
    # 检查是否有文件被上传
    if 'file' not in request.files:
        return 'No file part'
    file = request.files['file']

    # 如果用户没有选择文件，浏览器会发送一个空的文件名
    if file.filename == '':
        return 'No selected file'

    # 保存文件到指定目录
    file.save(os.path.join(app.config['UPLOAD_FOLDER'], file.filename))
    files, folder_names, folder_name = getfile(folder_name)
    return render_template('index.html', files=files, folder_names=folder_names, folder_name=folder_name)


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

    
    
    



文件上传下载列表
{% if folder_name == "" or folder_name %}
    {% if folder_name == "" %}
    
当前路径：
    {% else %}
        
当前路径:
            
返回：{{ folder_name|basename }}
    {% endif %}
    

        
        
        
    
    

        
文件:
        {% for filename in files %}
            
{{ filename|basename }}
        {% endfor %}
        
文件夹:
        {% for foldername in folder_names %}
            
{{ foldername|basename }}
        {% endfor %}
    
{% else %}
    
No folder selected
{% endif %}