1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
|
# coding: utf-8
"""
Hatchet API
The Hatchet API
The version of the OpenAPI document: 1.0.0
Generated by OpenAPI Generator (https://openapi-generator.tech)
Do not edit the class manually.
""" # noqa: E501
from __future__ import annotations
import json
import pprint
import re # noqa: F401
from datetime import datetime
from typing import Any, ClassVar, Dict, List, Optional, Set
from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
from typing_extensions import Annotated, Self
from hatchet_sdk.clients.rest.models.api_resource_meta import APIResourceMeta
from hatchet_sdk.clients.rest.models.recent_step_runs import RecentStepRuns
from hatchet_sdk.clients.rest.models.semaphore_slots import SemaphoreSlots
from hatchet_sdk.clients.rest.models.worker_label import WorkerLabel
from hatchet_sdk.clients.rest.models.worker_runtime_info import WorkerRuntimeInfo
from hatchet_sdk.clients.rest.models.worker_type import WorkerType
class Worker(BaseModel):
"""
Worker
""" # noqa: E501
metadata: APIResourceMeta
name: StrictStr = Field(description="The name of the worker.")
type: WorkerType
last_heartbeat_at: Optional[datetime] = Field(
default=None,
description="The time this worker last sent a heartbeat.",
alias="lastHeartbeatAt",
)
last_listener_established: Optional[datetime] = Field(
default=None,
description="The time this worker last sent a heartbeat.",
alias="lastListenerEstablished",
)
actions: Optional[List[StrictStr]] = Field(
default=None, description="The actions this worker can perform."
)
slots: Optional[List[SemaphoreSlots]] = Field(
default=None, description="The semaphore slot state for the worker."
)
recent_step_runs: Optional[List[RecentStepRuns]] = Field(
default=None,
description="The recent step runs for the worker.",
alias="recentStepRuns",
)
status: Optional[StrictStr] = Field(
default=None, description="The status of the worker."
)
max_runs: Optional[StrictInt] = Field(
default=None,
description="The maximum number of runs this worker can execute concurrently.",
alias="maxRuns",
)
available_runs: Optional[StrictInt] = Field(
default=None,
description="The number of runs this worker can execute concurrently.",
alias="availableRuns",
)
dispatcher_id: Optional[
Annotated[str, Field(min_length=36, strict=True, max_length=36)]
] = Field(
default=None,
description="the id of the assigned dispatcher, in UUID format",
alias="dispatcherId",
)
labels: Optional[List[WorkerLabel]] = Field(
default=None, description="The current label state of the worker."
)
webhook_url: Optional[StrictStr] = Field(
default=None, description="The webhook URL for the worker.", alias="webhookUrl"
)
webhook_id: Optional[StrictStr] = Field(
default=None, description="The webhook ID for the worker.", alias="webhookId"
)
runtime_info: Optional[WorkerRuntimeInfo] = Field(default=None, alias="runtimeInfo")
__properties: ClassVar[List[str]] = [
"metadata",
"name",
"type",
"lastHeartbeatAt",
"lastListenerEstablished",
"actions",
"slots",
"recentStepRuns",
"status",
"maxRuns",
"availableRuns",
"dispatcherId",
"labels",
"webhookUrl",
"webhookId",
"runtimeInfo",
]
@field_validator("status")
def status_validate_enum(cls, value):
"""Validates the enum"""
if value is None:
return value
if value not in set(["ACTIVE", "INACTIVE", "PAUSED"]):
raise ValueError(
"must be one of enum values ('ACTIVE', 'INACTIVE', 'PAUSED')"
)
return value
model_config = ConfigDict(
populate_by_name=True,
validate_assignment=True,
protected_namespaces=(),
)
def to_str(self) -> str:
"""Returns the string representation of the model using alias"""
return pprint.pformat(self.model_dump(by_alias=True))
def to_json(self) -> str:
"""Returns the JSON representation of the model using alias"""
# TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
return json.dumps(self.to_dict())
@classmethod
def from_json(cls, json_str: str) -> Optional[Self]:
"""Create an instance of Worker from a JSON string"""
return cls.from_dict(json.loads(json_str))
def to_dict(self) -> Dict[str, Any]:
"""Return the dictionary representation of the model using alias.
This has the following differences from calling pydantic's
`self.model_dump(by_alias=True)`:
* `None` is only added to the output dict for nullable fields that
were set at model initialization. Other fields with value `None`
are ignored.
"""
excluded_fields: Set[str] = set([])
_dict = self.model_dump(
by_alias=True,
exclude=excluded_fields,
exclude_none=True,
)
# override the default output from pydantic by calling `to_dict()` of metadata
if self.metadata:
_dict["metadata"] = self.metadata.to_dict()
# override the default output from pydantic by calling `to_dict()` of each item in slots (list)
_items = []
if self.slots:
for _item_slots in self.slots:
if _item_slots:
_items.append(_item_slots.to_dict())
_dict["slots"] = _items
# override the default output from pydantic by calling `to_dict()` of each item in recent_step_runs (list)
_items = []
if self.recent_step_runs:
for _item_recent_step_runs in self.recent_step_runs:
if _item_recent_step_runs:
_items.append(_item_recent_step_runs.to_dict())
_dict["recentStepRuns"] = _items
# override the default output from pydantic by calling `to_dict()` of each item in labels (list)
_items = []
if self.labels:
for _item_labels in self.labels:
if _item_labels:
_items.append(_item_labels.to_dict())
_dict["labels"] = _items
# override the default output from pydantic by calling `to_dict()` of runtime_info
if self.runtime_info:
_dict["runtimeInfo"] = self.runtime_info.to_dict()
return _dict
@classmethod
def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
"""Create an instance of Worker from a dict"""
if obj is None:
return None
if not isinstance(obj, dict):
return cls.model_validate(obj)
_obj = cls.model_validate(
{
"metadata": (
APIResourceMeta.from_dict(obj["metadata"])
if obj.get("metadata") is not None
else None
),
"name": obj.get("name"),
"type": obj.get("type"),
"lastHeartbeatAt": obj.get("lastHeartbeatAt"),
"lastListenerEstablished": obj.get("lastListenerEstablished"),
"actions": obj.get("actions"),
"slots": (
[SemaphoreSlots.from_dict(_item) for _item in obj["slots"]]
if obj.get("slots") is not None
else None
),
"recentStepRuns": (
[RecentStepRuns.from_dict(_item) for _item in obj["recentStepRuns"]]
if obj.get("recentStepRuns") is not None
else None
),
"status": obj.get("status"),
"maxRuns": obj.get("maxRuns"),
"availableRuns": obj.get("availableRuns"),
"dispatcherId": obj.get("dispatcherId"),
"labels": (
[WorkerLabel.from_dict(_item) for _item in obj["labels"]]
if obj.get("labels") is not None
else None
),
"webhookUrl": obj.get("webhookUrl"),
"webhookId": obj.get("webhookId"),
"runtimeInfo": (
WorkerRuntimeInfo.from_dict(obj["runtimeInfo"])
if obj.get("runtimeInfo") is not None
else None
),
}
)
return _obj
|
