Code Duplication    Length = 89-89 lines in 2 locations

gvm/protocols/gmpv214/gmpv214.py 2 locations

@@ 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,