Use openstack proxy for kms service (#127)

Use openstack proxy for kms service

Switch kms and obs to using openstack proxy instead of openstack_sdk proxy

Reviewed-by: None <None>
Reviewed-by: Artem Goncharov <[email protected]>

Author: Polina-Gubina

doc/source/sdk/guides/kms.rst

@@ -143,7 +143,7 @@ Encrypt Datakey
 This interface is used to encrypt a KMS data encryption key.
 
 .. literalinclude:: ../examples/kms/encrypt_datakey.py
-   :lines: 16-27
+   :lines: 16-25
 
 Decrypt Datakey
 ^^^^^^^^^^^^^^^

examples/kms/create_datakey.py

@@ -20,7 +20,7 @@
 conn = openstack.connect(cloud='otc')
 
 dek = conn.kms.create_datakey(
-    key_id='cmk_id',
+    cmk='cmk_id',
     datakey_length='512'
 )
 print(dek)

examples/kms/create_datakey_without_plaintext.py

@@ -20,7 +20,7 @@
 conn = openstack.connect(cloud='otc')
 
 dek = conn.kms.create_datakey_wo_plain(
-    key_id='cmk_id',
+    cmk='cmk_id',
     datakey_length='512'
 )
 print(dek)

examples/kms/decrypt_datakey.py

@@ -20,7 +20,7 @@
 conn = openstack.connect(cloud='otc')
 
 data = conn.kms.decrypt_datakey(
-    key_id='cmk_id',
+    cmk='cmk_id',
     cypher_text='64_bit_cypher',
     datakey_cypher_length='64'
 )

examples/kms/encrypt_datakey.py

@@ -20,8 +20,6 @@
 conn = openstack.connect(cloud='otc')
 
 data = conn.kms.encrypt_datakey(
-    key_id='cmk_id',
-    plain_text='64_bit_key',
-    datakey_plain_length='64'
+    datakey='datakey_id'
 )
 print(data)

otcextensions/sdk/kms/kms_service.py

@@ -11,12 +11,12 @@
 # under the License.
 from openstack import service_description
 
-from otcextensions.sdk.kms.v1 import _proxy
+from otcextensions.sdk.kms.v1 import _proxy as _proxy_v1
 
 
 class KmsService(service_description.ServiceDescription):
     """The KMS service."""
 
     supported_versions = {
-        '1': _proxy.Proxy
+        '1': _proxy_v1.Proxy
     }

otcextensions/sdk/kms/v1/_base.py

@@ -15,41 +15,37 @@
 from openstack import utils
 
 from otcextensions.i18n import _
-from otcextensions.sdk import sdk_resource
 
 _logger = _log.setup_logging('openstack')
 
 
-class Resource(sdk_resource.Resource):
+class Resource(resource.Resource):
     base_path = '/kms'
 
     #: Error code when create a secret key
     error_code = resource.Body('error_code')
     #: Error message when create a secret key
     error_msg = resource.Body('error_msg')
 
-    def create(self, session, prepend_key=True, requires_id=True,
-               endpoint_override=None, headers=None, uri=None):
+    def create(self, session, prepend_key=False, uri=None,
+               requires_id=True, **params):
         if not self.allow_create:
             raise exceptions.MethodNotSupported(self, "create")
 
         session = self._get_session(session)
 
         request = self._prepare_request(requires_id=False,
-                                        prepend_key=prepend_key)
+                                        prepend_key=prepend_key
+                                        )
         # PATH is different
         if uri:
             request.url = uri
         elif self.create_path:
             request.url = self.create_path
 
-        req_args = self._prepare_override_args(
-            endpoint_override=endpoint_override,
-            request_headers=request.headers,
-            additional_headers=headers)
-
         response = session.post(request.url,
-                                json=request.body, **req_args)
+                                json=request.body, headers=request.headers,
+                                params=params)
 
         if response.status_code == 400:
             body = response.json()
