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

yicloud/services/job/models.py

@@ -0,0 +1,268 @@
+# Code generated from Go SDK's services/job/models.go. DO NOT EDIT.
+
+from dataclasses import dataclass, field
+from typing import Optional, List, Dict
+
+
+@dataclass
+class TaskSpec:
+    """Task specification for job."""
+    Replicas: Optional[int] = None  # *int32 in Go, using Optional[int]
+    PositiveTags: List[str] = field(default_factory=list)
+    NegativeTags: List[str] = field(default_factory=list)
+    MaxWaitDuration: Optional[str] = None
+    MaxRunningDuration: Optional[str] = None
+    Image: str = ""
+    Command: List[str] = field(default_factory=list)
+    Env: Dict[str, str] = field(default_factory=dict)
+    CPU: Optional[str] = None
+    Memory: Optional[str] = None
+    GPU: Optional[str] = None
+    SKUId: Optional[str] = None
+    MountGpfs: Optional[str] = None
+
+
+@dataclass
+class TaskSpecResponse:
+    """Task specification response."""
+    Replicas: Optional[int] = None
+    PositiveTags: List[str] = field(default_factory=list)
+    NegativeTags: List[str] = field(default_factory=list)
+    MaxWaitDuration: Optional[str] = None
+    MaxRunningDuration: Optional[str] = None
+    Image: Optional[str] = None
+    Command: List[str] = field(default_factory=list)
+    Env: Dict[str, str] = field(default_factory=dict)
+    CPU: Optional[str] = None
+    Memory: Optional[str] = None
+    GPU: Optional[str] = None
+    SKUId: Optional[str] = None
+    MountGpfs: Optional[str] = None
+
+
+@dataclass
+class JobData:
+    """Job data."""
+    Name: Optional[str] = None
+    Project: Optional[str] = None
+    Labels: Dict[str, str] = field(default_factory=dict)
+    JobId: Optional[str] = None
+    TaskType: Optional[str] = None
+    QuotaGroup: Optional[str] = None
+    UsePrivateMachine: bool = False
+    SKUPrivate: bool = False
+    SKUPublic: bool = False
+    SKUPrivateTenant: bool = False
+    SKUPrivateProject: bool = False
+    Mount: Optional[str] = None
+    CreationTime: Optional[str] = None
+    Phase: Optional[str] = None
+    TaskSpecs: Dict[str, TaskSpecResponse] = field(default_factory=dict)
+    BackoffLimit: Optional[int] = None  # *int32 in Go
+    StopTime: Optional[str] = None
+    AutoDeleteDuration: Optional[str] = None
+
+
+@dataclass
+class ListJobsData:
+    """List jobs response data."""
+    Items: List[JobData] = field(default_factory=list)
+    Total: int = 0
+
+
+@dataclass
+class ListJobsReq:
+    """List jobs request."""
+    Project: str = ""
+    Offset: int = 0
+    Limit: int = 0
+    SortBy: str = ""
+    Creators: str = ""
+    JobNames: str = ""
+    JobIds: str = ""
+    Self: bool = False
+    JobType: str = ""
+
+
+@dataclass
+class GetJobReq:
+    """Get job request."""
+    Project: str = ""
+    JobId: str = ""
+
+
+@dataclass
+class CreateJobReq:
+    """Create job request."""
+    Name: str = ""
+    Project: str = ""
+    Labels: Dict[str, str] = field(default_factory=dict)
+    JobId: str = ""
+    TaskType: str = ""
+    QuotaGroup: str = ""
+    UsePrivateMachine: bool = False
+    SKUPrivate: bool = False
+    SKUPublic: bool = False
+    SKUPrivateTenant: bool = False
+    SKUPrivateProject: bool = False
+    Mount: str = ""
+    TaskSpecs: Dict[str, TaskSpec] = field(default_factory=dict)
+    BackoffLimit: Optional[int] = None
+    AutoDeleteDuration: str = ""
+
+
+@dataclass
+class CreateJobData:
+    """Create job response data."""
+    Name: str = ""
+    Project: str = ""
+    CreationTime: Optional[str] = None
+    JobId: str = ""
+
+
+@dataclass
+class EditJobReq:
+    """Edit job request."""
+    Project: str = ""
+    JobId: str = ""
+    Operation: str = ""
+    Priority: str = ""
+    TopAction: str = ""
+
+
+@dataclass
+class BatchDeleteJobsReq:
+    """Batch delete jobs request."""
+    Project: str = ""
+    JobIds: List[str] = field(default_factory=list)
+
+
+@dataclass
+class BatchStopJobsReq:
+    """Batch stop jobs request."""
+    Project: str = ""
+    JobIds: List[str] = field(default_factory=list)
+
+
+@dataclass
+class BatchStopJobsData:
+    """Batch stop jobs response data."""
+    Message: str = ""
+    Failed: List[str] = field(default_factory=list)
+    Reasons: Dict[str, str] = field(default_factory=dict)
+
+
+@dataclass
+class GetLogsReq:
+    """Get logs request."""
+    Project: str = ""
+    JobId: str = ""
+    Start: int = 0
+    End: int = 0
+    Limit: int = 0
+    Direction: str = ""
+    Keyword: str = ""
+    Regex: bool = False
+    CaseSensitive: bool = False
+
+
+@dataclass
+class LogEntry:
+    """Log entry."""
+    Timestamp: str = ""
+    Line: str = ""
+
+
+@dataclass
+class GetLogsResponse:
+    """Get logs response."""
+    Items: List[LogEntry] = field(default_factory=list)
+    Total: int = 0
+
+
+@dataclass
+class GetLogsVolumeReq:
+    """Get logs volume request."""
+    Project: str = ""
+    JobId: str = ""
+    Start: int = 0
+    End: int = 0
+    Keyword: str = ""
+    Regex: bool = False
+    CaseSensitive: bool = False
+
+
+@dataclass
+class VolumeDataPoint:
+    """Volume data point."""
+    Timestamp: int = 0
+    Count: str = ""
+
+
+@dataclass
+class VolumeResultItem:
+    """Volume result item."""
+    Metric: Dict[str, str] = field(default_factory=dict)
+    Values: List[VolumeDataPoint] = field(default_factory=list)
+
+
+@dataclass
+class VolumeMetadata:
+    """Volume metadata."""
+    Step: int = 0
+
+
+@dataclass
+class GetLogsVolumeResponse:
+    """Get logs volume response."""
+    Metadata: VolumeMetadata = field(default_factory=VolumeMetadata)
+    Result: List[VolumeResultItem] = field(default_factory=list)
+
+
+@dataclass
+class ListJobReplicasReq:
+    """List job replicas request."""
+    JobId: str = ""
+
+
+@dataclass
+class ReplicaData:
+    """Replica data."""
+    Name: str = ""
+    Project: str = ""
+    Phase: str = ""
+    NodeName: str = ""
+    PodIp: str = ""
+    StartTime: str = ""
+    TaskType: str = ""
+
+
+@dataclass
+class ListJobReplicasData:
+    """List job replicas response."""
+    Items: List[ReplicaData] = field(default_factory=list)
+
+
+@dataclass
+class GetJobReplicaReq:
+    """Get job replica request."""
+    JobId: str = ""
+    ReplicaName: str = ""
+
+
+@dataclass
+class GetJobReplicaPidReq:
+    """Get job replica PID request."""
+    JobId: str = ""
+    ReplicaName: str = ""
+
+
+@dataclass
+class PidInfoData:
+    """PID info data."""
+    Pid: str = ""
+    Ppid: str = ""
+    Pcpu: str = ""
+    Rss: str = ""
+    State: str = ""
+    Args: str = ""

