Skip to content

Commit f81dccd

Browse files
committed
cleanup @pcrespov
1 parent 6a9b046 commit f81dccd

File tree

1 file changed

+7
-10
lines changed
  • packages/service-library/src/servicelib/celery

1 file changed

+7
-10
lines changed

packages/service-library/src/servicelib/celery/models.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
TaskUUID: TypeAlias = UUID
1717
_TASK_ID_KEY_DELIMITATOR: Final[str] = ":"
1818
_WILDCARD: Final[str] = "*"
19+
_FORBIDDEN_CHARS = (_WILDCARD, _TASK_ID_KEY_DELIMITATOR, "=")
1920

2021

2122
class Wildcard: ...
@@ -50,17 +51,13 @@ class MyTaskFilter(TaskFilter):
5051

5152
@model_validator(mode="after")
5253
def _check_valid_filters(self) -> Self:
53-
for key in self.model_dump().keys():
54-
if _WILDCARD in key or _TASK_ID_KEY_DELIMITATOR in key or "=" in key:
54+
for key, value in self.model_dump().items():
55+
# forbidden keys
56+
if any(x in key for x in _FORBIDDEN_CHARS):
5557
raise ValueError(f"Invalid filter key: '{key}'")
56-
if (
57-
_WILDCARD in f"{getattr(self, key)}"
58-
or _TASK_ID_KEY_DELIMITATOR in f"{getattr(self, key)}"
59-
or "=" in f"{getattr(self, key)}"
60-
):
61-
raise ValueError(
62-
f"Invalid filter value for key '{key}': '{getattr(self, key)}'"
63-
)
58+
# forbidden values
59+
if any(x in f"{value}" for x in _FORBIDDEN_CHARS):
60+
raise ValueError(f"Invalid filter value for key '{key}': '{value}'")
6461
return self
6562

6663
def _build_task_id_prefix(self) -> str:

0 commit comments

Comments
 (0)