qwak-core 0.4.278__py3-none-any.whl → 0.5.19__py3-none-any.whl

_qwak_proto/qwak/fitness_service/fitness_service_pb2_grpc.py

@@ -1,169 +0,0 @@
-# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
-"""Client and server classes corresponding to protobuf-defined services."""
-import grpc
-
-from _qwak_proto.qwak.fitness_service import fitness_service_pb2 as qwak_dot_fitness__service_dot_fitness__service__pb2
-
-
-class FitnessServiceStub(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def __init__(self, channel):
-        """Constructor.
-
-        Args:
-            channel: A grpc.Channel.
-        """
-        self.BuildModel = channel.unary_unary(
-                '/qwak.fitness.service.FitnessService/BuildModel',
-                request_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelRequest.SerializeToString,
-                response_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelResponse.FromString,
-                )
-        self.CancelBuildModel = channel.unary_unary(
-                '/qwak.fitness.service.FitnessService/CancelBuildModel',
-                request_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelRequest.SerializeToString,
-                response_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelResponse.FromString,
-                )
-        self.GetRemoteBuildingStatus = channel.unary_unary(
-                '/qwak.fitness.service.FitnessService/GetRemoteBuildingStatus',
-                request_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusRequest.SerializeToString,
-                response_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusResponse.FromString,
-                )
-        self.CleanRemoteBuildResources = channel.unary_unary(
-                '/qwak.fitness.service.FitnessService/CleanRemoteBuildResources',
-                request_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesRequest.SerializeToString,
-                response_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesResponse.FromString,
-                )
-
-
-class FitnessServiceServicer(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def BuildModel(self, request, context):
-        """Build a serving model image
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def CancelBuildModel(self, request, context):
-        """Cancel an ongoing build
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetRemoteBuildingStatus(self, request, context):
-        """Get the status of remote build
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def CleanRemoteBuildResources(self, request, context):
-        """Clean the remote build resources
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-
-def add_FitnessServiceServicer_to_server(servicer, server):
-    rpc_method_handlers = {
-            'BuildModel': grpc.unary_unary_rpc_method_handler(
-                    servicer.BuildModel,
-                    request_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelRequest.FromString,
-                    response_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelResponse.SerializeToString,
-            ),
-            'CancelBuildModel': grpc.unary_unary_rpc_method_handler(
-                    servicer.CancelBuildModel,
-                    request_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelRequest.FromString,
-                    response_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelResponse.SerializeToString,
-            ),
-            'GetRemoteBuildingStatus': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetRemoteBuildingStatus,
-                    request_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusRequest.FromString,
-                    response_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusResponse.SerializeToString,
-            ),
-            'CleanRemoteBuildResources': grpc.unary_unary_rpc_method_handler(
-                    servicer.CleanRemoteBuildResources,
-                    request_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesRequest.FromString,
-                    response_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesResponse.SerializeToString,
-            ),
-    }
-    generic_handler = grpc.method_handlers_generic_handler(
-            'qwak.fitness.service.FitnessService', rpc_method_handlers)
-    server.add_generic_rpc_handlers((generic_handler,))
-
-
- # This class is part of an EXPERIMENTAL API.
-class FitnessService(object):
-    """Missing associated documentation comment in .proto file."""
-
-    @staticmethod
-    def BuildModel(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/qwak.fitness.service.FitnessService/BuildModel',
-            qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelRequest.SerializeToString,
-            qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def CancelBuildModel(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/qwak.fitness.service.FitnessService/CancelBuildModel',
-            qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelRequest.SerializeToString,
-            qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def GetRemoteBuildingStatus(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/qwak.fitness.service.FitnessService/GetRemoteBuildingStatus',
-            qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusRequest.SerializeToString,
-            qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def CleanRemoteBuildResources(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/qwak.fitness.service.FitnessService/CleanRemoteBuildResources',
-            qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesRequest.SerializeToString,
-            qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

frogml_storage/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.4.278"

frogml_storage/artifactory/__init__.py

@@ -1 +0,0 @@
-from ._artifactory_api import ArtifactoryApi, StartTransactionResponse

frogml_storage/artifactory/_artifactory_api.py

@@ -1,315 +0,0 @@
-import json
-import os
-from typing import Optional
-
-from requests import Response
-from tqdm.auto import tqdm
-from tqdm.utils import CallbackIOWrapper
-from urllib3 import Retry
-
-from frogml_storage.logging import logger
-from frogml_storage.utils import join_url
-from frogml_storage.constants import CHECKSUM_SHA2_HEADER
-from frogml_storage.models.entity_manifest import Checksums, EntityManifest
-from frogml_storage.models.frogml_entity_type_info import FrogMLEntityTypeInfo
-from frogml_storage.http import HTTPClient
-from frogml_storage.models import DownloadContext
-
-
-class StartTransactionResponse:
-    files_upload_path: str
-    lead_upload_path: str
-    dependencies_upload_path: str
-    code_upload_path: str
-    transaction_id: str
-
-    def __init__(
-        self,
-        files_upload_path,
-        lead_upload_path,
-        dependencies_upload_path,
-        code_upload_path,
-        transaction_id,
-    ):
-        self.files_upload_path = files_upload_path
-        self.lead_upload_path = lead_upload_path
-        self.dependencies_upload_path = dependencies_upload_path
-        self.code_upload_path = code_upload_path
-        self.transaction_id = transaction_id
-
-
-class ArtifactoryApi:
-    def __init__(self, uri, auth=None, http_client=None):
-        self.uri = uri
-        if http_client is not None:
-            self.http_client = http_client
-        else:
-            self.auth = auth
-            self.http_client = HTTPClient(auth=auth)
-
-    def start_transaction(
-        self,
-        entity_type_info: FrogMLEntityTypeInfo,
-        repository: str,
-        entity_name: str,
-        version: Optional[str],
-    ) -> StartTransactionResponse:
-        """
-        Initializes an upload. Returns transaction ID and upload path
-        """
-        if version is None:
-            start_transaction_url = (
-                f"{self.uri}/api/machinelearning/{repository}/"
-                f"{entity_type_info.entity_type}/{entity_name}/start-transaction"
-            )
-        else:
-            start_transaction_url = (
-                f"{self.uri}/api/machinelearning/{repository}/{entity_type_info.entity_type}"
-                f"/{entity_name}/start-transaction/{version}"
-            )
-        try:
-            response = self.http_client.post(start_transaction_url)
-            response.raise_for_status()
-            files_upload_path = response.json()["filesUploadPath"]
-            lead_upload_path = response.json()["leadUploadPath"]
-            dependencies_upload_path = response.json()["dependenciesUploadPath"]
-            code_upload_path = response.json()["codeUploadPath"]
-            transaction_id = response.json()["transactionId"]
-        except Exception as exception:
-            err = (
-                f"Error occurred while trying to start an upload transaction for "
-                f"{entity_type_info.entity_type}: '{entity_name}'"
-                f" Error: '{exception}'"
-            )
-            logger.error(err, exc_info=False)
-            raise exception
-        return StartTransactionResponse(
-            files_upload_path=files_upload_path,
-            lead_upload_path=lead_upload_path,
-            dependencies_upload_path=dependencies_upload_path,
-            code_upload_path=code_upload_path,
-            transaction_id=transaction_id,
-        )
-
-    def end_transaction(
-        self,
-        entity_type: FrogMLEntityTypeInfo,
-        repository: str,
-        entity_name: str,
-        entity_manifest: EntityManifest,
-        transaction_id: str,
-        version: str,
-        properties: Optional[dict[str, str]],
-    ) -> None:
-        """
-        Upload model-manifest.json | dataset-manifest.json file, makes the model | dataset available in the repository
-        """
-        filename = entity_type.metadata_file_name
-
-        url = join_url(
-            self.uri,
-            "api",
-            "machinelearning",
-            repository,
-            entity_type.entity_type,
-            "entity-manifest",
-            entity_name,
-            version,
-            transaction_id,
-        )
-
-        json_entity_manifest = entity_manifest.to_json()
-        self.upload_entity_manifest(
-            entity_type=entity_type,
-            filename=filename,
-            payload=json_entity_manifest,
-            url=url,
-            properties=properties,
-        )
-
-    def download_file(self, args: DownloadContext) -> None:
-        filename = os.path.basename(args.target_path)
-        try:
-            url = f"{self.uri}/{args.repo_key}/{args.source_url}"
-            with self.http_client.get(url=url, stream=True) as response:
-                response.raise_for_status()
-                total_size = int(response.headers.get("content-length", 0))
-                with open(args.target_path, "wb") as file:
-                    with self.__initialize_progress_bar(total_size, filename) as pbar:
-                        for chunk in response.iter_content(chunk_size=8192):
-                            if chunk:
-                                file.write(chunk)
-                                pbar.update(len(chunk))
-
-        except Exception as exception:
-            self.__handle_download_exception(exception, args.target_path, filename)
-
-    def get_entity_manifest(
-        self,
-        entity_type_info: FrogMLEntityTypeInfo,
-        repository: str,
-        entity_name: str,
-        namespace: Optional[str],
-        version: Optional[str],
-    ) -> dict:
-        url = join_url(
-            self.uri,
-            "api",
-            "machinelearning",
-            repository,
-            entity_type_info.entity_type,
-            "entity-manifest",
-            namespace,
-            entity_name,
-            version,
-        )
-        try:
-            with self.http_client.get(url=url) as r:
-                r.raise_for_status()
-                return r.json()
-        except Exception as exception:
-            err = f"Error occurred while trying to get {entity_type_info.entity_type} info file. Error: '{exception}'"
-            logger.error(err, exc_info=False)
-            raise exception
-
-    @staticmethod
-    def __handle_download_exception(
-        exception: Exception, target_path: str, filename: str
-    ) -> None:
-        if os.path.exists(target_path):
-            os.remove(target_path)
-        err = f"Error occurred while trying to download file: '{filename}' Error: '{exception}'"
-        logger.error(err, exc_info=False)
-        raise exception
-
-    def get_artifact_checksum(self, download_context: DownloadContext) -> str:
-        url = f"{self.uri}/{download_context.repo_key}/{download_context.source_url}"
-        try:
-            with self.http_client.head(url=url, stream=True) as response:
-                response.raise_for_status()
-                return response.headers.get(CHECKSUM_SHA2_HEADER)
-
-        except Exception as exception:
-            logger.error(exception.__cause__, exc_info=False)
-            raise exception
-
-    def upload_entity_manifest(
-        self,
-        entity_type: FrogMLEntityTypeInfo,
-        filename: str,
-        payload: str,
-        url: str,
-        properties: Optional[dict[str, str]],
-        stream: bool = False,
-    ) -> None:
-        body_part_name = f"{entity_type.body_part_stream}"
-
-        try:
-            files = {
-                f"{body_part_name}": (
-                    f"{body_part_name}",
-                    payload,
-                    "application/octet-stream",
-                ),  # Include the InputStream
-                "additionalData": (
-                    "additionalData",
-                    json.dumps(properties),
-                    "application/octet-stream",
-                ),  # Include the object
-            }
-            with self.http_client.put(url=url, files=files, stream=stream) as response:
-                response.raise_for_status()
-        except Exception as exception:
-            err = f"Error occurred while trying to upload file: '{filename}' Error: '{exception}'"
-            logger.error(err, exc_info=False)
-            raise exception
-
-    def upload_file(self, file_path: str, url: str) -> None:
-        wrapped_file = None
-        try:
-            file_size = os.stat(file_path).st_size
-            with (
-                self.__initialize_progress_bar(file_size, file_path) as pbar,
-                open(file_path, "rb") as file,
-            ):
-                wrapped_file = CallbackIOWrapper(pbar.update, file, "read")
-                with self.http_client.put(url=url, payload=wrapped_file) as response:
-                    response.raise_for_status()
-        except Exception as exception:
-            err = f"Error occurred while trying to upload file: '{file_path}' Error: '{exception}'"
-            logger.error(err, exc_info=False)
-            raise type(exception)(f"{err} File: {file_path}") from exception
-        finally:
-            if wrapped_file is not None:
-                wrapped_file.close()
-
-    def checksum_deployment(
-        self,
-        checksum: Checksums,
-        url: str,
-        full_path: str,
-        stream: bool = False,
-    ) -> bool:
-        response = self.http_client.put(
-            url=url,
-            headers={"X-Checksum-Sha256": checksum.sha2, "X-Checksum-Deploy": "true"},
-            stream=stream,
-        )
-        if response.status_code != 200 and response.status_code != 201:
-            return False
-        else:
-            file_size = os.path.getsize(full_path)
-            pbar = self.__initialize_progress_bar(file_size, full_path)
-            pbar.update(file_size)
-            pbar.close()
-            return True
-
-    @staticmethod
-    def __initialize_progress_bar(total_size: int, filename: str) -> tqdm:
-        return tqdm(
-            total=total_size, unit="B", unit_scale=True, desc=filename, initial=0
-        )
-
-    def encrypt_password(self) -> Response:
-        """
-        returns encrypted password as text
-        """
-        return self.http_client.get(
-            url=join_url(self.uri, "/api/security/encryptedPassword")
-        )
-
-    def ping(self) -> Response:
-        """
-        Sends a ping to Artifactory to validate login status
-        """
-        url = join_url(self.uri, "api/system/ping")
-        return self.http_client.get(url=url)
-
-    def get_artifactory_version(self) -> Response:
-        return self.http_client.get(url=join_url(self.uri, "/api/system/version"))
-
-    def create_machinelearning_local_repo(self, repo_name: str) -> Response:
-        data = {
-            "rclass": "local",
-            "packageType": "machinelearning",
-        }
-        return self.http_client.put(
-            url=join_url(self.uri, "/api/repositories/" + repo_name), json=data
-        )
-
-    def delete_frogml_local_repo(self, repo_name: str) -> Response:
-        return self.http_client.delete(
-            url=join_url(self.uri, "/api/repositories/" + repo_name)
-        )
-
-
-class RetryWithLog(Retry):
-    """
-    Adding extra logs before making a retry request
-    """
-
-    def __init__(self, *args, **kwargs):
-        history = kwargs.get("history")
-        if history is not None:
-            logger.debug(f"Error: ${history[-1].error}\nretrying...")
-        super().__init__(*args, **kwargs)

frogml_storage/authentication/login/__init__.py

@@ -1 +0,0 @@
-from ._login_cli import login as frogml_login