@@ 138-226 (lines=89) @@ | ||
135 | ||
136 | return self._send_xml_command(cmd) |
|
137 | ||
138 | def create_override( |
|
139 | self, |
|
140 | text: str, |
|
141 | nvt_oid: str, |
|
142 | *, |
|
143 | days_active: Optional[int] = None, |
|
144 | hosts: Optional[List[str]] = None, |
|
145 | port: Optional[int] = None, |
|
146 | result_id: Optional[str] = None, |
|
147 | severity: Optional[Severity] = None, |
|
148 | new_severity: Optional[Severity] = None, |
|
149 | task_id: Optional[str] = None, |
|
150 | threat: Optional[SeverityLevel] = None, |
|
151 | new_threat: Optional[SeverityLevel] = None, |
|
152 | ) -> Any: |
|
153 | """Create a new override |
|
154 | ||
155 | Arguments: |
|
156 | text: Text of the new override |
|
157 | nvt_id: OID of the nvt to which override applies |
|
158 | days_active: Days override will be active. -1 on always, 0 off |
|
159 | hosts: A list of host addresses |
|
160 | port: Port to which the override applies |
|
161 | result_id: UUID of a result to which override applies |
|
162 | severity: Severity to which override applies |
|
163 | new_severity: New severity for result |
|
164 | task_id: UUID of task to which override applies |
|
165 | threat: Severity level to which override applies. Will be converted |
|
166 | to severity. |
|
167 | new_threat: New severity level for results. Will be converted to |
|
168 | new_severity. |
|
169 | ||
170 | Returns: |
|
171 | The response. See :py:meth:`send_command` for details. |
|
172 | """ |
|
173 | if not text: |
|
174 | raise RequiredArgument( |
|
175 | function=self.create_override.__name__, argument='text' |
|
176 | ) |
|
177 | ||
178 | if not nvt_oid: |
|
179 | raise RequiredArgument( |
|
180 | function=self.create_override.__name__, argument='nvt_oid' |
|
181 | ) |
|
182 | ||
183 | cmd = XmlCommand("create_override") |
|
184 | cmd.add_element("text", text) |
|
185 | cmd.add_element("nvt", attrs={"oid": nvt_oid}) |
|
186 | ||
187 | if days_active is not None: |
|
188 | cmd.add_element("active", str(days_active)) |
|
189 | ||
190 | if hosts: |
|
191 | cmd.add_element("hosts", to_comma_list(hosts)) |
|
192 | ||
193 | if port: |
|
194 | cmd.add_element("port", str(port)) |
|
195 | ||
196 | if result_id: |
|
197 | cmd.add_element("result", attrs={"id": result_id}) |
|
198 | ||
199 | if severity: |
|
200 | cmd.add_element("severity", str(severity)) |
|
201 | ||
202 | if new_severity: |
|
203 | cmd.add_element("new_severity", str(new_severity)) |
|
204 | ||
205 | if task_id: |
|
206 | cmd.add_element("task", attrs={"id": task_id}) |
|
207 | ||
208 | if threat is not None: |
|
209 | deprecation( |
|
210 | "The threat parameter has been removed in GMP" |
|
211 | " version {}{}".format( |
|
212 | self.get_protocol_version()[0], |
|
213 | self.get_protocol_version()[1], |
|
214 | ) |
|
215 | ) |
|
216 | ||
217 | if new_threat is not None: |
|
218 | deprecation( |
|
219 | "The new_threat parameter has been removed in GMP" |
|
220 | " version {}{}".format( |
|
221 | self.get_protocol_version()[0], |
|
222 | self.get_protocol_version()[1], |
|
223 | ) |
|
224 | ) |
|
225 | ||
226 | return self._send_xml_command(cmd) |
|
227 | ||
228 | def modify_note( |
|
229 | self, |
|
@@ 301-389 (lines=89) @@ | ||
298 | ||
299 | return self._send_xml_command(cmd) |
|
300 | ||
301 | def modify_override( |
|
302 | self, |
|
303 | override_id: str, |
|
304 | text: str, |
|
305 | *, |
|
306 | days_active: Optional[int] = None, |
|
307 | hosts: Optional[List[str]] = None, |
|
308 | port: Optional[int] = None, |
|
309 | result_id: Optional[str] = None, |
|
310 | severity: Optional[Severity] = None, |
|
311 | new_severity: Optional[Severity] = None, |
|
312 | task_id: Optional[str] = None, |
|
313 | threat: Optional[SeverityLevel] = None, |
|
314 | new_threat: Optional[SeverityLevel] = None, |
|
315 | ) -> Any: |
|
316 | """Modifies an existing override. |
|
317 | ||
318 | Arguments: |
|
319 | override_id: UUID of override to modify. |
|
320 | text: The text of the override. |
|
321 | days_active: Days override will be active. -1 on always, |
|
322 | 0 off. |
|
323 | hosts: A list of host addresses |
|
324 | port: Port to which override applies. |
|
325 | result_id: Result to which override applies. |
|
326 | severity: Severity to which override applies. |
|
327 | new_severity: New severity score for result. |
|
328 | task_id: Task to which override applies. |
|
329 | threat: Threat level to which override applies. |
|
330 | Will be converted to severity. |
|
331 | new_threat: New threat level for results. Will be converted to |
|
332 | new_severity. |
|
333 | ||
334 | Returns: |
|
335 | The response. See :py:meth:`send_command` for details. |
|
336 | """ |
|
337 | if not override_id: |
|
338 | raise RequiredArgument( |
|
339 | function=self.modify_override.__name__, argument='override_id' |
|
340 | ) |
|
341 | if not text: |
|
342 | raise RequiredArgument( |
|
343 | function=self.modify_override.__name__, argument='text' |
|
344 | ) |
|
345 | ||
346 | cmd = XmlCommand("modify_override") |
|
347 | cmd.set_attribute("override_id", override_id) |
|
348 | cmd.add_element("text", text) |
|
349 | ||
350 | if days_active is not None: |
|
351 | cmd.add_element("active", str(days_active)) |
|
352 | ||
353 | if hosts: |
|
354 | cmd.add_element("hosts", to_comma_list(hosts)) |
|
355 | ||
356 | if port: |
|
357 | cmd.add_element("port", str(port)) |
|
358 | ||
359 | if result_id: |
|
360 | cmd.add_element("result", attrs={"id": result_id}) |
|
361 | ||
362 | if severity: |
|
363 | cmd.add_element("severity", str(severity)) |
|
364 | ||
365 | if new_severity: |
|
366 | cmd.add_element("new_severity", str(new_severity)) |
|
367 | ||
368 | if task_id: |
|
369 | cmd.add_element("task", attrs={"id": task_id}) |
|
370 | ||
371 | if threat is not None: |
|
372 | deprecation( |
|
373 | "The threat parameter has been removed in GMP" |
|
374 | " version {}{}".format( |
|
375 | self.get_protocol_version()[0], |
|
376 | self.get_protocol_version()[1], |
|
377 | ) |
|
378 | ) |
|
379 | ||
380 | if new_threat is not None: |
|
381 | deprecation( |
|
382 | "The new_threat parameter has been removed in GMP" |
|
383 | " version {}{}".format( |
|
384 | self.get_protocol_version()[0], |
|
385 | self.get_protocol_version()[1], |
|
386 | ) |
|
387 | ) |
|
388 | ||
389 | return self._send_xml_command(cmd) |
|
390 | ||
391 | def modify_user( |
|
392 | self, |