36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from flask import Flask, request, jsonify, send_from_directory
|
|
from flask_cors import CORS
|
|
import os
|
|
|
|
app = Flask(__name__)
|
|
CORS(app)
|
|
|
|
UPLOAD_FOLDER = 'uploads'
|
|
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
|
|
|
|
if not os.path.exists(UPLOAD_FOLDER):
|
|
os.makedirs(UPLOAD_FOLDER)
|
|
|
|
@app.route('/upload', methods=['POST'])
|
|
def upload_file():
|
|
if 'file' not in request.files:
|
|
return jsonify({'error': 'No file part'}), 400
|
|
file = request.files['file']
|
|
if file.filename == '':
|
|
return jsonify({'error': 'No selected file'}), 400
|
|
if file:
|
|
filename = file.filename
|
|
file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
|
|
return jsonify({'success': True, 'filename': filename})
|
|
|
|
@app.route('/images', methods=['GET'])
|
|
def get_images():
|
|
images = os.listdir(app.config['UPLOAD_FOLDER'])
|
|
return jsonify(images)
|
|
|
|
@app.route('/uploads/<filename>')
|
|
def uploaded_file(filename):
|
|
return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True) |