[Bamboo] Enable/disable elastic bamboo configuration (atlassian-api#1116

)
sebastiangollob · Jan 30, 2023 · 588ccae · 588ccae
1 parent e704bb7
commit 588ccae
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/atlassian/bamboo.py b/atlassian/bamboo.py
@@ -1200,6 +1200,22 @@ def delete_elastic_configuration(self, configuration_id):
         resource = "elasticConfiguration/{configuration_id}".format(configuration_id=configuration_id)
         return self.delete(self.resource_url(resource))
 
+    def get_elastic_bamboo(self):
+        """
+        Get elastic bamboo configuration
+        :return:
+        """
+        response = self.get("rest/admin/latest/elastic/config")
+        return response
+
+    def set_elastic_bamboo(self, data):
+        """
+        Set elastic bamboo configuration
+        :return:
+        """
+        response = self.put("rest/admin/latest/elastic/config", data=data)
+        return response
+
     def get_plugins_info(self):
         """
         Provide plugins info

diff --git a/docs/bamboo.rst b/docs/bamboo.rst
@@ -276,3 +276,12 @@ Elastic Bamboo
     # Delete elastic bamboo configuration
     delete_elastic_configuration('123456')
 
+    # Get elastic bamboo configuration
+    get_elastic_bamboo()
+
+    # Set elastic bamboo configuration
+    set_elastic_bamboo({"enabled": True, "awsCredentialsType": "INSTANCE_PROFILE", "region": "ASIA_PACIFIC_SE_2",
+    "privateKeyFile": "", "certificateFile": "", "maxNumOfElasticInstances": 1, "allocatePublicIpToVpcInstances": False,
+    "elasticInstanceManagement": {"type": "Disabled"}, "uploadAwsAccountIdentifierToElasticInstances": False,
+    "elasticAutoTermination": { "enabled": True, "shutdownDelay": 300}})
+