auto codegen for UK8S

ucloud-bot · web-flow · commit 06237deb5c70 · 2022-01-20T19:30:03.000+08:00
diff --git a/ucloud/services/uk8s/client.py b/ucloud/services/uk8s/client.py
@@ -53,6 +53,51 @@ def add_uk8s_existing_uhost(
         resp = self.invoke("AddUK8SExistingUHost", d, **kwargs)
         return apis.AddUK8SExistingUHostResponseSchema().loads(resp)
 
+    def add_uk8s_node_group(
+        self, req: typing.Optional[dict] = None, **kwargs
+    ) -> dict:
+        """AddUK8SNodeGroup - 添加UK8S节点池
+
+        **Request**
+
+        - **ProjectId** (str) - (Config) 项目ID。不填写为默认项目，子帐号必须填写。 请参考 `GetProjectList接口 <https://docs.ucloud.cn/api/summary/get_project_list>`_
+        - **Region** (str) - (Config) 地域。 参见  `地域和可用区列表 <https://docs.ucloud.cn/api/summary/regionlist>`_
+        - **ClusterId** (str) - (Required) 集群ID
+        - **NodeGroupName** (str) - (Required) 节点池名字
+        - **BootDiskType** (str) - 磁盘类型
+        - **CPU** (int) - GPU卡核心数。仅GPU机型支持此字段（可选范围与MachineType+GpuType相关）
+        - **ChargeType** (str) - 计费模式
+        - **DataDiskSize** (int) - 数据磁盘大小
+        - **DataDiskType** (str) - 磁盘类型
+        - **GPU** (int) - GPU卡核心数
+        - **GpuType** (str) - GPU类型
+        - **ImageId** (str) - 镜像ID
+        - **MachineType** (str) - 云主机机型。枚举值["N", "C", "G", "O", "OS"]。参考 `云主机机型说明 <https://docs.ucloud.cn/api/uhost-api/uhost_type>`_ 。
+        - **Mem** (int) - 内存大小。单位：MB
+        - **MinimalCpuPlatform** (str) - 最低cpu平台，枚举值["Intel/Auto", "Intel/IvyBridge", "Intel/Haswell", "Intel/Broadwell", "Intel/Skylake", "Intel/Cascadelake"；"Intel/CascadelakeR"; “Amd/Epyc2”,"Amd/Auto"],默认值是"Intel/Auto"
+        - **Tag** (str) - 业务组
+        - **Zone** (str) - 可用区。参见  `可用区列表 <https://docs.ucloud.cn/api/summary/regionlist>`_
+
+        **Response**
+
+        - **Message** (str) - 返回错误消息，当 RetCode 非 0 时提供详细的描述信息。
+        - **NodeGroupId** (str) - 节点池ID
+
+        """
+        # build request
+        d = {
+            "ProjectId": self.config.project_id,
+            "Region": self.config.region,
+        }
+        req and d.update(req)
+        d = apis.AddUK8SNodeGroupRequestSchema().dumps(d)
+
+        # build options
+        kwargs["max_retries"] = 0  # ignore retry when api is not idempotent
+
+        resp = self.invoke("AddUK8SNodeGroup", d, **kwargs)
+        return apis.AddUK8SNodeGroupResponseSchema().loads(resp)
+
     def add_uk8s_phost_node(
         self, req: typing.Optional[dict] = None, **kwargs
     ) -> dict:
@@ -320,6 +365,21 @@ def describe_uk8s_cluster(
 
         **Response Model**
 
+        **UhostInfo**
+        - **CPU** (int) - Cpu数量
+        - **CreateTime** (int) - 创建时间
+        - **DiskSet** (list) - 见 **DiskSet** 模型定义
+        - **ExpireTime** (int) - 到期时间
+        - **IPSet** (list) - 见 **IPSet** 模型定义
+        - **Memory** (int) - 内存
+        - **Name** (str) - 主机名称
+        - **NodeId** (str) - 主机ID
+        - **NodeType** (str) - 节点类型：uhost表示云主机;uphost表示物理云主机
+        - **OsName** (str) - 镜像信息
+        - **State** (str) - 主机状态
+        - **Zone** (str) - 所在机房
+
+
         **DiskSet**
         - **BackupType** (str) - 备份方案，枚举类型：BASIC_SNAPSHOT,普通快照；DATAARK,方舟。无快照则不返回该字段。
         - **DiskId** (str) - 磁盘长ID
@@ -341,21 +401,6 @@ def describe_uk8s_cluster(
         - **Type** (str) - 国际: Internation，BGP: Bgp，内网: Private
 
 
-        **UhostInfo**
-        - **CPU** (int) - Cpu数量
-        - **CreateTime** (int) - 创建时间
-        - **DiskSet** (list) - 见 **DiskSet** 模型定义
-        - **ExpireTime** (int) - 到期时间
-        - **IPSet** (list) - 见 **IPSet** 模型定义
-        - **Memory** (int) - 内存
-        - **Name** (str) - 主机名称
-        - **NodeId** (str) - 主机ID
-        - **NodeType** (str) - 节点类型：uhost表示云主机;uphost表示物理云主机
-        - **OsName** (str) - 镜像信息
-        - **State** (str) - 主机状态
-        - **Zone** (str) - 所在机房
-
-
         """
         # build request
         d = {
@@ -590,3 +635,79 @@ def list_uk8s_cluster_v2(
 
         resp = self.invoke("ListUK8SClusterV2", d, **kwargs)
         return apis.ListUK8SClusterV2ResponseSchema().loads(resp)
+
+    def list_uk8s_node_group(
+        self, req: typing.Optional[dict] = None, **kwargs
+    ) -> dict:
+        """ListUK8SNodeGroup - 列出UK8S节点池
+
+        **Request**
+
+        - **ProjectId** (str) - (Config) 项目ID。不填写为默认项目，子帐号必须填写。 请参考 `GetProjectList接口 <https://docs.ucloud.cn/api/summary/get_project_list>`_
+        - **Region** (str) - (Config) 地域。 参见  `地域和可用区列表 <https://docs.ucloud.cn/api/summary/regionlist>`_
+        - **ClusterId** (str) - (Required) 集群ID
+        - **Zone** (str) - 可用区。参见  `可用区列表 <https://docs.ucloud.cn/api/summary/regionlist>`_
+
+        **Response**
+
+        - **NodeGroupList** (list) - 见 **NodeGroupSet** 模型定义
+
+        **Response Model**
+
+        **NodeGroupSet**
+        - **BootDiskType** (str) - 系统盘类型
+        - **CPU** (int) - 虚拟CPU核数
+        - **ChargeType** (str) - 付费方式
+        - **DataDiskSize** (int) - 数据盘大小
+        - **DataDiskType** (str) - 数据盘类型
+        - **GPU** (int) - GPU卡核心数
+        - **GpuType** (str) - GPU类型
+        - **ImageId** (str) - 镜像ID
+        - **MachineType** (str) - 机型
+        - **Mem** (int) - 内存大小
+        - **MinimalCpuPlatform** (str) - cpu平台
+        - **NodeGroupId** (str) - 节点池ID
+        - **NodeGroupName** (str) - 节点池名字
+        - **NodeList** (list) - 节点id列表
+        - **Tag** (str) - 业务组
+
+
+        """
+        # build request
+        d = {
+            "ProjectId": self.config.project_id,
+            "Region": self.config.region,
+        }
+        req and d.update(req)
+        d = apis.ListUK8SNodeGroupRequestSchema().dumps(d)
+
+        resp = self.invoke("ListUK8SNodeGroup", d, **kwargs)
+        return apis.ListUK8SNodeGroupResponseSchema().loads(resp)
+
+    def remove_uk8s_node_group(
+        self, req: typing.Optional[dict] = None, **kwargs
+    ) -> dict:
+        """RemoveUK8SNodeGroup - 删除UK8S节点池
+
+        **Request**
+
+        - **ProjectId** (str) - (Config) 项目ID。不填写为默认项目，子帐号必须填写。 请参考 `GetProjectList接口 <https://docs.ucloud.cn/api/summary/get_project_list>`_
+        - **Region** (str) - (Config) 地域。 参见  `地域和可用区列表 <https://docs.ucloud.cn/api/summary/regionlist>`_
+        - **ClusterId** (str) - (Required) 集群id
+        - **NodeGroupId** (str) - (Required) 节点池Id
+        - **Zone** (str) - 可用区。参见  `可用区列表 <https://docs.ucloud.cn/api/summary/regionlist>`_
+
+        **Response**
+
+
+        """
+        # build request
+        d = {
+            "ProjectId": self.config.project_id,
+            "Region": self.config.region,
+        }
+        req and d.update(req)
+        d = apis.RemoveUK8SNodeGroupRequestSchema().dumps(d)
+
+        resp = self.invoke("RemoveUK8SNodeGroup", d, **kwargs)
+        return apis.RemoveUK8SNodeGroupResponseSchema().loads(resp)
diff --git a/ucloud/services/uk8s/schemas/apis.py b/ucloud/services/uk8s/schemas/apis.py
@@ -45,6 +45,48 @@ class AddUK8SExistingUHostResponseSchema(schema.ResponseSchema):
     }
 
 
+"""
+API: AddUK8SNodeGroup
+
+添加UK8S节点池
+"""
+
+
+class AddUK8SNodeGroupRequestSchema(schema.RequestSchema):
+    """AddUK8SNodeGroup - 添加UK8S节点池"""
+
+    fields = {
+        "BootDiskType": fields.Str(required=False, dump_to="BootDiskType"),
+        "CPU": fields.Int(required=False, dump_to="CPU"),
+        "ChargeType": fields.Str(required=False, dump_to="ChargeType"),
+        "ClusterId": fields.Str(required=True, dump_to="ClusterId"),
+        "DataDiskSize": fields.Int(required=False, dump_to="DataDiskSize"),
+        "DataDiskType": fields.Str(required=False, dump_to="DataDiskType"),
+        "GPU": fields.Int(required=False, dump_to="GPU"),
+        "GpuType": fields.Str(required=False, dump_to="GpuType"),
+        "ImageId": fields.Str(required=False, dump_to="ImageId"),
+        "MachineType": fields.Str(required=False, dump_to="MachineType"),
+        "Mem": fields.Int(required=False, dump_to="Mem"),
+        "MinimalCpuPlatform": fields.Str(
+            required=False, dump_to="MinimalCpuPlatform"
+        ),
+        "NodeGroupName": fields.Str(required=True, dump_to="NodeGroupName"),
+        "ProjectId": fields.Str(required=False, dump_to="ProjectId"),
+        "Region": fields.Str(required=True, dump_to="Region"),
+        "Tag": fields.Str(required=False, dump_to="Tag"),
+        "Zone": fields.Str(required=False, dump_to="Zone"),
+    }
+
+
+class AddUK8SNodeGroupResponseSchema(schema.ResponseSchema):
+    """AddUK8SNodeGroup - 添加UK8S节点池"""
+
+    fields = {
+        "Message": fields.Str(required=True, load_from="Message"),
+        "NodeGroupId": fields.Str(required=True, load_from="NodeGroupId"),
+    }
+
+
 """
 API: AddUK8SPHostNode
 
@@ -514,3 +556,58 @@ class ListUK8SClusterV2ResponseSchema(schema.ResponseSchema):
             models.ClusterSetSchema(), required=False, load_from="ClusterSet"
         ),
     }
+
+
+"""
+API: ListUK8SNodeGroup
+
+列出UK8S节点池
+"""
+
+
+class ListUK8SNodeGroupRequestSchema(schema.RequestSchema):
+    """ListUK8SNodeGroup - 列出UK8S节点池"""
+
+    fields = {
+        "ClusterId": fields.Str(required=True, dump_to="ClusterId"),
+        "ProjectId": fields.Str(required=False, dump_to="ProjectId"),
+        "Region": fields.Str(required=True, dump_to="Region"),
+        "Zone": fields.Str(required=False, dump_to="Zone"),
+    }
+
+
+class ListUK8SNodeGroupResponseSchema(schema.ResponseSchema):
+    """ListUK8SNodeGroup - 列出UK8S节点池"""
+
+    fields = {
+        "NodeGroupList": fields.List(
+            models.NodeGroupSetSchema(),
+            required=False,
+            load_from="NodeGroupList",
+        ),
+    }
+
+
+"""
+API: RemoveUK8SNodeGroup
+
+删除UK8S节点池
+"""
+
+
+class RemoveUK8SNodeGroupRequestSchema(schema.RequestSchema):
+    """RemoveUK8SNodeGroup - 删除UK8S节点池"""
+
+    fields = {
+        "ClusterId": fields.Str(required=True, dump_to="ClusterId"),
+        "NodeGroupId": fields.Str(required=True, dump_to="NodeGroupId"),
+        "ProjectId": fields.Str(required=False, dump_to="ProjectId"),
+        "Region": fields.Str(required=True, dump_to="Region"),
+        "Zone": fields.Str(required=False, dump_to="Zone"),
+    }
+
+
+class RemoveUK8SNodeGroupResponseSchema(schema.ResponseSchema):
+    """RemoveUK8SNodeGroup - 删除UK8S节点池"""
+
+    fields = {}
diff --git a/ucloud/services/uk8s/schemas/models.py b/ucloud/services/uk8s/schemas/models.py
@@ -3,18 +3,6 @@
 from ucloud.core.typesystem import schema, fields
 
 
-class IPSetSchema(schema.ResponseSchema):
-    """IPSet - 节点的IP信息"""
-
-    fields = {
-        "Bandwidth": fields.Int(required=False, load_from="Bandwidth"),
-        "Default": fields.Str(required=False, load_from="Default"),
-        "IP": fields.Str(required=False, load_from="IP"),
-        "IPId": fields.Str(required=False, load_from="IPId"),
-        "Type": fields.Str(required=False, load_from="Type"),
-    }
-
-
 class DiskSetSchema(schema.ResponseSchema):
     """DiskSet - 节点磁盘信息"""
 
@@ -32,6 +20,18 @@ class DiskSetSchema(schema.ResponseSchema):
     }
 
 
