Skip to content

Commit 84c1621

Browse files
committed
feat(schema): support labels in workers query interface
1 parent 8f2a12f commit 84c1621

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed

src/pynetmito/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353

5454
def get_logger():
5555
formatter = logging.Formatter(
56-
"%(asctime)s - %(levelname)s - %(message)s", datefmt="%Y-%m-%d %H:%M:%S"
56+
"%(asctime)s %(levelname)s %(name)s: %(message)s", datefmt="%Y-%m-%d %H:%M:%S"
5757
)
5858

5959
logger = logging.getLogger("pynetmito")

src/pynetmito/schemas.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -669,6 +669,7 @@ class WorkerQueryInfo(BaseAPIModel):
669669
worker_id: UUID4
670670
creator_username: str
671671
tags: list[str]
672+
labels: list[str]
672673
created_at: datetime
673674
updated_at: datetime
674675
state: WorkerState
@@ -712,6 +713,7 @@ class WorkersQueryReq(BaseAPIModel):
712713
group_name: Optional[str] = Field(default=None)
713714
role: Optional[Set[GroupWorkerRole]] = Field(default=None)
714715
tags: Optional[Set[str]] = Field(default=None)
716+
labels: Optional[Set[str]] = Field(default=None)
715717
creator_username: Optional[str] = Field(default=None)
716718
count: bool = Field(default=False)
717719

@@ -733,6 +735,15 @@ def serialize_tags(self, tags: Optional[Set[str]]):
733735
def deserialize_tags(cls, tags: Optional[list[str]]):
734736
return set(tags) if tags else None
735737

738+
@field_serializer("labels")
739+
def serialize_labels(self, labels: Optional[Set[str]]):
740+
return list(labels) if labels else None
741+
742+
@field_validator("labels", mode="before")
743+
@classmethod
744+
def deserialize_labels(cls, labels: Optional[list[str]]):
745+
return set(labels) if labels else None
746+
736747

737748
class WorkersQueryResp(BaseAPIModel):
738749
count: NonNegativeInt

0 commit comments

Comments
 (0)