Initial commit

4 files changed, 50 insertions, 0 deletions

diff --git a/opendc/api/v1/specifications/failure-models/__init__.py b/opendc/api/v1/specifications/failure-models/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/opendc/api/v1/specifications/failure-models/__init__.py
diff --git a/opendc/api/v1/specifications/failure-models/endpoint.py b/opendc/api/v1/specifications/failure-models/endpoint.py
new file mode 100644
index 00000000..2530d032
--- /dev/null
+++ b/opendc/api/v1/specifications/failure-models/endpoint.py
@@ -0,0 +1,17 @@
+from opendc.models.failure_model import FailureModel
+from opendc.util.rest import Response
+
+def GET(request):
+    """Get all Failure Models."""
+
+    # Get the FailureModels
+
+    failure_models = FailureModel.query()
+
+    # Return the FailureModels
+
+    return Response(
+        200,
+        'Successfully retrieved FailureModels.',
+        [x.to_JSON() for x in failure_models]
+    )
diff --git a/opendc/api/v1/specifications/failure-models/id/__init__.py b/opendc/api/v1/specifications/failure-models/id/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/opendc/api/v1/specifications/failure-models/id/__init__.py
diff --git a/opendc/api/v1/specifications/failure-models/id/endpoint.py b/opendc/api/v1/specifications/failure-models/id/endpoint.py
new file mode 100644
index 00000000..8a7d2e1d
--- /dev/null
+++ b/opendc/api/v1/specifications/failure-models/id/endpoint.py
@@ -0,0 +1,33 @@
+from opendc.models.failure_model import FailureModel
+from opendc.util import exceptions
+from opendc.util.rest import Response
+
+def GET(request):
+    """Get this Failure Model."""
+
+    # Make sure required parameters are there
+
+    try:
+        request.check_required_parameters(
+            path = {
+                'id': 'int'
+            }
+        )
+
+    except exceptions.ParameterError as e:
+        return Response(400, e.message)
+
+    # Instantiate a FailureModel and make sure it exists
+
+    failure_model = FailureModel.from_primary_key((request.params_path['id'],))
+
+    if not failure_model.exists():
+        return Response(404, '{} not found.'.format(failure_model))
+
+    # Return this FailureModel
+
+    return Response(
+        200,
+        'Successfully retrieved {}.'.format(failure_model),
+        failure_model.to_JSON()
+    )