feat(k8scluster): add supported cluster types

Author: snowch

hpecp/cli/k8scluster.py

@@ -443,6 +443,41 @@ def upgrade_cluster(
             id, k8s_upgrade_version, worker_upgrade_percent
         )
 
+    @base.intercept_exception
+    def import_cluster(
+        self,
+        cluster_type,
+        name,
+        description,
+        pod_dns_domain,
+        server_url,
+        ca,
+        bearer_token,
+    ):
+        """Import a k8s cluster.
+
+        TODO
+
+        Returns
+        -------
+        TODO
+
+        Raises
+        ------
+        APIException
+        """
+        print(
+            base.get_client().k8s_cluster.import_cluster(
+                cluster_type,
+                name,
+                description,
+                pod_dns_domain,
+                server_url,
+                ca,
+                bearer_token,
+            )
+        )
+
     @base.intercept_exception
     def import_generic_cluster(
         self, name, description, pod_dns_domain, server_url, ca, bearer_token

hpecp/k8s_cluster.py

@@ -586,10 +586,17 @@ def upgrade_cluster(
         )
         return response.json()
 
-    def import_generic_cluster(
-        self, name, description, pod_dns_domain, server_url, ca, bearer_token
+    def import_cluster(
+        self,
+        cluster_type,
+        name,
+        description,
+        pod_dns_domain,
+        server_url,
+        ca,
+        bearer_token,
     ):
-        """Import a generic k8s cluster.
+        """Import a k8s cluster.
 
         TODO
 
@@ -601,10 +608,18 @@ def import_generic_cluster(
         ------
         APIException
         """
+        assert cluster_type in [
+            "generic",
+            "eks",
+            "aks",
+            "gke",
+            "pks",
+        ], 'cluster_type must be either "generic", "eks", "aks", "gke", "pks"'
+
         data = {
             "label": {"name": name, "description": description},
             "pod_dns_domain": pod_dns_domain,
-            "type": "generic",
+            "type": cluster_type,
             "sysadmin_data": {
                 "server_url": server_url,
                 "ca": ca,
@@ -620,6 +635,19 @@ def import_generic_cluster(
         )
         return CaseInsensitiveDict(response.headers)["Location"]
 
+    def import_generic_cluster(
+        self, name, description, pod_dns_domain, server_url, ca, bearer_token
+    ):
+        return self.import_cluster(
+            "generic",
+            name,
+            description,
+            pod_dns_domain,
+            server_url,
+            ca,
+            bearer_token,
+        )
+
     def import_generic_cluster_with_json(self, json):
         """Import a generic k8s cluster.