Resolve "Extend NBI with CUD methods for entities in TFS API"

Closes #170 (closed)

src/nbi/service/rest_server/nbi_plugins/tfs_api/Resources.py

@@ -15,33 +15,71 @@
 import json
 from flask.json import jsonify
 from flask_restful import Resource, request
+from werkzeug.exceptions import BadRequest
 from common.proto.context_pb2 import Empty
 from common.tools.grpc.Tools import grpc_message_to_json
 from context.client.ContextClient import ContextClient
+from device.client.DeviceClient import DeviceClient
 from service.client.ServiceClient import ServiceClient
+from slice.client.SliceClient import SliceClient
 from .Tools import (
-    format_grpc_to_json, grpc_connection_id, grpc_context_id, grpc_device_id, grpc_link_id, grpc_policy_rule_id,
-    grpc_service_id, grpc_service, grpc_slice_id, grpc_topology_id)
+    format_grpc_to_json, grpc_connection_id, grpc_context, grpc_context_id, grpc_device,
+    grpc_device_id, grpc_link, grpc_link_id, grpc_policy_rule_id,
+    grpc_service_id, grpc_service, grpc_slice, grpc_slice_id, grpc_topology, grpc_topology_id
+)
 
 class _Resource(Resource):
     def __init__(self) -> None:
         super().__init__()
-        self.client = ContextClient()
+        self.context_client = ContextClient()
+        self.device_client  = DeviceClient()
         self.service_client = ServiceClient()
+        self.slice_client   = SliceClient()
 
 class ContextIds(_Resource):
     def get(self):
-        return format_grpc_to_json(self.client.ListContextIds(Empty()))
+        return format_grpc_to_json(self.context_client.ListContextIds(Empty()))
 
 class Contexts(_Resource):
     def get(self):
-        return format_grpc_to_json(self.client.ListContexts(Empty()))
+        return format_grpc_to_json(self.context_client.ListContexts(Empty()))
+
+    def post(self):
+        json_requests = request.get_json()
+        if 'contexts' in json_requests:
+            json_requests = json_requests['contexts']
+        return [
+            format_grpc_to_json(self.context_client.SetContext(grpc_context(context)))
+            for context in json_requests
+        ]
+
+class Context(_Resource):
+    def get(self, context_uuid : str):
+        return format_grpc_to_json(self.context_client.GetContext(grpc_context_id(context_uuid)))
+
+    def put(self, context_uuid : str):
+        context = request.get_json()
+        if context_uuid != context['context_id']['context_uuid']['uuid']:
+            raise BadRequest('Mismatching context_uuid')
+        return format_grpc_to_json(self.context_client.SetContext(grpc_context(context)))
+
+    def delete(self, context_uuid : str):
+        return format_grpc_to_json(self.context_client.RemoveContext(grpc_context_id(context_uuid)))
 
 class DummyContexts(_Resource):
     def get(self):
-        contexts = grpc_message_to_json(self.client.ListContexts(Empty()), use_integers_for_enums=True)['contexts']
-        devices = grpc_message_to_json(self.client.ListDevices(Empty()), use_integers_for_enums=True)['devices']
-        links = grpc_message_to_json(self.client.ListLinks(Empty()), use_integers_for_enums=True)['links']
+        contexts = grpc_message_to_json(
+            self.context_client.ListContexts(Empty()),
+            use_integers_for_enums=True
+        )['contexts']
+        devices = grpc_message_to_json(
+            self.context_client.ListDevices(Empty()),
+            use_integers_for_enums=True
+        )['devices']
+        links = grpc_message_to_json(
+            self.context_client.ListLinks(Empty()),
+            use_integers_for_enums=True
+        )['links']
 
         topologies  = list()
         slices      = list()
@@ -53,17 +91,17 @@ class DummyContexts(_Resource):
             context_id = grpc_context_id(context_uuid)
 
             topologies.extend(grpc_message_to_json(
-                self.client.ListTopologies(context_id),
+                self.context_client.ListTopologies(context_id),
                 use_integers_for_enums=True
             )['topologies'])
 
             slices.extend(grpc_message_to_json(
-                self.client.ListSlices(context_id),
+                self.context_client.ListSlices(context_id),
                 use_integers_for_enums=True
             )['slices'])
 
             context_services = grpc_message_to_json(
-                self.client.ListServices(context_id),
+                self.context_client.ListServices(context_id),
                 use_integers_for_enums=True
             )['services']
             services.extend(context_services)
