ExternalVendorCode/gns3-server
1
0
Fork 0
mirror of https://github.com/GNS3/gns3-server.git synced 2025-06-18 07:18:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

The upload interfaces allow user to choose an image type

Browse Source
This commit is contained in:
Julien Duponchelle
2015-03-07 13:52:40 +01:00
parent 1b68a54234
commit d126db1fe9
2 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gns3server/handlers/upload_handler.py
View File

@ -55,9 +55,13 @@ class UploadHandler:
response.redirect("/upload")
return
destination_path = os.path.join(UploadHandler.image_directory(), data["file"].filename)
if data["type"] not in ["IOU", "QEMU", "IOS"]:
raise HTTPForbidden("You are not authorized to upload this kind of image {}".format(data["type"]))
destination_dir = os.path.join(UploadHandler.image_directory(), data["type"])
destination_path = os.path.join(destination_dir, data["file"].filename)
try:
os.makedirs(UploadHandler.image_directory(), exist_ok=True)
os.makedirs(destination_dir, exist_ok=True)
with open(destination_path, "wb+") as f:
chunk = data["file"].file.read()
f.write(chunk)