yicloud-sdk-python 0.1.0__py3-none-any.whl

yicloud/services/iam/models.py

@@ -0,0 +1,276 @@
+from dataclasses import dataclass, field
+from typing import Optional, List
+
+
+# Enum: Role
+class Role:
+    MANAGER = "manager"
+    OPERATOR = "operator"
+
+
+# Enum: UserType
+class UserType:
+    TENANT_OPERATOR = "tenantOperator"
+    TENANT_MANAGER = "tenantManager"
+    TENANT_OBSERVER = "noPermission"
+
+
+# Enum: AccessKeyStatus
+class AccessKeyStatus:
+    ACTIVE = "Active"
+    INACTIVE = "Inactive"
+
+
+# Enum: StatementType
+class StatementType:
+    CLASSIC_WHITE_LIST = "ClassicWhiteList"
+
+
+# Models
+@dataclass
+class ProjectRole:
+    Name: str  # json:"Name"
+    Role: str  # json:"Role"
+
+
+@dataclass
+class CreateUserReq:
+    Email: str  # json:"Email"
+    FirstName: str  # json:"FirstName"
+    LastName: str  # json:"LastName"
+    Password: str  # json:"Password"
+    UserName: str  # json:"UserName"
+    UserType: str  # json:"UserType"
+    ProjectsRoles: Optional[List[ProjectRole]] = None  # json:"ProjectsRoles,omitempty"
+
+
+@dataclass
+class UserData:
+    CreationTime: str  # json:"CreationTime"
+    DisplayName: str  # json:"DisplayName"
+    Email: str  # json:"Email"
+    UserName: str  # json:"UserName"
+    UserType: str  # json:"UserType"
+    ProjectsRoles: Optional[List[ProjectRole]] = None  # json:"ProjectsRoles,omitempty"
+
+
+@dataclass
+class ListUsersData:
+    Items: Optional[List[UserData]] = None  # json:"Items,omitempty"
+    Limit: int = 10  # json:"Limit,omitempty"
+    Offset: int = 0  # json:"Offset,omitempty"
+    Total: int = 0  # json:"Total,omitempty"
+
+
+@dataclass
+class DeleteUserReq:
+    UserName: str  # json:"UserName"
+
+
+@dataclass
+class CreateAccessKeyReq:
+    UserName: str  # json:"UserName"
+
+
+@dataclass
+class AccessKeyData:
+    AccessKey: str  # json:"AccessKey"
+    CreationTime: str  # json:"CreationTime"
+    SecretKey: str  # json:"SecretKey"
+    Status: str  # json:"Status"
+
+
+@dataclass
+class DeleteAccessKeyReq:
+    AccessKey: str  # json:"AccessKey"
+
+
+@dataclass
+class UpdateAccessKeyReq:
+    AccessKey: str  # json:"AccessKey"
+    Status: str  # json:"Status"
+
+
+@dataclass
+class ListAccessKeyData:
+    Items: Optional[List[AccessKeyData]] = None  # json:"Items,omitempty"
+    Total: int = 0  # json:"Total,omitempty"
+
+
+@dataclass
+class Statement:
+    Type: str  # json:"Type"
+    IPList: List[str]  # json:"IPList"
+
+
+@dataclass
+class AccessKeyPolicyData:
+    Type: str  # json:"Type"
+    IPList: List[str]  # json:"IPList"
+    Status: str  # json:"Status"
+
+
+@dataclass
+class SetAccessKeyPolicyReq:
+    AccessKey: str  # json:"AccessKey"
+    AccessKeyPolicy: Optional[AccessKeyPolicyData] = None  # json:"AccessKeyPolicy"
+
+
+@dataclass
+class CreateProjectReq:
+    ProjectName: str  # json:"ProjectName"
+
+
+@dataclass
+class DeleteProjectReq:
+    ProjectName: str  # json:"ProjectName"
+
+
+@dataclass
+class ProjectData:
+    CreationTime: str  # json:"CreationTime,omitempty"
+    ProjectName: str  # json:"ProjectName,omitempty"
+
+
+@dataclass
+class ListProjectsData:
+    Items: Optional[List[ProjectData]] = None  # json:"Items,omitempty"
+    Limit: int = 10  # json:"Limit,omitempty"
+    Offset: int = 0  # json:"Offset,omitempty"
+    Total: int = 0  # json:"Total,omitempty"
+
+
+@dataclass
+class BindProjectUserReq:
+    ProjectName: str  # json:"ProjectName"
+    Role: str  # json:"Role"
+    UserName: str  # json:"UserName"
+
+
+@dataclass
+class RemoveProjectUserReq:
+    ProjectName: str  # json:"ProjectName"
+    Role: str  # json:"Role"
+    UserName: str  # json:"UserName"
+
+
+@dataclass
+class ListProjectUsersReq:
+    ProjectName: str  # query:"ProjectName"
+    Role: str  # query:"Role"
+    Limit: int = 10  # query:"Limit,omitempty"
+    Offset: int = 0  # query:"Offset,omitempty"
+
+
+@dataclass
+class ProjectUser:
+    CreationTime: str  # json:"CreationTime,omitempty"
+    Role: str  # json:"Role,omitempty"
+    UserName: str  # json:"UserName,omitempty"
+
+
+@dataclass
+class ListProjectUsersData:
+    Items: Optional[List[ProjectUser]] = None  # json:"Items,omitempty"
+    Limit: int = 10  # json:"Limit,omitempty"
+    Offset: int = 0  # json:"Offset,omitempty"
+    Total: int = 0  # json:"Total,omitempty"
+
+
+@dataclass
+class QuotaGroupRole:
+    Role: str  # json:"Role"
+    UserName: str  # json:"UserName"
+
+
+@dataclass
+class CreateQuotaGroupReq:
+    ProjectName: str  # json:"ProjectName"
+    QuotaGroupName: str  # json:"QuotaGroupName"
+    QuotaGroupRoles: Optional[List[QuotaGroupRole]] = None  # json:"QuotaGroupRoles,omitempty"
+    Quota: Optional[str] = None  # json:"Quota"
+
+
+@dataclass
+class QuotaGroupData:
+    CreationTime: str  # json:"CreationTime,omitempty"
+    ProjectName: str  # json:"ProjectName,omitempty"
+    QuotaGroupId: str  # json:"QuotaGroupId,omitempty"
+    QuotaGroupName: str  # json:"QuotaGroupName,omitempty"
+    QuotaGroupRoles: Optional[List[QuotaGroupRole]] = None  # json:"QuotaGroupRoles,omitempty"
+
+
+@dataclass
+class QuotaGroupBasicData:
+    CreationTime: str  # json:"CreationTime,omitempty"
+    ProjectName: str  # json:"ProjectName,omitempty"
+    QuotaGroupId: str  # json:"QuotaGroupId,omitempty"
+    QuotaGroupName: str  # json:"QuotaGroupName,omitempty"
+
+
+@dataclass
+class ListQuotaGroupData:
+    Items: Optional[List[QuotaGroupBasicData]] = None  # json:"Items,omitempty"
+    Limit: int = 10  # json:"Limit,omitempty"
+    Offset: int = 0  # json:"Offset,omitempty"
+    Total: int = 0  # json:"Total,omitempty"
+
+
+# GET request models
+@dataclass
+class GetUserReq:
+    UserName: str  # query:"UserName"
+
+
+@dataclass
+class ListUsersReq:
+    Limit: int = 10  # query:"Limit,omitempty"
+    Offset: int = 0  # query:"Offset,omitempty"
+    UserName: str = ""  # query:"UserName,omitempty"
+
+
+@dataclass
+class GetAccessKeyReq:
+    AccessKey: str  # query:"AccessKey"
+
+
+@dataclass
+class ListAccessKeysReq:
+    UserName: str  # query:"UserName"
+
+
+@dataclass
+class GetAccessKeyPolicyReq:
+    AccessKey: str  # query:"AccessKey"
+
+
+@dataclass
+class GetProjectReq:
+    ProjectName: str  # query:"ProjectName"
+
+
+@dataclass
+class ListProjectsReq:
+    Limit: int = 10  # query:"Limit,omitempty"
+    Offset: int = 0  # query:"Offset,omitempty"
+    ProjectName: str = ""  # query:"ProjectName,omitempty"
+
+
+@dataclass
+class GetQuotaGroupReq:
+    ProjectName: str  # query:"ProjectName"
+    QuotaGroupName: str  # query:"QuotaGroupName"
+
+
+@dataclass
+class DeleteQuotaGroupReq:
+    ProjectName: str  # query:"ProjectName"
+    QuotaGroupName: str  # query:"QuotaGroupName"
+
+
+@dataclass
+class ListQuotaGroupsReq:
+    Limit: int = 10  # query:"Limit,omitempty"
+    Offset: int = 0  # query:"Offset,omitempty"
+    ProjectName: str = ""  # query:"ProjectName,omitempty"
+    QuotaGroupName: str = ""  # query:"QuotaGroupName,omitempty"