@@ -72,7 +110,7 @@ class DummyContexts(_Resource):
                 service_uuid = service['service_id']['service_uuid']['uuid']
                 service_id = grpc_service_id(context_uuid, service_uuid)
                 connections.extend(grpc_message_to_json(
-                    self.client.ListConnections(service_id),
+                    self.context_client.ListConnections(service_id),
                     use_integers_for_enums=True
                 )['connections'])
 
@@ -97,115 +135,191 @@ class DummyContexts(_Resource):
         if len(connections) > 0: dummy_context['connections'] = connections
         return jsonify(dummy_context)
 
-class Context(_Resource):
-    def get(self, context_uuid : str):
-        return format_grpc_to_json(self.client.GetContext(grpc_context_id(context_uuid)))
-
 class TopologyIds(_Resource):
     def get(self, context_uuid : str):
-        return format_grpc_to_json(self.client.ListTopologyIds(grpc_context_id(context_uuid)))
+        return format_grpc_to_json(self.context_client.ListTopologyIds(grpc_context_id(context_uuid)))
 
 class Topologies(_Resource):
     def get(self, context_uuid : str):
-        return format_grpc_to_json(self.client.ListTopologies(grpc_context_id(context_uuid)))
+        return format_grpc_to_json(self.context_client.ListTopologies(grpc_context_id(context_uuid)))
+
+    def post(self, context_uuid : str):
+        json_requests = request.get_json()
+        if 'topologies' in json_requests:
+            json_requests = json_requests['topologies']
+        for topology in json_requests:
+            if context_uuid != topology['topology_id']['context_id']['context_uuid']['uuid']:
+                raise BadRequest('Mismatching context_uuid')
+        return [
+            format_grpc_to_json(self.context_client.SetTopology(grpc_topology(**topology)))
+            for topology in json_requests
+        ]
 
 class Topology(_Resource):
     def get(self, context_uuid : str, topology_uuid : str):
-        return format_grpc_to_json(self.client.GetTopology(grpc_topology_id(context_uuid, topology_uuid)))
+        return format_grpc_to_json(self.context_client.GetTopology(grpc_topology_id(context_uuid, topology_uuid)))
+
+    def put(self, context_uuid : str, topology_uuid : str):
+        topology = request.get_json()
+        if context_uuid != topology['topology_id']['context_id']['context_uuid']['uuid']:
+            raise BadRequest('Mismatching context_uuid')
+        if topology_uuid != topology['topology_id']['topology_uuid']['uuid']:
+            raise BadRequest('Mismatching topology_uuid')
+        return format_grpc_to_json(self.context_client.SetTopology(grpc_topology(topology)))
+
+    def delete(self, context_uuid : str, topology_uuid : str):
+        return format_grpc_to_json(self.context_client.RemoveTopology(grpc_topology_id(context_uuid, topology_uuid)))
 
 class ServiceIds(_Resource):
     def get(self, context_uuid : str):
-        return format_grpc_to_json(self.client.ListServiceIds(grpc_context_id(context_uuid)))
+        return format_grpc_to_json(self.context_client.ListServiceIds(grpc_context_id(context_uuid)))
 
 class Services(_Resource):
     def get(self, context_uuid : str):
-        return format_grpc_to_json(self.client.ListServices(grpc_context_id(context_uuid)))
+        return format_grpc_to_json(self.context_client.ListServices(grpc_context_id(context_uuid)))
+
+    def post(self, context_uuid : str):
+        json_requests = request.get_json()
+        if 'services' in json_requests:
+            json_requests = json_requests['services']
+        for service in json_requests:
+            if context_uuid != service['service_id']['context_id']['context_uuid']['uuid']:
+                raise BadRequest('Mismatching context_uuid')
+        return [
+            format_grpc_to_json(self.service_client.CreateService(grpc_service(**service)))
+            for service in json_requests
+        ]
 
 class Service(_Resource):
     def get(self, context_uuid : str, service_uuid : str):
