diff --git a/src/pytmv1/api/alert.py b/src/pytmv1/api/alert.py index 4bf17b0..56c077d 100644 --- a/src/pytmv1/api/alert.py +++ b/src/pytmv1/api/alert.py @@ -121,6 +121,8 @@ def consume( consumer: Callable[[Union[SaeAlert, TiAlert]], None], start_time: Optional[str] = None, end_time: Optional[str] = None, + op: QueryOp = QueryOp.AND, + **fields: str, ) -> Result[ConsumeLinkableResp]: """Retrieves and consume workbench alerts. @@ -134,6 +136,11 @@ def consume( time range (yyyy-MM-ddThh:mm:ssZ). Defaults to the time the request is made. :type end_time: Optional[str] + :param op: Operator to apply between fields (ie: ... OR ...). + :type op: QueryOp + :param fields: Field/value used to filter result (i.e:fileName="1.sh"), + check Vision One API documentation for full list of supported fields. + :type fields: Dict[str, str] :rtype: Result[ConsumeLinkableResp]: """ return self._core.send_linkable( @@ -147,4 +154,5 @@ def consume( "orderBy": "createdDateTime desc", } ), + headers=utils.tmv1_filter(op, fields), ) diff --git a/src/pytmv1/model/common.py b/src/pytmv1/model/common.py index ae7290f..e4a58e5 100644 --- a/src/pytmv1/model/common.py +++ b/src/pytmv1/model/common.py @@ -33,7 +33,7 @@ Status, ) -CFG = ConfigDict(alias_generator=to_camel, populate_by_name=True) +CFG = ConfigDict(alias_generator=to_camel, populate_by_name=True, protected_namespaces=()) class BaseModel(PydanticBaseModel):