+class IPSetSchema(schema.ResponseSchema):
+    """IPSet - 节点的IP信息"""
+
+    fields = {
+        "Bandwidth": fields.Int(required=False, load_from="Bandwidth"),
+        "Default": fields.Str(required=False, load_from="Default"),
+        "IP": fields.Str(required=False, load_from="IP"),
+        "IPId": fields.Str(required=False, load_from="IPId"),
+        "Type": fields.Str(required=False, load_from="Type"),
+    }
+
+
 class UhostInfoSchema(schema.ResponseSchema):
     """UhostInfo - 机器信息"""
 
@@ -77,14 +77,6 @@ class K8SNodeConditionSchema(schema.ResponseSchema):
     }
 
 
-class KubeProxySchema(schema.ResponseSchema):
-    """KubeProxy - KubeProxy信息"""
-
-    fields = {
-        "Mode": fields.Str(required=False, load_from="Mode"),
-    }
-
-
 class UHostIPSetSchema(schema.ResponseSchema):
     """UHostIPSet - 云主机IP信息"""
 
@@ -102,6 +94,14 @@ class UHostIPSetSchema(schema.ResponseSchema):
     }
 
 
+class KubeProxySchema(schema.ResponseSchema):
+    """KubeProxy - KubeProxy信息"""
+
+    fields = {
+        "Mode": fields.Str(required=False, load_from="Mode"),
+    }
+
+
 class NodeInfoV2Schema(schema.ResponseSchema):
     """NodeInfoV2 - UK8S 节点信息"""
 
