如何在 flask 中使用自定义名称将图像保存在上传文件夹中？

Question

我想用自定义名称保存此图像，以便让我的 HTML 文件轻松显示该图像。 我怎样才能做到这一点？

Answer 1

import pathlib

def generate_custom_name(original_file_name):
    return "my_custom_file_name" + pathlib.Path(original_file_name).suffix
    
@app.route('/prediction', methods=['POST'])
def success():
    if request.method == 'POST':
        f = request.files['file']
        file_name = generate_custom_name(f.filename)
        f.save(os.path.join(UPLOAD_FOLDER, file_name))
        return "", 200