# -*- coding: utf-8 -*-
# Copyright (c) Alibaba, Inc. and its affiliates.

from typing import Optional
from dashscope.client.base_api import (
    CreateMixin,
    DeleteMixin,
    GetMixin,
    ListMixin,
    PutMixin,
    StreamEventMixin,
)
from dashscope.finetune.customize_types import (
    Deployment,
    DeploymentDelete,
    DeploymentList,
)


class Deployments(
    CreateMixin,
    DeleteMixin,
    ListMixin,
    GetMixin,
    StreamEventMixin,
    PutMixin,
):
    SUB_PATH = "deployments"
    """Deploy a model.
    """

    @classmethod
    # type: ignore[override]
    def call(  # pylint: disable=arguments-renamed  # type: ignore[override]
        cls,
        model: str,
        capacity: int,
        version: str = None,
        suffix: Optional[str] = None,
        api_key: str = None,
        workspace: str = None,
        **kwargs,
    ) -> Deployment:
        """Call to deployment a model service.

        Args:
            model (str): The model name.
            version (str, optional): The model version, unnecessary
                for fine-tuned model. Defaults to None.
            suffix (str, optional): The name suffix of the model deployment,
                If specified, the final model name will be model_suffix.
                Defaults to None.
            capacity (int, optional): The model service capacity.
            api_key (str, optional): The api-key. Defaults to None.
            workspace (str): The dashscope workspace id.

        Returns:
            Deployment: _description_
        """
        req = {"model_name": model, "capacity": capacity}

        if version is not None:
            req["model_version"] = version
        if suffix is not None:
            req["suffix"] = suffix
        response = super().call(
            req,
            api_key=api_key,
            workspace=workspace,
            **kwargs,
        )
        return Deployment(**response)

    @classmethod
    def list(  # type: ignore[override]
        cls,
        page_no=1,
        page_size=10,
        api_key: str = None,
        workspace: str = None,
        **kwargs,
    ) -> DeploymentList:
        """List deployments.

        Args:
            api_key (str, optional): The api api_key, if not present,
                will get by default rule(TODO: api key doc). Defaults to None.
            page_no (int, optional): Page number. Defaults to 1.
            page_size (int, optional): Items per page. Defaults to 10.
            workspace (str): The dashscope workspace id.

        Returns:
            Deployment: The deployment list.
        """
        response = super().list(
            page_no=page_no,
            page_size=page_size,
            api_key=api_key,
            workspace=workspace,
            **kwargs,
        )
        return DeploymentList(**response)

    @classmethod
    def get(  # type: ignore[override]
        cls,
        deployed_model: str,
        api_key: str = None,
        workspace: str = None,
        **kwargs,
    ) -> Deployment:
        """Get model deployment information.

        Args:
            deployed_model (str): The deployment_id.
            api_key (str, optional): The api key. Defaults to None.
            workspace (str): The dashscope workspace id.

        Returns:
            Deployment: The deployment information.
        """
        response = super().get(
            deployed_model,
            api_key=api_key,
            workspace=workspace,
            **kwargs,
        )
        return Deployment(**response)

    @classmethod
    def delete(  # type: ignore[override]
        cls,
        deployed_model: str,
        api_key: str = None,
        workspace: str = None,
        **kwargs,
    ) -> DeploymentDelete:
        """Delete model deployment.

        Args:
            deployed_model (str): The deployment id.
            api_key (str, optional): The api key. Defaults to None.
            workspace (str): The dashscope workspace id.

        Returns:
            Deployment: The delete result.
        """
        response = super().delete(
            deployed_model,
            api_key=api_key,
            workspace=workspace,
            **kwargs,
        )
        return DeploymentDelete(**response)

    @classmethod
    def scale(
        cls,
        deployed_model: str,
        capacity: int,
        api_key: str = None,
        workspace: str = None,
        **kwargs,
    ) -> Deployment:
        """Scaling model deployment.

        Args:
            deployment_id (str): The deployment id.
            capacity (int): The target service capacity.
            api_key (str, optional): The api key. Defaults to None.

        Returns:
            Deployment: The delete result.
        """
        req = {"deployed_model": deployed_model, "capacity": capacity}
        path = f"{cls.SUB_PATH.lower()}/{deployed_model}/scale"
        response = super().put(
            deployed_model,
            req,
            path=path,
            api_key=api_key,
            workspace=workspace,
            **kwargs,
        )
        return Deployment(**response)