@@ -152,3 +152,27 @@ class ClusterSetSchema(schema.ResponseSchema):
         "SubnetId": fields.Str(required=True, load_from="SubnetId"),
         "VPCId": fields.Str(required=True, load_from="VPCId"),
     }
+
+
+class NodeGroupSetSchema(schema.ResponseSchema):
+    """NodeGroupSet - 节点池配置"""
+
+    fields = {
+        "BootDiskType": fields.Str(required=False, load_from="BootDiskType"),
+        "CPU": fields.Int(required=False, load_from="CPU"),
+        "ChargeType": fields.Str(required=False, load_from="ChargeType"),
+        "DataDiskSize": fields.Int(required=False, load_from="DataDiskSize"),
+        "DataDiskType": fields.Str(required=False, load_from="DataDiskType"),
+        "GPU": fields.Int(required=False, load_from="GPU"),
+        "GpuType": fields.Str(required=False, load_from="GpuType"),
+        "ImageId": fields.Str(required=False, load_from="ImageId"),
+        "MachineType": fields.Str(required=False, load_from="MachineType"),
+        "Mem": fields.Int(required=False, load_from="Mem"),
+        "MinimalCpuPlatform": fields.Str(
+            required=False, load_from="MinimalCpuPlatform"
+        ),
+        "NodeGroupId": fields.Str(required=False, load_from="NodeGroupId"),
+        "NodeGroupName": fields.Str(required=False, load_from="NodeGroupName"),
+        "NodeList": fields.List(fields.Str()),
+        "Tag": fields.Str(required=False, load_from="Tag"),
+    }
