Code Duplication    Length = 111-141 lines in 2 locations

gvm/protocols/gmpv7/gmpv7.py 1 location

@@ 1986-2126 (lines=141) @@
1983
        cmd.add_element("copy", target_id)
1984
        return self._send_xml_command(cmd)
1985
1986
    def create_task(
1987
        self,
1988
        name: str,
1989
        config_id: str,
1990
        target_id: str,
1991
        scanner_id: str,
1992
        *,
1993
        alterable: Optional[bool] = None,
1994
        hosts_ordering: Optional[HostsOrdering] = None,
1995
        schedule_id: Optional[str] = None,
1996
        alert_ids: Optional[List[str]] = None,
1997
        comment: Optional[str] = None,
1998
        schedule_periods: Optional[int] = None,
1999
        observers: Optional[List[str]] = None,
2000
        preferences: Optional[dict] = None
2001
    ) -> Any:
2002
        """Create a new task
2003
2004
        Arguments:
2005
            name: Name of the task
2006
            config_id: UUID of scan config to use by the task
2007
            target_id: UUID of target to be scanned
2008
            scanner_id: UUID of scanner to use for scanning the target
2009
            comment: Comment for the task
2010
            alterable: Whether the task should be alterable
2011
            alert_ids: List of UUIDs for alerts to be applied to the task
2012
            hosts_ordering: The order hosts are scanned in
2013
            schedule_id: UUID of a schedule when the task should be run.
2014
            schedule_periods: A limit to the number of times the task will be
2015
                scheduled, or 0 for no limit
2016
            observers: List of names or ids of users which should be allowed to
2017
                observe this task
2018
            preferences: Name/Value pairs of scanner preferences.
2019
2020
        Returns:
2021
            The response. See :py:meth:`send_command` for details.
2022
        """
2023
        if not name:
2024
            raise RequiredArgument(
2025
                function=self.create_task.__name__, argument='name'
2026
            )
2027
2028
        if not config_id:
2029
            raise RequiredArgument(
2030
                function=self.create_task.__name__, argument='config_id'
2031
            )
2032
2033
        if not target_id:
2034
            raise RequiredArgument(
2035
                function=self.create_task.__name__, argument='target_id'
2036
            )
2037
2038
        if not scanner_id:
2039
            raise RequiredArgument(
2040
                function=self.create_task.__name__, argument='scanner_id'
2041
            )
2042
2043
        # don't allow to create a container task with create_task
2044
        if target_id == '0':
2045
            raise InvalidArgument(
2046
                'Invalid argument {} for target_id'.format(target_id)
2047
            )
2048
2049
        cmd = XmlCommand("create_task")
2050
        cmd.add_element("name", name)
2051
        cmd.add_element("config", attrs={"id": config_id})
2052
        cmd.add_element("target", attrs={"id": target_id})
2053
        cmd.add_element("scanner", attrs={"id": scanner_id})
2054
2055
        if comment:
2056
            cmd.add_element("comment", comment)
2057
2058
        if alterable is not None:
2059
            cmd.add_element("alterable", _to_bool(alterable))
2060
2061
        if hosts_ordering:
2062
            if not isinstance(hosts_ordering, HostsOrdering):
2063
                raise InvalidArgumentType(
2064
                    function=self.create_task.__name__,
2065
                    argument='hosts_ordering',
2066
                    arg_type=HostsOrdering.__name__,
2067
                )
2068
            cmd.add_element("hosts_ordering", hosts_ordering.value)
2069
2070
        if alert_ids:
2071
            if isinstance(alert_ids, str):
2072
                deprecation(
2073
                    "Please pass a list as alert_ids parameter to create_task. "
2074
                    "Passing a string is deprecated and will be removed in "
2075
                    "future."
2076
                )
2077
2078
                # if a single id is given as a string wrap it into a list
2079
                alert_ids = [alert_ids]
2080
            if _is_list_like(alert_ids):
2081
                # parse all given alert id's
2082
                for alert in alert_ids:
2083
                    cmd.add_element("alert", attrs={"id": str(alert)})
2084
2085
        if schedule_id:
2086
            cmd.add_element("schedule", attrs={"id": schedule_id})
2087
2088
            if schedule_periods is not None:
2089
                if (
2090
                    not isinstance(schedule_periods, numbers.Integral)
2091
                    or schedule_periods < 0
2092
                ):
2093
                    raise InvalidArgument(
2094
                        "schedule_periods must be an integer greater or equal "
2095
                        "than 0"
2096
                    )
2097
                cmd.add_element("schedule_periods", str(schedule_periods))
2098
2099
        if observers is not None:
2100
            if not _is_list_like(observers):
2101
                raise InvalidArgumentType(
2102
                    function=self.create_task.__name__,
2103
                    argument='observers',
2104
                    arg_type='list',
2105
                )
2106
2107
            # gvmd splits by comma and space
2108
            # gvmd tries to lookup each value as user name and afterwards as
2109
            # user id. So both user name and user id are possible
2110
            cmd.add_element("observers", _to_comma_list(observers))
2111
2112
        if preferences is not None:
2113
            if not isinstance(preferences, collections.abc.Mapping):
