Support for DFX (#17)

* Add BAD_HEALTH to failed status for DFX
* Correct not found error for DFX Environment to return None instead of an empty list
* Correct spelling of DFX authorisedIpRanges to authorizedIpRanges

Signed-off-by: Daniel Chaffelson <[email protected]>

Author: Chaffelson

src/cdpy/cdpy.py

@@ -12,6 +12,7 @@
 from cdpy.ml import CdpyMl
 from cdpy.opdb import CdpyOpdb
 from cdpy.dw import CdpyDw
+from cdpy.df import CdpyDf
 
 
 class Cdpy(CdpSdkBase):
@@ -25,3 +26,4 @@ def __init__(self, *args, **kwargs):
         self.ml = CdpyMl(*args, **kwargs)
         self.opdb = CdpyOpdb(*args, **kwargs)
         self.dw = CdpyDw(*args, **kwargs)
+        self.df = CdpyDf(*args, **kwargs)

src/cdpy/common.py

@@ -156,22 +156,25 @@ def _warning_format(message, category, filename, lineno, line=None):
             'STACK_CREATION_IN_PROGRESS',
             'CREATION_INITIATED',
             'FREEIPA_CREATION_IN_PROGRESS',
-            'STARTING'
+            'STARTING',
+            'ENABLING'  # DF
         ]
 
         self.TERMINATION_STATES = [
             'EXTERNAL_DATABASE_DELETION_IN_PROGRESS',
             'STACK_DELETION_IN_PROGRESS',
             'FREEIPA_DELETE_IN_PROGRESS',
-            'STOPPING'
+            'STOPPING',
+            'DISABLING'  # DF
         ]
 
         self.STARTED_STATES = [
             'EXTERNAL_DATABASE_START_IN_PROGRESS',
             'AVAILABLE',
             'START_IN_PROGRESS',
             'RUNNING',
-            'Running'
+            'Running',  # DW
+            'GOOD_HEALTH'  # DF
         ]
 
         self.STOPPED_STATES = [
@@ -189,13 +192,15 @@ def _warning_format(message, category, filename, lineno, line=None):
             'TIMEDOUT',
             'DELETE_FAILED',
             'Error',  # DW
-            'installation:failed'  # ML
+            'installation:failed',  # ML
+            'BAD_HEALTH'  # DF
         ]
 
         self.REMOVABLE_STATES = [
             'AVAILABLE', 'UPDATE_FAILED', 'CREATE_FAILED', 'ENABLE_SECURITY_FAILED', 'DELETE_FAILED',
             'DELETE_COMPLETED', 'DELETED_ON_PROVIDER_SIDE', 'STOPPED', 'START_FAILED', 'STOP_FAILED',
-            'Error', 'Running'
+            'Error', 'Running',  # DW
+            'GOOD_HEALTH', 'CONCERNING_HEALTH', 'BAD_HEALTH'  # DF
         ]
 
         # common regex patterns

src/cdpy/df.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+
+from cdpy.common import CdpSdkBase, Squelch
+
+
+class CdpyDf(CdpSdkBase):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def list_environments(self, only_enabled=False):
+        result = self.sdk.call(
+            svc='df', func='list_environments', ret_field='environments', squelch=[
+                Squelch(value='NOT_FOUND', default=list(),
+                        warning='No Environments for DataFlow found in Tenant')
+            ],
+            pageSize=self.sdk.DEFAULT_PAGE_SIZE
+        )
+        if only_enabled:
+            return [x for x in result if x['status']['state'] not in ['NOT_ENABLED']]
+        return result
+
+    def describe_environment(self, env_crn: str):
+        return self.sdk.call(
+            svc='df', func='get_environment', ret_field='environment', squelch=[
+                Squelch(value='NOT_FOUND',
+                        warning='No Environment with crn %s for DataFlow found in Tenant' % env_crn)
+            ],
+            crn=env_crn
+        )
+
+    def enable_environment(self, env_crn: str, authorized_ips: list = None, min_nodes: int = 3, max_nodes: int = 3,
+                           enable_public_ip: bool = True):
+        self.sdk.validate_crn(env_crn)
+        return self.sdk.call(
+            svc='df', func='enable_environment', ret_field='environment',
+            crn=env_crn, minK8sNodeCount=min_nodes, maxK8sNodeCount=max_nodes,
+            usePublicLoadBalancer=enable_public_ip, authorizedIpRanges=authorized_ips
+        )
+
+    def disable_environment(self, env_crn: str, persist: bool = False):
+        self.sdk.validate_crn(env_crn)
+        return self.sdk.call(
+            svc='df', func='disable_environment', ret_field='status',
+            crn=env_crn, persist=persist
+        )