snuffster/szurubooru
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

server/api: support file uploads

Browse source
This commit is contained in:
rr- 2016-04-08 22:06:15 +02:00
parent b78d018870
commit 3234da76ad
1 changed files with 17 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
server/szurubooru/middleware/json_translator.py
View file

@ -1,3 +1,4 @@
import cgi
import datetime
import json
import falcon
@ -19,14 +20,28 @@ class JsonTranslator(object):
if request.content_length in (None, 0):
return
body = request.stream.read()
request.context.files = {}
if 'multipart/form-data' in (request.content_type or ''):
# obscure, claims to "avoid a bug in cgi.FieldStorage"
request.env.setdefault('QUERY_STRING', '')
form = cgi.FieldStorage(fp=request.stream, environ=request.env)
for key in form:
if key != 'metadata':
request.context.files[key] = (
form.getvalue(key),
getattr(form[key], 'filename', None))
body = form.getvalue('metadata')
else:
body = request.stream.read().decode('utf-8')
if not body:
raise falcon.HTTPBadRequest(
'Empty request body',
'A valid JSON document is required.')
try:
request.context.request = json.loads(body.decode('utf-8'))
request.context.request = json.loads(body)
except (ValueError, UnicodeDecodeError):
raise falcon.HTTPError(
falcon.HTTP_401,