python-api-server/app.py
2023-04-13 11:46:17 +02:00

72 lines
2.2 KiB
Python

import os
import re
import uuid
from flask import Flask, request, jsonify, send_from_directory
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = os.environ.get('UPLOAD_FOLDER', './uploads')
app.config['MAX_CONTENT_LENGTH'] = 5 * 1024 * 1024 # 5MB
VALID_FILENAME_REGEX = r'^[a-zA-Z0-9\-_\.]+$'
def is_valid_filename(filename):
return bool(re.match(VALID_FILENAME_REGEX, filename))
@app.route('/upload', methods=['POST'])
def upload_file():
if 'file' not in request.files:
return jsonify({'error': 'No file part in the request'}), 400
file = request.files['file']
if file.filename == '':
return jsonify({'error': 'No file selected for upload'}), 400
if not is_valid_filename(file.filename):
return jsonify({'error': 'Invalid filename. Only alphanumeric characters, hyphens, underscores, and periods are allowed.'}), 400
filename = file.filename
file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
return jsonify({'success': 'File {} successfully uploaded'.format(filename)})
@app.route('/download/<filename>', methods=['GET'])
def download_file(filename):
try:
return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
except FileNotFoundError:
return jsonify({'error': 'File not found'}), 404
@app.route('/list', methods=['GET'])
def list_files():
files = []
total_size = 0
for filename in os.listdir(app.config['UPLOAD_FOLDER']):
path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
if os.path.isfile(path):
size = os.path.getsize(path)
files.append({'filename': filename, 'size': size})
total_size += size
return jsonify({'files': files, 'count': len(files), 'total_size': total_size})
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0', port=int(os.environ.get('PORT', 5000)))
# Upload
# curl -X POST -H "Authorization: Bearer myuploadtoken" -F "file=@/path/to/file" http://docker10.grote.lan:5040/upload
# Download
# curl -X GET http://docker10.grote.lan:5040/download/filename.ext > filename.ext
# List
# curl -X GET http://docker10.grote.lan:5040/files
# Metriken
# curl -X GET http://docker10.grote.lan:5040/metrics
# upload check
# download
# farben im putput weg
# wsgi server