lightning-sdk 2025.12.17__py3-none-any.whl → 2026.1.22__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_kubevirt_vm_resources.py

@@ -1,201 +0,0 @@
-# coding: utf-8
-
-"""
-    external/v1/auth_service.proto
-
-    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
-
-    OpenAPI spec version: version not set
-    
-    Generated by: https://github.com/swagger-api/swagger-codegen.git
-
-    NOTE
-    ----
-    standard swagger-codegen-cli for this python client has been modified
-    by custom templates. The purpose of these templates is to include
-    typing information in the API and Model code. Please refer to the
-    main grid repository for more info
-"""
-
-import pprint
-import re  # noqa: F401
-
-from typing import TYPE_CHECKING
-
-import six
-
-if TYPE_CHECKING:
-    from datetime import datetime
-    from lightning_sdk.lightning_cloud.openapi.models import *
-
-class V1KubevirtVmResources(object):
-    """NOTE: This class is auto generated by the swagger code generator program.
-
-    Do not edit the class manually.
-    """
-    """
-    Attributes:
-      swagger_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
-    """
-    swagger_types = {
-        'cpu': 'int',
-        'gpus': 'int',
-        'ram': 'str',
-        'storage': 'int'
-    }
-
-    attribute_map = {
-        'cpu': 'cpu',
-        'gpus': 'gpus',
-        'ram': 'ram',
-        'storage': 'storage'
-    }
-
-    def __init__(self, cpu: 'int' =None, gpus: 'int' =None, ram: 'str' =None, storage: 'int' =None):  # noqa: E501
-        """V1KubevirtVmResources - a model defined in Swagger"""  # noqa: E501
-        self._cpu = None
-        self._gpus = None
-        self._ram = None
-        self._storage = None
-        self.discriminator = None
-        if cpu is not None:
-            self.cpu = cpu
-        if gpus is not None:
-            self.gpus = gpus
-        if ram is not None:
-            self.ram = ram
-        if storage is not None:
-            self.storage = storage
-
-    @property
-    def cpu(self) -> 'int':
-        """Gets the cpu of this V1KubevirtVmResources.  # noqa: E501
-
-
-        :return: The cpu of this V1KubevirtVmResources.  # noqa: E501
-        :rtype: int
-        """
-        return self._cpu
-
-    @cpu.setter
-    def cpu(self, cpu: 'int'):
-        """Sets the cpu of this V1KubevirtVmResources.
-
-
-        :param cpu: The cpu of this V1KubevirtVmResources.  # noqa: E501
-        :type: int
-        """
-
-        self._cpu = cpu
-
-    @property
-    def gpus(self) -> 'int':
-        """Gets the gpus of this V1KubevirtVmResources.  # noqa: E501
-
-
-        :return: The gpus of this V1KubevirtVmResources.  # noqa: E501
-        :rtype: int
-        """
-        return self._gpus
-
-    @gpus.setter
-    def gpus(self, gpus: 'int'):
-        """Sets the gpus of this V1KubevirtVmResources.
-
-
-        :param gpus: The gpus of this V1KubevirtVmResources.  # noqa: E501
-        :type: int
-        """
-
-        self._gpus = gpus
-
-    @property
-    def ram(self) -> 'str':
-        """Gets the ram of this V1KubevirtVmResources.  # noqa: E501
-
-
-        :return: The ram of this V1KubevirtVmResources.  # noqa: E501
-        :rtype: str
-        """
-        return self._ram
-
-    @ram.setter
-    def ram(self, ram: 'str'):
-        """Sets the ram of this V1KubevirtVmResources.
-
-
-        :param ram: The ram of this V1KubevirtVmResources.  # noqa: E501
-        :type: str
-        """
-
-        self._ram = ram
-
-    @property
-    def storage(self) -> 'int':
-        """Gets the storage of this V1KubevirtVmResources.  # noqa: E501
-
-
-        :return: The storage of this V1KubevirtVmResources.  # noqa: E501
-        :rtype: int
-        """
-        return self._storage
-
-    @storage.setter
-    def storage(self, storage: 'int'):
-        """Sets the storage of this V1KubevirtVmResources.
-
-
-        :param storage: The storage of this V1KubevirtVmResources.  # noqa: E501
-        :type: int
-        """
-
-        self._storage = storage
-
-    def to_dict(self) -> dict:
-        """Returns the model properties as a dict"""
-        result = {}
-
-        for attr, _ in six.iteritems(self.swagger_types):
-            value = getattr(self, attr)
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
-                    value
-                ))
-            elif hasattr(value, "to_dict"):
-                result[attr] = value.to_dict()
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], item[1].to_dict())
-                    if hasattr(item[1], "to_dict") else item,
-                    value.items()
-                ))
-            else:
-                result[attr] = value
-        if issubclass(V1KubevirtVmResources, dict):
-            for key, value in self.items():
-                result[key] = value
-
-        return result
-
-    def to_str(self) -> str:
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self) -> str:
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other: 'V1KubevirtVmResources') -> bool:
-        """Returns true if both objects are equal"""
-        if not isinstance(other, V1KubevirtVmResources):
-            return False
-
-        return self.__dict__ == other.__dict__
-
-    def __ne__(self, other: 'V1KubevirtVmResources') -> bool:
-        """Returns true if both objects are not equal"""
-        return not self == other