yicloud/services/job/__init__.py

@@ -0,0 +1,44 @@
+from yicloud.services.job.client import JobClient as Client, client, use_client
+from . import actions, models
+
+from .actions import (
+    list_jobs,
+    get_job,
+    create_job,
+    edit_job,
+    batch_delete_jobs,
+    batch_stop_jobs,
+    get_logs,
+    get_logs_volume,
+    list_job_replicas,
+    get_job_replica,
+    get_job_replica_pid,
+)
+from .models import (
+    TaskSpec,
+    ListJobsReq,
+    JobData,
+    TaskSpecResponse,
+    ListJobsData,
+    GetJobReq,
+    CreateJobReq,
+    CreateJobData,
+    EditJobReq,
+    BatchDeleteJobsReq,
+    BatchStopJobsReq,
+    BatchStopJobsData,
+    GetLogsReq,
+    LogEntry,
+    GetLogsResponse,
+    GetLogsVolumeReq,
+    VolumeDataPoint,
+    VolumeResultItem,
+    VolumeMetadata,
+    GetLogsVolumeResponse,
+    ListJobReplicasReq,
+    ReplicaData,
+    ListJobReplicasData,
+    GetJobReplicaReq,
+    GetJobReplicaPidReq,
+    PidInfoData,
+)