-        return format_grpc_to_json(self.client.GetService(grpc_service_id(context_uuid, service_uuid)))
-
-    def post(self, context_uuid : str, service_uuid : str): # pylint: disable=unused-argument
-        service = request.get_json()['services'][0]
-        return format_grpc_to_json(self.service_client.CreateService(grpc_service(
-            service_uuid = service['service_id']['service_uuid']['uuid'],
-            service_type = service['service_type'],
-            context_uuid = service['service_id']['context_id']['context_uuid']['uuid'],
-        )))
-
-    def put(self, context_uuid : str, service_uuid : str):  # pylint: disable=unused-argument
-        service = request.get_json()['services'][0]
-        return format_grpc_to_json(self.service_client.UpdateService(grpc_service(
-            service_uuid = service['service_id']['service_uuid']['uuid'],
-            service_type = service['service_type'],
-            context_uuid = service['service_id']['context_id']['context_uuid']['uuid'],
-            status       = service['service_status']['service_status'],
-            endpoint_ids = service['service_endpoint_ids'],
-            constraints  = service['service_constraints'],
-            config_rules = service['service_config']['config_rules']
-        )))
+        return format_grpc_to_json(self.context_client.GetService(grpc_service_id(context_uuid, service_uuid)))
+
+    def put(self, context_uuid : str, service_uuid : str):
+        service = request.get_json()
+        if context_uuid != service['service_id']['context_id']['context_uuid']['uuid']:
+            raise BadRequest('Mismatching context_uuid')
+        if service_uuid != service['service_id']['service_uuid']['uuid']:
+            raise BadRequest('Mismatching service_uuid')
+        return format_grpc_to_json(self.service_client.UpdateService(grpc_service(service)))
 
     def delete(self, context_uuid : str, service_uuid : str):
-        return format_grpc_to_json(self.service_client.DeleteService(grpc_service_id(
-            context_uuid, service_uuid,
-        )))
+        return format_grpc_to_json(self.service_client.DeleteService(grpc_service_id(context_uuid, service_uuid)))
 
 class SliceIds(_Resource):
     def get(self, context_uuid : str):
-        return format_grpc_to_json(self.client.ListSliceIds(grpc_context_id(context_uuid)))
+        return format_grpc_to_json(self.context_client.ListSliceIds(grpc_context_id(context_uuid)))
 
 class Slices(_Resource):
     def get(self, context_uuid : str):
-        return format_grpc_to_json(self.client.ListSlices(grpc_context_id(context_uuid)))
+        return format_grpc_to_json(self.context_client.ListSlices(grpc_context_id(context_uuid)))
+
+    def post(self, context_uuid : str):
+        json_requests = request.get_json()
+        if 'slices' in json_requests:
+            json_requests = json_requests['slices']
+        for slice_ in json_requests:
+            if context_uuid != slice_['slice_id']['context_id']['context_uuid']['uuid']:
+                raise BadRequest('Mismatching context_uuid')
+        return [
+            format_grpc_to_json(self.slice_client.CreateSlice(grpc_slice(**slice_)))
+            for slice_ in json_requests
+        ]
 
 class Slice(_Resource):
     def get(self, context_uuid : str, slice_uuid : str):
-        return format_grpc_to_json(self.client.GetSlice(grpc_slice_id(context_uuid, slice_uuid)))
+        return format_grpc_to_json(self.context_client.GetSlice(grpc_slice_id(context_uuid, slice_uuid)))
+
+    def put(self, context_uuid : str, slice_uuid : str):
+        slice_ = request.get_json()
+        if context_uuid != slice_['slice_id']['context_id']['context_uuid']['uuid']:
+            raise BadRequest('Mismatching context_uuid')
+        if slice_uuid != slice_['slice_id']['slice_uuid']['uuid']:
+            raise BadRequest('Mismatching slice_uuid')
+        return format_grpc_to_json(self.slice_client.UpdateSlice(grpc_slice(slice_)))
+
+    def delete(self, context_uuid : str, slice_uuid : str):
+        return format_grpc_to_json(self.slice_client.DeleteSlice(grpc_slice_id(context_uuid, slice_uuid)))
 
 class DeviceIds(_Resource):
     def get(self):
-        return format_grpc_to_json(self.client.ListDeviceIds(Empty()))
+        return format_grpc_to_json(self.context_client.ListDeviceIds(Empty()))
 
 class Devices(_Resource):
     def get(self):