lightning_sdk/lightning_cloud/source_code/logs_socket_api.py

@@ -1,103 +0,0 @@
-from urllib.parse import urlparse
-
-from lightning_sdk.lightning_cloud.login import Auth
-from lightning_sdk.lightning_cloud.openapi import ApiClient, AuthServiceApi, V1LoginRequest
-import websocket
-
-
-class LightningLogsSocketAPI:
-
-    def __init__(self, api_client=None):
-        if api_client is None:
-            api_client = ApiClient()
-        self.api_client = api_client
-        self._auth = None
-        self._auth_service = None
-
-    def _init_auth(self):
-        self._auth = Auth()
-        self._auth.authenticate()
-        self._auth_service = AuthServiceApi(self.api_client)
-
-    def _get_api_token(self):
-        if not self._auth_service:
-            self._init_auth()
-        token_resp = self._auth_service.auth_service_login(body=V1LoginRequest(
-            username=self._auth.user_id,
-            api_key=self._auth.api_key,
-        ))
-        return token_resp.token
-
-    @staticmethod
-    def _socket_url(host: str, project_id: str, app_id: str, token: str,
-                    component: str):
-        return (f"wss://{host}/v1/projects/{project_id}/apps/{app_id}/logs?"
-                f"token={token}&component={component}&follow=true")
-
-    def create_lightning_logs_socket(
-        self,
-        project_id: str,
-        app_id: str,
-        component: str,
-        on_message_callback,
-        on_error_callback=None,
-    ) -> websocket.WebSocketApp:
-        """
-        Creates and returns WebSocketApp to listen to lightning app logs.
-
-        Usage example synchronous:
-            def print_log_msg(ws_app, msg):
-                print(msg)
-
-            flow_logs_socket = client.create_lightning_logs_socket('project_id', 'app_id', 'flow', print_log_msg)
-            flow_socket.run_forever()
-
-        Usage example asynchronous, multiple components:
-            def print_log_msg(ws_app, msg):
-                print(msg)
-
-            flow_logs_socket = client.create_lightning_logs_socket('project_id', 'app_id', 'flow', print_log_msg)
-            work_logs_socket = client.create_lightning_logs_socket('project_id', 'app_id', 'work_1', print_log_msg)
-
-            flow_logs_thread = Thread(target=flow_logs_socket.run_forever)
-            work_logs_thread = Thread(target=work_logs_socket.run_forever)
-
-            flow_logs_thread.start()
-            work_logs_thread.start()
-            .......
-
-            flow_logs_socket.close()
-            work_logs_thread.close()
-
-        Parameters
-        ----------
-        project_id: str
-        app_id: str
-        component: str
-        on_message_callback: function
-            Callback object which is called when received data.
-            on_message_callback has 2 arguments.
-            The 1st argument is the WebSocketApp object.
-            The 2nd argument is utf-8 data received from the server.
-        on_error_callback: function
-            Callback object which is called when we get error.
-            on_error has 2 arguments.
-            The 1st argument is this class object.
-            The 2nd argument is exception object.
-
-        Returns
-        -------
-        WebSocketApp of the wanted socket
-        """
-        clean_ws_host = urlparse(self.api_client.configuration.host).netloc
-        socket_url = self._socket_url(
-            host=clean_ws_host,
-            project_id=project_id,
-            app_id=app_id,
-            token=self._get_api_token(),
-            component=component,
-        )
-
-        return websocket.WebSocketApp(socket_url,
-                                      on_message=on_message_callback,
-                                      on_error=on_error_callback)