Add start/stop functions for OPDB & DH

cloudera-labs · Oct 19, 2021 · e609c7a · e609c7a
1 parent 8d0e88c
commit e609c7a
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/src/cdpy/datahub.py b/src/cdpy/datahub.py
@@ -72,3 +72,15 @@ def list_cluster_definitions(self):
     def describe_cluster_definition(self, name):
         return self.sdk.call(svc='datahub', func='describe_cluster_definition', ret_field='clusterDefinition', squelch=[
             Squelch(value='NOT_FOUND')], clusterDefinitionName=name)
+
+    def start_cluster(self, name):
+        return self.sdk.call(
+            svc='datahub', func='start_cluster', squelch=[Squelch('NOT_FOUND')],
+            clusterName=name
+        )
+
+    def stop_cluster(self, name):
+        return self.sdk.call(
+            svc='datahub', func='stop_cluster', squelch=[Squelch('NOT_FOUND')],
+            clusterName=name
+        )
diff --git a/src/cdpy/opdb.py b/src/cdpy/opdb.py
@@ -49,3 +49,21 @@ def create_database(self, name, env):
             databaseName=name,
             environmentName=env,
         )
+
+    def start_database(self, name, env):
+        return self.sdk.call(
+            svc='opdb', func='start_database', squelch=[
+                Squelch('NOT_FOUND'),
+            ],
+            databaseName=name,
+            environmentName=env,
+        )
+
+    def stop_database(self, name, env):
+        return self.sdk.call(
+            svc='opdb', func='stop_database', squelch=[
+                Squelch('NOT_FOUND'),
+            ],
+            databaseName=name,
+            environmentName=env,
+        )