-        return format_grpc_to_json(self.client.ListDevices(Empty()))
+        return format_grpc_to_json(self.context_client.ListDevices(Empty()))
+
+    def post(self):
+        json_requests = request.get_json()
+        if 'devices' in json_requests:
+            json_requests = json_requests['devices']
+        return [
+            format_grpc_to_json(self.device_client.AddDevice(grpc_device(device)))
+            for device in json_requests
+        ]
 
 class Device(_Resource):
     def get(self, device_uuid : str):
-        return format_grpc_to_json(self.client.GetDevice(grpc_device_id(device_uuid)))
+        return format_grpc_to_json(self.context_client.GetDevice(grpc_device_id(device_uuid)))
+
+    def put(self, device_uuid : str):
+        device = request.get_json()
+        if device_uuid != device['device_id']['device_uuid']['uuid']:
+            raise BadRequest('Mismatching device_uuid')
+        return format_grpc_to_json(self.device_client.ConfigureDevice(grpc_device(device)))
+
+    def delete(self, device_uuid : str):
+        return format_grpc_to_json(self.device_client.DeleteDevice(grpc_device_id(device_uuid)))
 
 class LinkIds(_Resource):
     def get(self):
-        return format_grpc_to_json(self.client.ListLinkIds(Empty()))
+        return format_grpc_to_json(self.context_client.ListLinkIds(Empty()))
 
 class Links(_Resource):
     def get(self):
-        return format_grpc_to_json(self.client.ListLinks(Empty()))
+        return format_grpc_to_json(self.context_client.ListLinks(Empty()))
+
+    def post(self):
+        json_requests = request.get_json()
+        if 'links' in json_requests:
+            json_requests = json_requests['links']
+        return [
+            format_grpc_to_json(self.context_client.SetLink(grpc_link(link)))
+            for link in json_requests
+        ]
 
 class Link(_Resource):
     def get(self, link_uuid : str):
-        return format_grpc_to_json(self.client.GetLink(grpc_link_id(link_uuid)))
+        return format_grpc_to_json(self.context_client.GetLink(grpc_link_id(link_uuid)))
+
+    def put(self, link_uuid : str):
+        link = request.get_json()
+        if link_uuid != link['link_id']['link_uuid']['uuid']:
+            raise BadRequest('Mismatching link_uuid')
+        return format_grpc_to_json(self.context_client.SetLink(grpc_link(link)))
+
+    def delete(self, link_uuid : str):
+        return format_grpc_to_json(self.context_client.RemoveLink(grpc_link_id(link_uuid)))
 
 class ConnectionIds(_Resource):
     def get(self, context_uuid : str, service_uuid : str):
-        return format_grpc_to_json(self.client.ListConnectionIds(grpc_service_id(context_uuid, service_uuid)))
+        return format_grpc_to_json(self.context_client.ListConnectionIds(grpc_service_id(context_uuid, service_uuid)))
 
 class Connections(_Resource):
     def get(self, context_uuid : str, service_uuid : str):
-        return format_grpc_to_json(self.client.ListConnections(grpc_service_id(context_uuid, service_uuid)))
+        return format_grpc_to_json(self.context_client.ListConnections(grpc_service_id(context_uuid, service_uuid)))
 
 class Connection(_Resource):
     def get(self, connection_uuid : str):
-        return format_grpc_to_json(self.client.GetConnection(grpc_connection_id(connection_uuid)))
+        return format_grpc_to_json(self.context_client.GetConnection(grpc_connection_id(connection_uuid)))
 
 class PolicyRuleIds(_Resource):
     def get(self):
-        return format_grpc_to_json(self.client.ListPolicyRuleIds(Empty()))
+        return format_grpc_to_json(self.context_client.ListPolicyRuleIds(Empty()))
 
 class PolicyRules(_Resource):
     def get(self):
-        return format_grpc_to_json(self.client.ListPolicyRules(Empty()))
+        return format_grpc_to_json(self.context_client.ListPolicyRules(Empty()))
 
 class PolicyRule(_Resource):
     def get(self, policy_rule_uuid : str):
-        return format_grpc_to_json(self.client.GetPolicyRule(grpc_policy_rule_id(policy_rule_uuid)))
+        return format_grpc_to_json(self.context_client.GetPolicyRule(grpc_policy_rule_id(policy_rule_uuid)))