yicloud/services/job/actions.py

@@ -0,0 +1,167 @@
+from typing import Optional
+
+from yicloud.base import msgs
+from yicloud.services.job import models
+from yicloud.services.job.client import client
+
+_PRODUCT_CODE = "job"
+_LAST_VERSION = "v1alpha1"
+
+
+def list_jobs(ctx: Optional[dict], req: models.ListJobsReq) -> models.ListJobsData:
+    """
+    GET /job/v1alpha1/ListJobs
+    Action: ListJobs
+
+    Returns ListJobsData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/ListJobs"
+    rsp = msgs.Rsp[models.ListJobsData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def get_job(ctx: Optional[dict], req: models.GetJobReq) -> models.JobData:
+    """
+    GET /job/v1alpha1/GetJob
+    Action: GetJob
+
+    Returns JobData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/GetJob"
+    rsp = msgs.Rsp[models.JobData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def create_job(ctx: Optional[dict], req: models.CreateJobReq) -> models.CreateJobData:
+    """
+    POST /job/v1alpha1/CreateJob
+    Action: CreateJob
+
+    Returns CreateJobData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/CreateJob"
+    rsp = msgs.Rsp[models.CreateJobData]()
+    client.base.post(ctx, path, req.__dict__, rsp)
+    return rsp.get_typed_data()
+
+
+def edit_job(ctx: Optional[dict], req: models.EditJobReq) -> None:
+    """
+    POST /job/v1alpha1/EditJob
+    Action: EditJob
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/EditJob"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)
+
+
+def batch_delete_jobs(ctx: Optional[dict], req: models.BatchDeleteJobsReq) -> None:
+    """
+    POST /job/v1alpha1/BatchDeleteJobs
+    Action: BatchDeleteJobs
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/BatchDeleteJobs"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)
+
+
+def batch_stop_jobs(ctx: Optional[dict], req: models.BatchStopJobsReq) -> models.BatchStopJobsData:
+    """
+    POST /job/v1alpha1/BatchStopJobs
+    Action: BatchStopJobs
+
+    Returns BatchStopJobsData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/BatchStopJobs"
+    rsp = msgs.Rsp[models.BatchStopJobsData]()
+    client.base.post(ctx, path, req.__dict__, rsp)
+    return rsp.get_typed_data()
+
+
+def get_logs(ctx: Optional[dict], req: models.GetLogsReq) -> models.GetLogsResponse:
+    """
+    GET /job/v1alpha1/GetLogs
+    Action: GetLogs
+
+    Returns GetLogsResponse on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/GetLogs"
+    rsp = msgs.Rsp[models.GetLogsResponse]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def get_logs_volume(ctx: Optional[dict], req: models.GetLogsVolumeReq) -> models.GetLogsVolumeResponse:
+    """
+    GET /job/v1alpha1/GetLogsVolume
+    Action: GetLogsVolume
+
+    Returns GetLogsVolumeResponse on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/GetLogsVolume"
+    rsp = msgs.Rsp[models.GetLogsVolumeResponse]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def list_job_replicas(ctx: Optional[dict], req: models.ListJobReplicasReq) -> models.ListJobReplicasData:
+    """
+    GET /job/v1alpha1/ListJobReplicas
+    Action: ListJobReplicas
+
+    Returns ListJobReplicasData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/ListJobReplicas"
+    rsp = msgs.Rsp[models.ListJobReplicasData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def get_job_replica(ctx: Optional[dict], req: models.GetJobReplicaReq) -> models.ReplicaData:
+    """
+    GET /job/v1alpha1/GetJobReplica
+    Action: GetJobReplica
+
+    Returns ReplicaData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/GetJobReplica"
+    rsp = msgs.Rsp[models.ReplicaData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def get_job_replica_pid(ctx: Optional[dict], req: models.GetJobReplicaPidReq) -> Optional[dict]:
+    """
+    GET /job/v1alpha1/GetJobReplicaPid
+    Action: GetJobReplicaPid
+
+    Returns list of PIDs.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/GetJobReplicaPid"
+    rsp = msgs.Rsp[None]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()

yicloud/services/job/client.py

@@ -0,0 +1,19 @@
+from yicloud.base import Client
+
+
+class JobClient:
+    """Job service client, matching Go SDK's services/job/client.go."""
+
+    def __init__(self, base_client: Client = None):
+        self.base = base_client
+        self.product_code = "job"
+        self.last_version = "v1alpha1"
+
+
+# Singleton matching Go SDK
+client: JobClient = JobClient()
+
+
+def use_client(cli: Client) -> None:
+    """Set the base client for Job service."""
+    client.base = cli