1
0
mirror of https://github.com/GNS3/gns3-server synced 2024-11-13 20:08:55 +00:00

Improve memory consumption of file upload with the HTML form

Fix #86
This commit is contained in:
Julien Duponchelle 2015-06-10 14:20:06 +02:00
parent c10e692ca2
commit b36c5f25d2
2 changed files with 9 additions and 4 deletions

View File

@ -71,8 +71,11 @@ class UploadHandler:
try:
os.makedirs(destination_dir, exist_ok=True)
with open(destination_path, "wb+") as f:
chunk = data["file"].file.read()
f.write(chunk)
while True:
chunk = data["file"].file.read(512)
if not chunk:
break
f.write(chunk)
st = os.stat(destination_path)
os.chmod(destination_path, st.st_mode | stat.S_IXUSR)
except OSError as e:

View File

@ -31,8 +31,10 @@ def test_index_upload(server):
def test_upload(server, tmpdir):
content = ''.join(['a' for _ in range(0, 1025)])
with open(str(tmpdir / "test"), "w+") as f:
f.write("TEST")
f.write(content)
body = aiohttp.FormData()
body.add_field("type", "QEMU")
body.add_field("file", open(str(tmpdir / "test"), "rb"), content_type="application/iou", filename="test2")
@ -41,6 +43,6 @@ def test_upload(server, tmpdir):
response = server.post('/upload', api_version=None, body=body, raw=True)
with open(str(tmpdir / "QEMU" / "test2")) as f:
assert f.read() == "TEST"
assert f.read() == content
assert "test2" in response.body.decode("utf-8")