python + flask 做一个图床

from flask import Flask, render_template, request, redirect, url_for, flash from werkzeug.utils import secure_filename import os import uuid app = Flask(__name__) app.config['UPLOAD_FOLDER'] = 'static/uploads' app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 # 限制上传文件大小为16MB app.secret_key = 'your_secret_key' # 用于 flash 消息 # 允许的图片扩展名 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'bmp', 'ico', 'tif', 'tiff', 'psd', 'eps', 'ai'} # 检查文件扩展名是否允许 def allowed_file(filename): return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS # 确保上传文件夹存在 if not os.path.exists(app.config['UPLOAD_FOLDER']): os.makedirs(app.config['UPLOAD_FOLDER']) @app.route('/') def index(): return render_template('index.html') @app.route('/upload', methods=['POST']) def upload_file(): if 'file' not in request.files: flash('没有选择文件') return redirect(url_for('index')) file = request.files['file'] if file.filename == '': flash('没有选择文件') return redirect(url_for('index')) if file and allowed_file(file.filename): # 生成唯一的文件名 filename = str(uuid.uuid4()) + '.' + file.filename.rsplit('.', 1)[1].lower() file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename)) # 生成图片的访问链接 image_url = url_for('static', filename='uploads/' + filename, _external=True) flash(f'图片上传成功！链接: {image_url}') return redirect(url_for('index')) flash('不支持的文件类型') return redirect(url_for('index')) if __name__ == '__main__': app.run(debug=True)