yicloud/services/mc/__init__.py

@@ -0,0 +1,59 @@
+from yicloud.services.mc.client import MCClient as Client, client, use_client
+from . import actions, models
+
+from .actions import (
+    assign_nodes,
+    list_nodes,
+    list_skus,
+    bind_project_user,
+    create_project,
+    delete_project,
+    get_project,
+    list_project_users,
+    list_projects,
+    remove_project_user,
+    update_project,
+    create_quota_group,
+    delete_quota_group,
+    get_quota_group,
+    list_quota_groups,
+    update_quota_group,
+)
+from .models import (
+    Role,
+    AssignNodeReq,
+    ResourceDetail,
+    NodeConfig,
+    NodeData,
+    ListNodesData,
+    SKUData,
+    ListSKUsData,
+    BindProjectUserReq,
+    Quota,
+    CreateProjectReq,
+    DeleteProjectReq,
+    ProjectStatus,
+    QuotaStatus,
+    ProjectData,
+    ProjectUser,
+    ListProjectUsersData,
+    ListProjectsData,
+    RemoveProjectUserReq,
+    UpdateProjectReq,
+    QuotaGroupUser,
+    UsageType,
+    MachineType,
+    CreateQuotaGroupReq,
+    QuotaGroupData,
+    QuotaGroupBasicData,
+    ListQuotaGroupData,
+    UpdateQuotaGroupReq,
+    ListNodesReq,
+    ListSKUsReq,
+    GetProjectReq,
+    ListProjectUsersReq,
+    ListProjectsReq,
+    DeleteQuotaGroupReq,
+    GetQuotaGroupReq,
+    ListQuotaGroupsReq,
+)

yicloud/services/mc/actions.py

