import dynamix_sdk.base as _base
import dynamix_sdk.api._nested as _nested


class CloudbrokerDisksMigrateStatusResultModel(_base.BaseAPIResultModel):
    bandwidth_limit: int | None = None
    current_bytes: int | str | None = None  # BDX-9583
    progress_percent: int | None = None
    total_bytes: str | None = None  # BDX-9583
    type: _nested.DiskMigrateJobType | None = None


class CloudbrokerDisksMigrateStatusProtocol(
    _base.BaseGetAPIFunctionProtocol
):
    def migrate_status(
        self,
        *,
        disk_id: int,
    ) -> CloudbrokerDisksMigrateStatusResultModel:
        ...
