File tree Expand file tree Collapse file tree 1 file changed +7
-10
lines changed
packages/service-library/src/servicelib/celery Expand file tree Collapse file tree 1 file changed +7
-10
lines changed Original file line number Diff line number Diff line change 1616TaskUUID : TypeAlias = UUID
1717_TASK_ID_KEY_DELIMITATOR : Final [str ] = ":"
1818_WILDCARD : Final [str ] = "*"
19+ _FORBIDDEN_CHARS = (_WILDCARD , _TASK_ID_KEY_DELIMITATOR , "=" )
1920
2021
2122class 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 :
You can’t perform that action at this time.
0 commit comments