from flask import Flask, request, jsonify
from werkzeug.utils import secure_filename
import os 

app = Flask(__name__)

UPLOAD_FOLDER = ''  # 上传文件保存的目录
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}  # 允许上传的图片文件类型

app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

def allowed_file(filename):
    '''检查文件扩展名是否合法'''
    return '.' in filename and \
           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

@app.route('/upload', methods=['POST'])
def upload():
    '''接收并保存上传的图片'''
    file = request.files['file']

    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
        file.save(os.path.join(app.config['UPLOAD_FOLDER'], "1" + filename))
        return jsonify({'result': 'Upload success.'})
    else:
        return jsonify({'error': 'Invalid file type.'})

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