@@ -0,0 +1,221 @@
+from typing import Optional
+
+from yicloud.base import msgs
+from yicloud.services.mc import models
+from yicloud.services.mc.client import client
+
+_PRODUCT_CODE = "mc"
+_LAST_VERSION = "v1alpha1"
+
+
+def assign_nodes(ctx: Optional[dict], req: models.AssignNodeReq) -> None:
+    """
+    POST /mc/v1alpha1/AssignNodes
+    Action: AssignNodes
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/AssignNodes"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)
+
+
+def list_nodes(ctx: Optional[dict], req: models.ListNodesReq) -> models.ListNodesData:
+    """
+    GET /mc/v1alpha1/ListNodes
+    Action: ListNodes
+
+    Returns ListNodesData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/ListNodes"
+    rsp = msgs.Rsp[models.ListNodesData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def list_skus(ctx: Optional[dict], req: models.ListSKUsReq) -> models.ListSKUsData:
+    """
+    GET /mc/v1alpha1/ListSKUs
+    Action: ListSKUs
+
+    Returns ListSKUsData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/ListSKUs"
+    rsp = msgs.Rsp[models.ListSKUsData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def bind_project_user(ctx: Optional[dict], req: models.BindProjectUserReq) -> None:
+    """
+    POST /mc/v1alpha1/BindProjectUser
+    Action: BindProjectUser
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/BindProjectUser"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)
+
+
+def create_project(ctx: Optional[dict], req: models.CreateProjectReq) -> None:
+    """
+    POST /mc/v1alpha1/CreateProject
+    Action: CreateProject
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/CreateProject"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)
+
+
+def delete_project(ctx: Optional[dict], req: models.DeleteProjectReq) -> None:
+    """
+    POST /mc/v1alpha1/DeleteProject
+    Action: DeleteProject
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/DeleteProject"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)
+
+
+def get_project(ctx: Optional[dict], req: models.GetProjectReq) -> models.ProjectData:
+    """
+    GET /mc/v1alpha1/GetProject
+    Action: GetProject
+
+    Returns ProjectData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/GetProject"
+    rsp = msgs.Rsp[models.ProjectData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def list_project_users(ctx: Optional[dict], req: models.ListProjectUsersReq) -> models.ListProjectUsersData:
+    """
+    GET /mc/v1alpha1/ListProjectUsers
+    Action: ListProjectUsers
+
+    Returns ListProjectUsersData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/ListProjectUsers"
+    rsp = msgs.Rsp[models.ListProjectUsersData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def list_projects(ctx: Optional[dict], req: models.ListProjectsReq) -> models.ListProjectsData:
+    """
+    GET /mc/v1alpha1/ListProjects
+    Action: ListProjects
+
+    Returns ListProjectsData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/ListProjects"
+    rsp = msgs.Rsp[models.ListProjectsData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def remove_project_user(ctx: Optional[dict], req: models.RemoveProjectUserReq) -> None:
+    """
+    POST /mc/v1alpha1/RemoveProjectUser
+    Action: RemoveProjectUser
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/RemoveProjectUser"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)
+
+
+def update_project(ctx: Optional[dict], req: models.UpdateProjectReq) -> None:
+    """
+    POST /mc/v1alpha1/UpdateProject
+    Action: UpdateProject
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/UpdateProject"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)
+
+
+def create_quota_group(ctx: Optional[dict], req: models.CreateQuotaGroupReq) -> None:
+    """
+    POST /mc/v1alpha1/CreateQuotaGroup
+    Action: CreateQuotaGroup
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/CreateQuotaGroup"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)
+
+
+def delete_quota_group(ctx: Optional[dict], req: models.DeleteQuotaGroupReq) -> None:
+    """
+    POST /mc/v1alpha1/DeleteQuotaGroup
+    Action: DeleteQuotaGroup
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/DeleteQuotaGroup"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)
+
+
+def get_quota_group(ctx: Optional[dict], req: models.GetQuotaGroupReq) -> models.QuotaGroupData:
+    """
+    GET /mc/v1alpha1/GetQuotaGroup
+    Action: GetQuotaGroup
+
+    Returns QuotaGroupData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/GetQuotaGroup"
+    rsp = msgs.Rsp[models.QuotaGroupData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def list_quota_groups(ctx: Optional[dict], req: models.ListQuotaGroupsReq) -> models.ListQuotaGroupData:
+    """
+    GET /mc/v1alpha1/ListQuotaGroups
+    Action: ListQuotaGroups
+
+    Returns ListQuotaGroupData on success.
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/ListQuotaGroups"
+    rsp = msgs.Rsp[models.ListQuotaGroupData]()
+    client.base.get(ctx, path, req, rsp)
+    return rsp.get_typed_data()
+
+
+def update_quota_group(ctx: Optional[dict], req: models.UpdateQuotaGroupReq) -> None:
+    """
+    POST /mc/v1alpha1/UpdateQuotaGroup
+    Action: UpdateQuotaGroup
+
+    Raises YiCloudException on error.
+    """
+    path = f"/{_PRODUCT_CODE}/{_LAST_VERSION}/UpdateQuotaGroup"
+    rsp = msgs.Rsp[None].empty()
+    client.base.post(ctx, path, req.__dict__, rsp)

yicloud/services/mc/client.py

@@ -0,0 +1,21 @@
+# Code generated from Go SDK's services/mc/client.go. DO NOT EDIT.
+
+from yicloud.base import Client
+
+
+class MCClient:
+    """Machine Cluster service client, matching Go SDK's services/mc/client.go."""
+
+    def __init__(self, base_client: Client = None):
+        self.base = base_client
+        self.product_code = "mc"
+        self.last_version = "v1alpha1"
+
+
+# Singleton matching Go SDK
+client: MCClient = MCClient()
+
+
+def use_client(cli: Client) -> None:
+    """Set the base client for MC service."""
+    client.base = cli