yellowdog-sdk 8.1.7__py3-none-any.whl → 8.1.9__py3-none-any.whl

yellowdog_client/_version.py

@@ -1,2 +1,2 @@
 
-__version__ = '8.1.7'  # YEL-12854
+__version__ = '8.1.9'  # YEL-12856

yellowdog_client/model/__init__.py

@@ -215,6 +215,8 @@ from .permission import Permission
 from .permission_detail import PermissionDetail
 from .price import Price
 from .processor_architecture import ProcessorArchitecture
+from .provider_node_summary import ProviderNodeSummary
+from .provider_region_node_summary import ProviderRegionNodeSummary
 from .provision_template_worker_pool_request import ProvisionTemplateWorkerPoolRequest
 from .provisioned_worker_pool import ProvisionedWorkerPool
 from .provisioned_worker_pool_properties import ProvisionedWorkerPoolProperties
@@ -273,6 +275,7 @@ from .user_portal_context import UserPortalContext
 from .user_search import UserSearch
 from .waterfall_provision_strategy import WaterfallProvisionStrategy
 from .work_requirement import WorkRequirement
+from .work_requirement_search import WorkRequirementSearch
 from .work_requirement_status import WorkRequirementStatus
 from .work_requirement_summary import WorkRequirementSummary
 from .worker import Worker