2114
                raise InvalidArgumentType(
2115
                    function=self.create_task.__name__,
2116
                    argument='preferences',
2117
                    arg_type=collections.abc.Mapping.__name__,
2118
                )
2119
2120
            _xmlprefs = cmd.add_element("preferences")
2121
            for pref_name, pref_value in preferences.items():
2122
                _xmlpref = _xmlprefs.add_element("preference")
2123
                _xmlpref.add_element("scanner_name", pref_name)
2124
                _xmlpref.add_element("value", str(pref_value))
2125
2126
        return self._send_xml_command(cmd)
2127
2128
    def create_container_task(
2129
        self, name: str, *, comment: Optional[str] = None

gvm/protocols/gmpv9/gmpv9.py 1 location

@@ 1152-1262 (lines=111) @@
1149
1150
        return self._send_xml_command(cmd)
1151
1152
    def __create_task(
1153
        self,
1154
        name: str,
1155
        config_id: str,
1156
        target_id: str,
1157
        scanner_id: str,
1158
        usage_type: UsageType,
1159
        function: str,
1160
        *,
1161
        alterable: Optional[bool] = None,
1162
        hosts_ordering: Optional[HostsOrdering] = None,
1163
        schedule_id: Optional[str] = None,
1164
        alert_ids: Optional[List[str]] = None,
1165
        comment: Optional[str] = None,
1166
        schedule_periods: Optional[int] = None,
1167
        observers: Optional[List[str]] = None,
1168
        preferences: Optional[dict] = None
1169
    ) -> Any:
1170
        if not name:
1171
            raise RequiredArgument(function=function, argument='name')
1172
1173
        if not config_id:
1174
            raise RequiredArgument(function=function, argument='config_id')
1175
1176
        if not target_id:
1177
            raise RequiredArgument(function=function, argument='target_id')
1178
1179
        if not scanner_id:
1180
            raise RequiredArgument(function=function, argument='scanner_id')
1181
1182
        # don't allow to create a container task with create_task
1183
        if target_id == '0':
1184
            raise InvalidArgument(function=function, argument='target_id')
1185
1186
        cmd = XmlCommand("create_task")
1187
        cmd.add_element("name", name)
1188
        cmd.add_element("usage_type", usage_type.value)
1189
        cmd.add_element("config", attrs={"id": config_id})
1190
        cmd.add_element("target", attrs={"id": target_id})
1191
        cmd.add_element("scanner", attrs={"id": scanner_id})
1192
1193
        if comment:
1194
            cmd.add_element("comment", comment)
1195
1196
        if alterable is not None:
1197
            cmd.add_element("alterable", _to_bool(alterable))
1198
1199
        if hosts_ordering:
1200
            if not isinstance(hosts_ordering, self.types.HostsOrdering):
1201
                raise InvalidArgumentType(
1202
                    function=function,
1203
                    argument='hosts_ordering',
1204
                    arg_type=HostsOrdering.__name__,
1205
                )
1206
            cmd.add_element("hosts_ordering", hosts_ordering.value)
1207
1208
        if alert_ids:
1209
            if isinstance(alert_ids, str):
1210
                deprecation(
1211
                    "Please pass a list as alert_ids parameter to {}. "
1212
                    "Passing a string is deprecated and will be removed in "
1213
                    "future.".format(function)
1214
                )
1215
1216
                # if a single id is given as a string wrap it into a list
1217
                alert_ids = [alert_ids]
1218
            if _is_list_like(alert_ids):
1219
                # parse all given alert id's
1220
                for alert in alert_ids:
1221
                    cmd.add_element("alert", attrs={"id": str(alert)})
1222
1223
        if schedule_id:
1224
            cmd.add_element("schedule", attrs={"id": schedule_id})
1225
1226
            if schedule_periods is not None:
1227
                if (
1228
                    not isinstance(schedule_periods, numbers.Integral)
1229
                    or schedule_periods < 0
1230
                ):
1231
                    raise InvalidArgument(
1232
                        "schedule_periods must be an integer greater or equal "
1233
                        "than 0"
1234
                    )
1235
                cmd.add_element("schedule_periods", str(schedule_periods))
1236
1237
        if observers is not None:
1238
            if not _is_list_like(observers):
1239
                raise InvalidArgumentType(
1240
                    function=function, argument='observers', arg_type='list'
1241
                )
1242
1243
            # gvmd splits by comma and space
1244
            # gvmd tries to lookup each value as user name and afterwards as
1245
            # user id. So both user name and user id are possible
1246
            cmd.add_element("observers", _to_comma_list(observers))
1247
1248
        if preferences is not None:
1249
            if not isinstance(preferences, collections.abc.Mapping):
1250
                raise InvalidArgumentType(
1251
                    function=function,
1252
                    argument='preferences',
1253
                    arg_type=collections.abc.Mapping.__name__,
1254
                )
1255
1256
            _xmlprefs = cmd.add_element("preferences")
1257
            for pref_name, pref_value in preferences.items():
1258
                _xmlpref = _xmlprefs.add_element("preference")
1259
                _xmlpref.add_element("scanner_name", pref_name)
1260
                _xmlpref.add_element("value", str(pref_value))
1261
1262
        return self._send_xml_command(cmd)
1263
1264
    def __create_config(
1265
        self,