@@ -69,7 +65,7 @@ def create(self, session, prepend_key=True, requires_id=True,
         return self
 
     def _action(self, session, url_part, body):
-        """Preform actions given the message body.
+        """Perform actions given the message body.
         """
         url = utils.urljoin(self.base_path, url_part)
         response = session.post(

otcextensions/sdk/kms/v1/_proxy.py

@@ -9,16 +9,23 @@
 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 # License for the specific language governing permissions and limitations
 # under the License.
-from otcextensions.sdk import sdk_proxy
+from openstack import proxy
 from otcextensions.sdk.kms.v1 import data_key as _data_key
 from otcextensions.sdk.kms.v1 import key as _key
 from otcextensions.sdk.kms.v1 import misc as _misc
 
 
-class Proxy(sdk_proxy.Proxy):
+class Proxy(proxy.Proxy):
 
     skip_discovery = True
 
+    def __init__(self, session, *args, **kwargs):
+        super(Proxy, self).__init__(session=session, *args, **kwargs)
+        self.additional_headers = {
+            'Content-Type': 'application/json',
+            'X-Language': 'en-us'
+        }
+
     # ======== CMK Keys ========
     def keys(self, **query):
         """List all master keys.
@@ -233,7 +240,7 @@ def get_instance_number(self):
             :class:`~otcextensions.sdk.kms.v1.key.InstanceNumber`
         """
         instance_num_obj = _misc.InstanceNumber()
-        return instance_num_obj.get(self)
+        return instance_num_obj.fetch(self)
 
     def quotas(self):
         """List quota resources for KMS service

otcextensions/sdk/kms/v1/data_key.py

@@ -32,6 +32,7 @@ class DataKey(_base.Resource):
 
     allow_create = True
     allow_update = True
+    allow_fetch = False
 
     # Properties
     #: Secret key (CMK) ID
@@ -60,12 +61,11 @@ class DataKey(_base.Resource):
     #: *Type:str*
     cipher_text = resource.Body('cipher_text')
 
-    def create_wo_plain(self, session, prepend_key=True, requires_id=True,
-                        endpoint_override=None, headers=None):
+    def create_wo_plain(self, session, prepend_key=True, requires_id=True):
         return super(DataKey, self).create(
-            session, prepend_key=prepend_key, requires_id=requires_id,
-            endpoint_override=endpoint_override, headers=headers,
-            uri='/kms/create-datakey-without-plaintext')
+            session, prepend_key=prepend_key,
+            uri='/kms/create-datakey-without-plaintext',
+            requires_id=requires_id)
 
     def encrypt(self, session):
         """Encrypt DEK

otcextensions/sdk/kms/v1/key.py

@@ -24,8 +24,9 @@ class Key(_base.Resource):
     resource_key = 'key_info'
 
     allow_list = True
-    allow_get = True
     allow_create = True
+    allow_update = False
+    allow_fetch = True
 
     # Properties
     #: Secret key ID
@@ -55,8 +56,8 @@ class Key(_base.Resource):
     #: Error message when create a secret key
     error_msg = resource.Body('error_msg')
 
-    def get(self, session, error_message=None, requires_id=True,
-            endpoint_override=None, headers=None):
+    def fetch(self, session, requires_id=None,
+              base_path=None, error_message=None):
         if not self.allow_get:
             raise exceptions.MethodNotSupported(self, "get")
         url = self.get_path
@@ -93,8 +94,7 @@ def cancel_deletion(self, session):
         )
 
     @classmethod
-    def list(cls, session, paginated=True,
-             endpoint_override=None, headers=None, **kwargs):
+    def list(cls, session, paginated=True, base_path=None, **kwargs):
 
         if not cls.allow_list:
             raise exceptions.MethodNotSupported(cls, "list")
@@ -108,25 +108,21 @@ def list(cls, session, paginated=True,
         body = {}
         limit = None
         if 'limit' in kwargs:
-            limit = kwargs['limit']
+            limit = kwargs.pop('limit')
             body['limit'] = limit
         if 'marker' in kwargs:
-            body['marker'] = kwargs['marker']
+            body['marker'] = kwargs.pop('marker')
         if 'key_state' in kwargs:
-            body['key_state'] = kwargs['key_state']
+            body['key_state'] = kwargs.pop('key_state')
         if 'sequence' in kwargs:
-            body['sequence'] = kwargs['sequence']
+            body['sequence'] = kwargs.pop('sequence')
 
         total_yielded = 0
         while uri:
 
             session = cls._get_session(session)
 
-            args = cls._prepare_override_args(
-                endpoint_override=endpoint_override,
-                additional_headers=headers)
-
-            response = session.post(uri, json=body, **args)
+            response = session.post(uri, json=body, **kwargs)
 
             data = response.json()