Add DB handlers and rename to v2

5 files changed, 49 insertions, 0 deletions

diff --git a/opendc/api/v2/room-types/__init__.py b/opendc/api/v2/room-types/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/opendc/api/v2/room-types/__init__.py
diff --git a/opendc/api/v2/room-types/endpoint.py b/opendc/api/v2/room-types/endpoint.py
new file mode 100644
index 00000000..fe00f226
--- /dev/null
+++ b/opendc/api/v2/room-types/endpoint.py
@@ -0,0 +1,18 @@
+from opendc.models.room_type import RoomType
+from opendc.util.rest import Response
+
+
+def GET(request):
+    """Get all available room types."""
+
+    # Get the RoomTypes
+
+    room_types = RoomType.query()
+
+    # Return the RoomTypes
+
+    return Response(
+        200,
+        'Successfully retrieved RoomTypes.',
+        [x.to_JSON() for x in room_types]
+    )
diff --git a/opendc/api/v2/room-types/name/__init__.py b/opendc/api/v2/room-types/name/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/opendc/api/v2/room-types/name/__init__.py
diff --git a/opendc/api/v2/room-types/name/allowed-objects/__init__.py b/opendc/api/v2/room-types/name/allowed-objects/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/opendc/api/v2/room-types/name/allowed-objects/__init__.py
diff --git a/opendc/api/v2/room-types/name/allowed-objects/endpoint.py b/opendc/api/v2/room-types/name/allowed-objects/endpoint.py
new file mode 100644
index 00000000..a574591c
--- /dev/null
+++ b/opendc/api/v2/room-types/name/allowed-objects/endpoint.py
@@ -0,0 +1,31 @@
+from opendc.models.allowed_object import AllowedObject
+from opendc.util import exceptions
+from opendc.util.rest import Response
+
+
+def GET(request):
+    """Get this room's allowed objects."""
+
+    # Make sure required parameters are there
+
+    try:
+        request.check_required_parameters(
+            path={
+                'name': 'string'
+            }
+        )
+
+    except exceptions.ParameterError as e:
+        return Response(400, e.message)
+
+    # Get the AllowedObjects
+
+    allowed_objects = AllowedObject.query('room_type', request.params_path['name'])
+
+    # Return the AllowedObjects
+
+    return Response(
+        200,
+        'Successfully retrieved AllowedObjects.',
+        [x.to_JSON() for x in allowed_objects]
+    )