Implement limits API for newton

Change-Id: Ie7063c484af69686f6acd7085e73c16f2f84bf0d
Issue-Id: MULTIVIM-19
Signed-off-by: Bin Yang <bin.yang@windriver.com>

2 files changed, 71 insertions, 0 deletions

diff --git a/newton/newton/requests/urls.py b/newton/newton/requests/urls.py
index e90d9ab6..d14c03dc 100644
--- a/newton/newton/requests/urls.py
+++ b/newton/newton/requests/urls.py
@@ -16,12 +16,14 @@ from django.conf.urls import url
 from rest_framework.urlpatterns import format_suffix_patterns
 
 from views import network
+from views import limits
 
 urlpatterns = [
     url(r'^networks(/(?P<networkid>[0-9a-zA-Z_-]+))?',
         network.Networks.as_view()),
     url(r'^subnets/(?P<subnetid>[0-9a-zA-Z_-]+)',
         network.Subnets.as_view()),
+    url(r'^limits$', limits.Limits.as_view()),
 ]
 
 urlpatterns = format_suffix_patterns(urlpatterns)
diff --git a/newton/newton/requests/views/limits.py b/newton/newton/requests/views/limits.py
new file mode 100644
index 00000000..16c7f72a
--- /dev/null
+++ b/newton/newton/requests/views/limits.py
@@ -0,0 +1,69 @@
+# Copyright (c) 2017 Wind River Systems, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import logging
+import json
+
+from rest_framework import status
+from rest_framework.response import Response
+from rest_framework.views import APIView
+
+from newton.pub.exceptions import VimDriverNewtonException
+
+from util import VimDriverUtils
+
+logger = logging.getLogger(__name__)
+
+
+class Limits(APIView):
+    service = {'service_type': 'compute',
+               'interface': 'public',
+               'region_name': 'RegionOne'}
+
+    service_network = {'service_type': 'network',
+               'interface': 'public',
+               'region_name': 'RegionOne'}
+
+    def get(self, request, vimid="", tenantid=""):
+        logger.debug("Limits--get::> %s" % request.data)
+        try:
+            #get limits first
+            # prepare request resource to vim instance
+            req_resouce = "/limits"
+            vim = VimDriverUtils.get_vim_info(vimid)
+            sess = VimDriverUtils.get_session(vim, tenantid)
+            resp = sess.get(req_resouce, endpoint_filter=self.service)
+            content = resp.json()
+            content_all =content['limits']['absolute']
+
+            vim_dict = {
+                "vimName": vim["name"],
+                "vimId": vim["vimId"],
+                "tenantId": tenantid,
+            }
+            content_all.update(vim_dict)
+
+            #now get quota
+            # prepare request resource to vim instance
+            req_resouce = "/v2.0/quotas/%s" % tenantid
+            resp = sess.get(req_resouce, endpoint_filter=self.service_network)
+            content = resp.json()
+            content_all.update(content['quota'])
+
+            return Response(data=content_all, status=resp.status_code)
+        except VimDriverNewtonException as e:
+            return Response(data={'error': e.content}, status=e.status_code)
+        except Exception as e:
+            return Response(data={'error': str(e)},
+                            status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+