Merge pull request #51 from mkiwala/no-such-entity-410

Make 404 configurable

ptero_common/exceptions.py

@@ -0,0 +1,2 @@
+class NoSuchEntityError(RuntimeError):
+    pass

ptero_common/view_wrapper.py

@@ -0,0 +1,18 @@
+from functools import wraps
+from os import environ
+from ptero_common.exceptions import NoSuchEntityError
+
+
+NO_SUCH_ENTITY_STATUS_CODE = int(environ.get(
+    'PTERO_NO_SUCH_ENTITY_STATUS_CODE', 404))
+
+
+def handles_no_such_entity_error(target):
+    @wraps(target)
+    def wrapper(*args, **kwargs):
+        try:
+            result = target(*args, **kwargs)
+        except NoSuchEntityError as e:
+            return {'error': e.message}, NO_SUCH_ENTITY_STATUS_CODE
+        return result
+    return wrapper