Skip to content

Commit 5d38591

Browse files
bjoernricksgreenbonebot
authored andcommitted
Add: Add an add_filter method to XmlCommand
Allow to set a filter on an XmlCommand instance. This allows to deprecated the add_filter utils function.
1 parent 6475ff7 commit 5d38591

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

gvm/xml.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,19 @@ class XmlCommand(XmlCommandElement):
8686
def __init__(self, name: str) -> None:
8787
super().__init__(create_element(name))
8888

89+
def add_filter(
90+
self,
91+
filter_string: Optional[str],
92+
filter_id: Optional[str],
93+
) -> "XmlCommand":
94+
if filter_string:
95+
self.set_attribute("filter", filter_string)
96+
97+
if filter_id:
98+
self.set_attribute("filt_id", filter_id)
99+
100+
return self
101+
89102

90103
def pretty_print(
91104
xml: Union[str, List[Union[Element, str]], Element],

0 commit comments

Comments
 (0)