Respond with 404 Not Found

hel-repo · Nov 23, 2016 · 64cafe3 · 64cafe3
1 parent a28915d
commit 64cafe3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/hel/views.py b/hel/views.py
@@ -164,6 +164,8 @@ def auth(request):
              context=Package,
              permission='pkg_update')
 def update_package(context, request):
+    if context.retrieve() is None:
+        raise HTTPNotFound()
     try:
         params = request.json_body
     except:
@@ -460,6 +462,8 @@ def list_packages(context, request):
              context=User,
              permission='user_update')
 def update_user(context, request):
+    if context.retrieve() is None:
+        raise HTTPNotFound()
     try:
         params = request.json_body
     except:
@@ -507,6 +511,8 @@ def get_user(context, request):
              context=User,
              permission='user_delete')
 def delete_user(context, request):
+    if context.retrieve() is None:
+        raise HTTPNotFound()
     context.delete()
 
     raise HTTPNoContent()