From 807997f239dfd940135e962b7c0a64ef8003b862 Mon Sep 17 00:00:00 2001 From: Tris Date: Thu, 10 Sep 2020 12:03:56 +1000 Subject: [PATCH] Fixed key/location --- interface/views.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/interface/views.py b/interface/views.py index d6c9279..c1140d6 100644 --- a/interface/views.py +++ b/interface/views.py @@ -73,8 +73,13 @@ class S3CompleteMixin(View): self.object = self.get_object() if self.always_set or not self.object.key: - uri = urlparse(request.GET['location']) - self.object.key = uri.path[1:] + if 'location' in request.GET: + uri = urlparse(request.GET['location']) + self.object.key = uri.path[1:] + elif 'key' in request.GET: + self.object.key = request.GET['key'] + else: + raise KeyError("No key or location found") self.object.save() return super().get(request, *args, **kwargs)