wip: First Draft of news Request Version

slapos/recipe/request.py

@@ -30,6 +30,7 @@ from slapos.recipe.librecipe import wrap, JSON_SERIALISED_MAGIC_KEY
 import json
 from slapos import slap as slapmodule
 from slapos.slap import SoftwareProductCollection
+from slapos.slap.slap import json_loads_byteified
 import slapos.recipe.librecipe.generic as librecipe
 import traceback
 
@@ -137,6 +138,7 @@ class Recipe(object):
       options.get('key-file'),
       options.get('cert-file'),
     )
+    if not slap.jio_api_connector:
       request = slap.registerComputerPartition(
         options['computer-id'],
         options['partition-id'],
@@ -158,6 +160,56 @@ class Recipe(object):
     self._raise_request_exception_formatted = None
     self.instance = None
 
+    if slap.jio_api_connector:
+      request_dict = {
+        "title": name,
+        "software_type": software_type,
+        "software_release_uri": software_url,
+        "portal_type": "Software Instance",
+      }
+      if partition_parameter_kw:
+        request_dict["parameters"] = json.dumps(partition_parameter_kw)
+      if filter_kw:
+        request_dict["sla_parameters"] = filter_kw
+      if slave:
+        request_dict["shared"] = True
+      if requested_state:
+        request_dict["state"] = requested_state
+
+      partition_dict = slap.jio_api_connector.post(request_dict)
+      if "$schema" in partition_dict and "error-response-schema.json" in partition_dict["$schema"]:
+        self.logger.warning(
+          'Request for %(request_name)r with software release '
+          '%(software_release)r and software type %(software_type)r failed '
+          'with partition_parameter_kw=%(partition_parameter_kw)r, '
+          'filter_kw=%(filter_kw)r, shared=%(shared)r, state=%(state)r.', dict(
+            software_release=software_url,
+            software_type=software_type,
+            request_name=name,
+            partition_parameter_kw=partition_parameter_kw,
+            filter_kw=filter_kw,
+            shared=slave,
+            state=requested_state
+          )
+        )
+        self._raise_request_exception = slapmodule.NotFoundError
+        self._raise_request_exception_formatted = str(partition_dict["message"])
+        return_parameter_dict = {}
+      else:
+        return_parameter_dict =  self._getFilteredParameterDict(
+          partition_dict,
+          return_parameters
+        )
+        # Fetch the instance-guid and the instance-state
+        # Note: SlapOS Master does not support it for slave instances
+        if not slave:
+          options['instance-guid'] =  partition_dict["reference"]
+          # XXX: deprecated, to be removed
+          options['instance_guid'] = options['instance-guid']
+          options['instance-state'] = options['state']
+          options['instance-status'] = options.get('access_status_message')
+
+    else:
       # Try to do the request and fetch parameter dict...
       try:
         self.instance = request(software_url, software_type,
@@ -221,6 +273,14 @@ class Recipe(object):
         pass
     return result
 
+  def _getFilteredParameterDict(self, partition_dict, return_parameter_list):
+    result = {}
+    parameters = json_loads_byteified(partition.get("parameters", "{}"))
+    for key in return_parameter_list:
+      if key in parameters:
+        result[key] = parameters["key"]
+    return result
+
   def install(self):
     if self._raise_request_exception:
       raise self._raise_request_exception
@@ -286,6 +346,12 @@ class JSONCodec(object):
     except slapmodule.NotFoundError:
       return {}
 
+  def _getFilteredParameterDict(self, partition_dict, return_parameter_list):
+    parameters = json_loads_byteified(partition.get("parameters", "{}"))
+    if JSON_SERIALISED_MAGIC_KEY in parameters:
+      return json.loads(parameters[JSON_SERIALISED_MAGIC_KEY])
+    return {}
+
 class RequestJSONEncoded(JSONCodec, Recipe):
   """
   Like Recipe, but serialised with JSONCodec.

slapos/test/recipe/test_request.py

@@ -2,6 +2,7 @@ import mock
 import unittest
 from collections import defaultdict
 from slapos.recipe import request
+from test_slaposconfiguration import APIRequestHandler
 from testfixtures import LogCapture
 
 
@@ -25,6 +26,7 @@ class RecipeTestMixin(object):
     self.request_instance.return_value = requested_instance
     register_instance.request = self.request_instance
     slap_instance.registerComputerPartition.return_value = register_instance
+    slap_instance.jio_api_connector = None
     slap.return_value = slap_instance
     self.instance_getConnectionParameter = \
         requested_instance.getConnectionParameter
@@ -149,3 +151,6 @@ class RequestOptionalJSONEncodedTest(RecipeTestMixin, unittest.TestCase):
   return_value = '{"anything": "done"}'
   raises = False
   called_partition_parameter_kw = {'_': '{}'}
+
+class RecipejIOTestMixin:
+  pass
\ No newline at end of file