Update wrappers for PvC - cdpcli 0.9.105

Update wrappers for PvC - cdpcli 0.9.105
- add drscp.py for PvC DRS Backup/Restore
- add metrics.py for PvC Metrics Export
- update environments.py for 3 new PvC functions

Signed-off-by: Chuck Levesque <clevesque@cloudera.com>

Author: clevesque

src/cdpy/cdpy.py

@@ -14,6 +14,8 @@
 from cdpy.opdb import CdpyOpdb
 from cdpy.dw import CdpyDw
 from cdpy.df import CdpyDf
+from cdpy.drscp import CdpyDrscp
+from cdpy.metrics import CdpyMetrics
 
 
 class Cdpy(CdpSdkBase):
@@ -30,3 +32,5 @@ def __init__(self, *args, **kwargs):
         self.dw = CdpyDw(*args, **kwargs)
         self.df = CdpyDf(*args, **kwargs)
         self.de = CdpyDe(*args, **kwargs)
+        self.drscp = CdpyDrscp(*args, **kwargs)
+        self.metrics = CdpyMetrics(*args, **kwargs)

src/cdpy/drscp.py

@@ -0,0 +1,93 @@
+# -*- coding: utf-8 -*-
+
+from cdpy.common import CdpSdkBase, Squelch, CdpcliWrapper
+
+ENTITLEMENT_DISABLED='DRS not enabled on CDP Tenant'
+
+
+class CdpyDrscp(CdpSdkBase):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def create_backup(self, backup_name=None, item_name=None ):
+        return self.sdk.call(
+            svc='drscp', func='create_backup', ret_field='backupCrn', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            backupName=backup_name,
+            itemName=item_name
+        )
+
+    def delete_backup(self, backup_crn):
+        return self.sdk.call(
+            svc='drscp', func='delete_backup', ret_field='deleteBackupCrn', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            backupCrn=backup_crn
+        )
+
+    def describe_backup(self, backup_crn):
+        return self.sdk.call(
+            svc='drscp', func='describe_backup', ret_field='backup', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            backupCrn=backup_crn
+        )
+
+    def describe_restore(self, restore_crn):
+        return self.sdk.call(
+            svc='drscp', func='describe_restore', ret_field='restore', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            restoreCrn=restore_crn
+        )
+
+    def get_logs(self, crn ):
+        return self.sdk.call(
+            svc='drscp', func='get_logs', ret_field='logs', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            crn=crn
+        )
+
+    def list_backup_entities(self):
+        return self.sdk.call(
+            svc='drscp', func='list_backup_entities', ret_field='items', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ]
+        )
+
+    def list_backups(self, backup_name=None, job_states=None):
+        return self.sdk.call(
+            svc='drscp', func='list_backups', ret_field='backup', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            backupName=backup_name,
+            jobStates=job_states
+        )
+
+    def list_restores(self, job_states=None, backup_crn=None):
+        return self.sdk.call(
+            svc='drscp', func='list_restores', ret_field='restores', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            jobStates=job_states,
+            backupCrn=backup_crn
+        )
+
+    def restore_backup(self, backup_crn ):
+        return self.sdk.call(
+            svc='drscp', func='restore_backup', ret_field='restoreCrn', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            backupCrn=backup_crn
+        )

src/cdpy/environments.py

@@ -140,6 +140,35 @@ def create_gcp_environment(self, **kwargs):
             self.sdk.throw_error(resp)
         return resp
 
+    def create_private_environment(self, env_name, address, user, authentication_token, cluster_names,
+                                  kube_config=None, authentication_token_type="CLEARTEXT_PASSWORD", 
+                                  namespace_prefix=None, domain=None, platform=None, docker_config_json=None, 
+                                  docker_user_pass=None, description=None, storage_class=None):
+        resp = self.sdk.call(
+            svc='environments', func='create_private_environment', ret_field='environment', ret_error=True,
+            squelch=[
+                Squelch(value='NOT_FOUND', default=list(),
+                        warning='No Workspaces found in Tenant'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED,
+                        default=list())
+            ],
+            environmentName=env_name,
+            address=address,
+            user=user,
+            authenticationToken=authentication_token,
+            clusterNames=cluster_names,
+            kubeConfig=kube_config,
+            authenticationTokenType=authentication_token_type,
+            namespacePrefix=namespace_prefix,
+            domain=domain,
+            platform=platform,
+            dockerConfigJson=docker_config_json,
+            dockerUserPass=docker_user_pass,
+            description=description,
+            storageClass=storage_class
+        )
+
+
     def stop_environment(self, name):
         return self.sdk.call(
             svc='environments', func='stop_environment', ret_field='environment', squelch=[
@@ -336,3 +365,36 @@ def resolve_environment_crn(self, env: Union[str, None]):
                 return env_desc['crn'] if env_desc else None
         else:
             return None
+            
+    def check_database_connectivity(self, host, port, name, username, password):
+        return self.sdk.call(
+            svc='environments', func='check_database_connectivity',
+            ret_field='result',
+            host=host,
+            port=port,
+            name=name,
+            userName=username,
+            password=password
+        )
+
+    def check_environment_connectivity(self, address, user, authentication_token,
+                      authentication_token_type=None, cluster_names=None):
+        return self.sdk.call(
+            svc='environments', func='check_environment_connectivity',
+            ret_field='clusters',
+            address=address,
+            user=user,
+            authenticationToken=authentication_token,
+            authenticationTokenType=authentication_token_type,
+            clusterNames=cluster_names
+        )
+
+    def check_kubernetes_connectivity(self, kube_config, format=None):
+        return self.sdk.call(
+            svc='environments', func='check_kubernetes_connectivity',
+            ret_field='status',
+            kubeConfig=kube_config,
+            format=format
+        )    
+    
+

src/cdpy/metrics.py

@@ -0,0 +1,52 @@
+from cdpy.common import CdpSdkBase, Squelch
+
+
+class CdpyMetrics(CdpSdkBase):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        
+    def create_remote_write_config(self, remote_write_config=None):
+        return self.sdk.call(
+            svc='metrics', func='create_remote_write_config', ret_field='remoteWriteConfig', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            remoteWriteConfig=remote_write_config
+        )
+
+    def delete_remote_write_config(self, id=None):
+        return self.sdk.call(
+            svc='metrics', func='delete_remote_write_config', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            id=id
+        )
+
+    def describe_remote_write_config(self, id=None):
+        return self.sdk.call(
+            svc='metrics', func='describe_remote_write_config',
+                ret_field='remoteWriteConfig', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            id=id
+        )
+
+    def list_remote_write_configs(self):
+        return self.sdk.call(
+            svc='metrics', func='list_remote_write_configs',
+                ret_field='remoteWriteConfigs', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ]
+        )
+
+    def update_remote_write_config(self, remote_write_config=None):
+        return self.sdk.call(
+            svc='metrics', func='update_remote_write_config', ret_field='remoteWriteConfig', squelch=[
+                Squelch('NOT_FOUND'), Squelch('INVALID_ARGUMENT'),
+                Squelch(value='PATH_DISABLED', warning=ENTITLEMENT_DISABLED)
+            ],
+            remoteWriteConfig=remote_write_config
+        )