@@ -504,6 +507,8 @@ __all__ = [
     "PermissionDetail",
     "Price",
     "ProcessorArchitecture",
+    "ProviderNodeSummary",
+    "ProviderRegionNodeSummary",
     "ProvisionTemplateWorkerPoolRequest",
     "ProvisionedWorkerPool",
     "ProvisionedWorkerPoolProperties",
@@ -562,6 +567,7 @@ __all__ = [
     "UserSearch",
     "WaterfallProvisionStrategy",
     "WorkRequirement",
+    "WorkRequirementSearch",
     "WorkRequirementStatus",
     "WorkRequirementSummary",
     "Worker",

yellowdog_client/model/provider_node_summary.py

@@ -0,0 +1,12 @@
+from dataclasses import dataclass
+from typing import Optional
+
+from .cloud_provider import CloudProvider
+
+
+@dataclass
+class ProviderNodeSummary:
+    """A summary of a group of nodes."""
+    provider: Optional[CloudProvider] = None
+    totalNodes: Optional[int] = None
+    workingNodes: Optional[int] = None

yellowdog_client/model/provider_region_node_summary.py

@@ -0,0 +1,10 @@
+from dataclasses import dataclass
+from typing import Optional
+
+
+@dataclass
+class ProviderRegionNodeSummary:
+    """A summary of a group of nodes."""
+    region: Optional[str] = None
+    totalNodes: Optional[int] = None
+    workingNodes: Optional[int] = None

yellowdog_client/model/work_requirement_search.py

@@ -0,0 +1,18 @@
+from dataclasses import dataclass
+from typing import List, Optional
+
+from .instant_range import InstantRange
+from .sort_direction import SortDirection
+from .work_requirement_status import WorkRequirementStatus
+
+
+@dataclass
+class WorkRequirementSearch:
+    namespace: Optional[str] = None
+    name: Optional[str] = None
+    tag: Optional[str] = None
+    statuses: Optional[List[WorkRequirementStatus]] = None
+    isHealthy: Optional[bool] = None
+    createdTime: Optional[InstantRange] = None
+    sortField: Optional[str] = None
+    sortDirection: Optional[SortDirection] = None

yellowdog_client/scheduler/work_client.py

@@ -6,7 +6,7 @@ from typing import List
 from .work_requirement_helper import WorkRequirementHelper
 from yellowdog_client.common import Closeable, SearchClient
 from yellowdog_client.common.server_sent_events import SubscriptionEventListener
-from yellowdog_client.model import Slice, SliceReference, Task, TaskGroup, TaskSearch, WorkRequirement, WorkRequirementSummary
+from yellowdog_client.model import Slice, SliceReference, Task, TaskGroup, TaskSearch, WorkRequirement, WorkRequirementSearch, WorkRequirementSummary
 
 
 class WorkClient(ABC, Closeable):
@@ -197,6 +197,18 @@ class WorkClient(ABC, Closeable):
         Returns summaries of all existing work requirements within the system for the requesting user.
 
         :return: a list of work requirement summaries
+        @deprecated use {@link #getWorkRequirements(WorkRequirementSearch) instead to search tasks.
+        """
+
+        pass
+
+    @abstractmethod
+    def get_work_requirements(self, search: WorkRequirementSearch) -> SearchClient[WorkRequirementSummary]:
+        """
+        Returns a SearchClient that offers the ability to search work requirements.
+
+        :param search: the search
+        :return: the search client
         """
 
         pass

yellowdog_client/scheduler/work_client_impl.py

@@ -12,12 +12,14 @@ from yellowdog_client.model import TaskSearch
 from yellowdog_client.model import WorkRequirement
 from yellowdog_client.model import WorkRequirementStatus
 from yellowdog_client.model import WorkRequirementSummary
+from yellowdog_client.model import WorkRequirementSearch
 from .work_client import WorkClient
 from .work_requirement_helper import WorkRequirementHelper
 from .work_service_proxy import WorkServiceProxy
 from yellowdog_client.common import SearchClient
 
 
+
 class WorkClientImpl(WorkClient):
     def __init__(self, service_proxy: WorkServiceProxy) -> None:
         self.__service_proxy = service_proxy
@@ -71,7 +73,15 @@ class WorkClientImpl(WorkClient):
         return self.__service_proxy.transition_work_requirement(work_requirement_id, WorkRequirementStatus.CANCELLING)
 
     def find_all_work_requirements(self) -> List[WorkRequirementSummary]:
-        return self.__service_proxy.find_all_work_requirements()
+        search = WorkRequirementSearch()
+        return paginate(lambda sr: self._get_work_requirements_slice(search, sr))
+
+    def get_work_requirements(self, search: WorkRequirementSearch) -> SearchClient[WorkRequirementSummary]:
+        get_next_slice_function = lambda slice_reference: self._get_work_requirements_slice(search, slice_reference)
+        return SearchClient(get_next_slice_function)
+
+    def _get_work_requirements_slice(self, search: WorkRequirementSearch, slice_reference: SliceReference) -> Slice[WorkRequirementSummary]:
+        return self.__service_proxy.find_work_requirements(search, slice_reference)
 
     def add_work_requirement_listener(self, work_requirement: WorkRequirement, listener: SubscriptionEventListener[WorkRequirement]) -> None:
         self._check_has_id(work_requirement)

yellowdog_client/scheduler/work_service_proxy.py

@@ -4,6 +4,8 @@ from yellowdog_client.common import Proxy
 from yellowdog_client.model import WorkRequirement, WorkRequirementStatus, WorkRequirementSummary, Task, \
     TaskSearch, SliceReference, Slice
 
+from yellowdog_client.model import WorkRequirementSearch
+
 
 class WorkServiceProxy:
     def __init__(self, proxy: Proxy) -> None:
@@ -26,8 +28,10 @@ class WorkServiceProxy:
         url = "requirements/%s/transition/%s" % (requirement_id, str(next_status))
         return self._proxy.put(WorkRequirement, url=url)
 
-    def find_all_work_requirements(self) -> List[WorkRequirementSummary]:
-        return self._proxy.get(List[WorkRequirementSummary], "requirements")
+    def find_work_requirements(self, search: WorkRequirementSearch, slice_reference: SliceReference) -> Slice[WorkRequirementSummary]:
+        params = self._proxy.to_params(search, slice_reference)
+        params["sliced"] = "true"
+        return self._proxy.get(Slice[WorkRequirementSummary], "requirements", params)
 
     def stream_work_requirement_updates(self, requirement_id: str):
         return self._proxy.stream(f"requirements/{requirement_id}/updates")

{yellowdog_sdk-8.1.7.dist-info → yellowdog_sdk-8.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: yellowdog-sdk
-Version: 8.1.7
+Version: 8.1.9
 Summary: SDK for the YellowDog Platform
 Author-email: YellowDog Limited <support@yellowdog.co>
 Project-URL: Homepage, https://yellowdog.co

{yellowdog_sdk-8.1.7.dist-info → yellowdog_sdk-8.1.9.dist-info}/RECORD

@@ -1,6 +1,6 @@
 __init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 yellowdog_client/__init__.py,sha256=xHGTw5UbjkeEl_hC8_gJCacfji6462qJWD1nvJdFssE,13162
-yellowdog_client/_version.py,sha256=Fe7vbk642Dh860I7r7P9fBSyPWCbJX14uwPUG6_80gE,35
+yellowdog_client/_version.py,sha256=7oZ5haqHFA1WNA5I-pSeMOkupCNRF0JDvfTxfBHZmoQ,35
 yellowdog_client/client_collection.py,sha256=VSEzjf6iR1qCQ0YGLyDq_Kgvw8r832QDwTp6-MLB4Vs,388
 yellowdog_client/platform_client.py,sha256=h_9sd35e0GMdVGjSjG0KGcG3w1qLGY1Vf1U1gSor-HE,7052
 yellowdog_client/account/__init__.py,sha256=wx5GbsoODMdXbOmRoNZu5VYgDJX-RqsFb9ph14HXoV4,235
@@ -50,7 +50,7 @@ yellowdog_client/images/images_service_proxy.py,sha256=eyP4u6FoIH1WJvNM_m8tacTCs
 yellowdog_client/images/page.py,sha256=UIvlxvzdcfnKvbcq2Cn6IB7ZtQMc3dzcBTUfElvVPwQ,391
 yellowdog_client/images/pageable.py,sha256=msD8uGGJ2F5jEqTNDYaFrh6z6drlxOXZ1AmB0T3edM0,296
 yellowdog_client/images/sort.py,sha256=YS05DlIRg1Cm3QLBi6KFjFdB3g-b3WrqFFitlMJUEMM,167
-yellowdog_client/model/__init__.py,sha256=6MaYoWtGLPuPh-6C9a9TNYvBo7sNh_H3bBPDQf2rIPo,23769
+yellowdog_client/model/__init__.py,sha256=klJsMejlPvrWun1-wE-kigPjOwgnDeTrQe-Lv57eEIk,24040
 yellowdog_client/model/access_delegate.py,sha256=gZYX7Hqw_eBzBIHzbxyZfrhko0xZIsPCsf4gSzFalwc,568
 yellowdog_client/model/account.py,sha256=r_-7J-JjjMcQPZzaQ4_o9y6XY7-GMPRLWt9mGOucYU0,394
 yellowdog_client/model/account_allowance.py,sha256=c-MkSSRLB5Y5TMDZl5h5fipj4vzLLlO_AQXPeRJ0AYw,1021
@@ -269,6 +269,8 @@ yellowdog_client/model/permission.py,sha256=3V-0Ba97BSTjpEcpCQtfTPdeK8GSh7_hvo0X
 yellowdog_client/model/permission_detail.py,sha256=ukTbqQgmU782PZJ7i0H2KAojgmRG4flzwR2ywZICOOQ,207
 yellowdog_client/model/price.py,sha256=kPZw5VGPg5vKRXL5brX8eZrH_GePvlBhdfMsm5fF6Dc,194
 yellowdog_client/model/processor_architecture.py,sha256=hkbyplz2R5SeVtINaB3CcU86yR64elm5qi9W48LHVms,157
+yellowdog_client/model/provider_node_summary.py,sha256=maYsTKxTR7jNjZDr_BNs23rV91k_WL44Q3BVXGdKFSk,307
+yellowdog_client/model/provider_region_node_summary.py,sha256=n2VHXgPassz9zRrLoyGXOnpvO1oLnUE-ctfGkqkSQ4k,258
 yellowdog_client/model/provision_template_worker_pool_request.py,sha256=FEGUv-882NQJOF8KH22wQatzHwiz9_TPSUBWWSE2H-E,416
 yellowdog_client/model/provisioned_worker_pool.py,sha256=U00_pczMkTsu9QhioeRUub0srMJLW3ntGbTnYjLXVTA,1083
 yellowdog_client/model/provisioned_worker_pool_properties.py,sha256=ObTIEl06JBevzfW8metBmX2UlrjCBnKUTVvW2-F_Hbs,1481
@@ -327,6 +329,7 @@ yellowdog_client/model/user_portal_context.py,sha256=ad0GuMXA9hKobFXf4in91xcXkiP
 yellowdog_client/model/user_search.py,sha256=XHEx0_yIqZnvi5A0zgvAnEbZ3oBASjqcWgM9GroVZn0,285
 yellowdog_client/model/waterfall_provision_strategy.py,sha256=1mAIACbWcVrsy3Bjulgh3TaPx83WoqRM443-jQ9FgaQ,727
 yellowdog_client/model/work_requirement.py,sha256=4-kqhjS33HEjf62Z_GjcgVxHZ4Xa5c54TR1DIDgYPuY,1714
+yellowdog_client/model/work_requirement_search.py,sha256=4qh3tWWF4C5UOC7XUjSN3Zylw3Cr1j3mn38QXZcXIyg,578
 yellowdog_client/model/work_requirement_status.py,sha256=w_XCB-SDBJQjHnkM28UivdA53MVaAlDk0PzUcZyZYmk,1301
 yellowdog_client/model/work_requirement_summary.py,sha256=jEvsqI6IBykpP5_LAyoTiS5elHXfaQJ2Zj1FOwxQtMU,1374
 yellowdog_client/model/worker.py,sha256=KqUNIYwy9F-R4eY3Vk2JkZIyhERE-2CxmQe-6JgkI5Q,964
@@ -421,10 +424,10 @@ yellowdog_client/object_store/utils/memory_mapped_file_writter.py,sha256=8l9E_m8
 yellowdog_client/scheduler/__init__.py,sha256=ZORtc_LUBoTcplpoDhCHZNannI3qjeohghNcoBYU3ts,927
 yellowdog_client/scheduler/predicated_work_subscription_event_listener.py,sha256=mLfaVRmIMtQbj9u6d8O_B-VaZxOy5l1kuGCfTsS_xPU,2056
 yellowdog_client/scheduler/predicated_worker_pool_subscription_event_listener.py,sha256=z7tCniTF8WL6HSV7QcMOWFvC9EGSRopBGLODFOQinTA,2080
-yellowdog_client/scheduler/work_client.py,sha256=KqgE4Yi6yd0XjGtR_CCkVhMi2PJscp0mYTL-_QdwkrE,12348
-yellowdog_client/scheduler/work_client_impl.py,sha256=Jt6ddC5FNXRCSwZmlGXpZkBamYqVSjBGPqvHyT6h5oE,6652
+yellowdog_client/scheduler/work_client.py,sha256=Z0bz-SvjMiTDQ-fhGW3dSHTQrgipORJqPpDRDYiw98g,12792
+yellowdog_client/scheduler/work_client_impl.py,sha256=Pi8r7A8nvi6NUZUHlXXyV1IMxi3H4SaBNjvV1oHM3FU,7273
 yellowdog_client/scheduler/work_requirement_helper.py,sha256=vQMoI9emTMKLLmZSil3IGNUtxUl50MFuITJTv5Okvlo,4167
-yellowdog_client/scheduler/work_service_proxy.py,sha256=_sjFile_mV3BKkMaz3Ty_3vW78JeS5XgBkwDg1G3w1w,2694
+yellowdog_client/scheduler/work_service_proxy.py,sha256=ds0qiYHiOq9hdInjt1ynWE4GtifZlJEeh5HVDbKtTGc,2920
 yellowdog_client/scheduler/worker_pool_client.py,sha256=CDJKKpAy91VNn8udCQ5gFO4FSm-vqj8tZ7KbM-4Grc4,15021
 yellowdog_client/scheduler/worker_pool_client_impl.py,sha256=dSuPBDMioNz4DUa-YuJ9o4see2QV4NwPNMnV9OBni6Y,8371
 yellowdog_client/scheduler/worker_pool_helper.py,sha256=TZWm4x5S4atmjvZjgt7xhx8hcsnGI087WJrqllJDK-U,1340
@@ -433,8 +436,8 @@ yellowdog_client/usage/__init__.py,sha256=XQwRJqTdxKZa1QUTsxBEL0TqQJeQHGyPklFeqc
 yellowdog_client/usage/allowances_client.py,sha256=H6n63jXjT4OwuWJgFUXSjSmvGTZz9uspy3kj3upinaA,1337
 yellowdog_client/usage/allowances_client_impl.py,sha256=nQPnSzJKhL3WvyCn5fmiDkwE84xZryH9YvV5Z1GjU4M,2061
 yellowdog_client/usage/allowances_service_proxy.py,sha256=uO6LWnpjIzUcZTGdOxPXn7SyYX7NMRqO5KUiHUGr490,1320
-yellowdog_sdk-8.1.7.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-yellowdog_sdk-8.1.7.dist-info/METADATA,sha256=PuoEA0X4lz29RQn97LIjbrsIZWqHgLg6h1JTmWr44zw,3044
-yellowdog_sdk-8.1.7.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-yellowdog_sdk-8.1.7.dist-info/top_level.txt,sha256=6PH16DcoqpYHhQ5A0UJOjf0tg-1rTrNC9C2CLqCMuFo,26
-yellowdog_sdk-8.1.7.dist-info/RECORD,,
+yellowdog_sdk-8.1.9.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+yellowdog_sdk-8.1.9.dist-info/METADATA,sha256=-58o2-n0D2AxPqTCsT_FfE7f-SN5-2EB1mreVZBC5s4,3044
+yellowdog_sdk-8.1.9.dist-info/WHEEL,sha256=a7TGlA-5DaHMRrarXjVbQagU3Man_dCnGIWMJr5kRWo,91
+yellowdog_sdk-8.1.9.dist-info/top_level.txt,sha256=6PH16DcoqpYHhQ5A0UJOjf0tg-1rTrNC9C2CLqCMuFo,26
+yellowdog_sdk-8.1.9.dist-info/RECORD,,

{yellowdog_sdk-8.1.7.dist-info → yellowdog_sdk-8.1.9.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.1.0)
+Generator: setuptools (75.4.0)
 Root-Is-Purelib: true
 Tag: py3-none-any