fix: raw body upload (no base64), remove form parsing, bump 1.0.20
Deploy loadtest / validate (push) Waiting to run
Deploy loadtest / validate (push) Waiting to run
This commit is contained in:
+4
-28
@@ -1,13 +1,12 @@
|
||||
import time
|
||||
import json
|
||||
import base64
|
||||
from flask import Flask, render_template, request
|
||||
from flask_sock import Sock
|
||||
|
||||
app = Flask(__name__)
|
||||
sock = Sock(app)
|
||||
|
||||
VERSION = '1.0.19'
|
||||
VERSION = '1.0.20'
|
||||
|
||||
|
||||
@app.route('/')
|
||||
@@ -39,32 +38,9 @@ def ws_upload(ws):
|
||||
@app.route('/upload', methods=['POST'])
|
||||
def upload():
|
||||
t0 = time.time()
|
||||
total = 0
|
||||
# base64-режим: клиент отправляет поле "data"
|
||||
data = request.form.get('data', '')
|
||||
if data:
|
||||
# убрать префикс data:...;base64,
|
||||
if ',' in data:
|
||||
data = data.split(',', 1)[1]
|
||||
# form-encode превращает + в пробелы — восстановить
|
||||
data = data.replace(' ', '+')
|
||||
# добавить padding если потерялся
|
||||
missing = len(data) % 4
|
||||
if missing:
|
||||
data += '=' * (4 - missing)
|
||||
raw = base64.b64decode(data)
|
||||
total = len(raw)
|
||||
# файловый режим
|
||||
elif request.files:
|
||||
for f in request.files.values():
|
||||
while True:
|
||||
chunk = f.read(8192)
|
||||
if not chunk:
|
||||
break
|
||||
total += len(chunk)
|
||||
# raw-режим: тело как есть
|
||||
else:
|
||||
total = len(request.get_data())
|
||||
# raw-режим: тело как есть (клиент шлёт файл напрямую через xhr.send(file))
|
||||
raw = request.get_data()
|
||||
total = len(raw)
|
||||
elapsed = round((time.time() - t0) * 1000)
|
||||
return {'ok': True, 'bytes': total, 'elapsed_ms': elapsed}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user