gmp.xml._GmpCommandFactory.create_schedule_command() - Code Metrics - Inspection of "Convert create_*_command() methods to use the new..." - greenbone/gvm-tools - Measure and Improve Code Quality continuously with Scrutinizer

Passed
Pull Request — master (#116)

by Juan José
created 2018-10-30 07:15 UTC
_GmpCommandFactory.create_schedule_command() B

↳ Parent: gmp.xml
Complexity

Conditions
Size

Total Lines	46
Code Lines	36
Duplication

Lines	0
Ratio	0 %
Importance

Changes
Metric	Value
cc	8
eloc	36
nop	3
dl	0
loc	46
rs	7.1493
c	0
b	0
f	0
# -*- coding: utf-8 -*-
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
# Copyright (C) 2018 Greenbone Networks GmbH
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import defusedxml.lxml as secET

from lxml import etree

FILTER_NAMES = [
    'Agent',
    'Alert',
    'Asset',
    'Config',
    'Credential',
    'Filter',
    'Group',
    'Note',
    'Override',
    'Permission',
    'Port List',
    'Report',
    'Report Format',
    'Result',
    'Role',
    'Schedule',
    'SecInfo',
    'Tag',
    'Target',
    'Task',
    'User',
]

class XmlCommandElement:
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

    def __init__(self, element):
        self._element = element

    def add_element(self, name, text=None, attrs=None):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        node = etree.SubElement(self._element, name, attrib=attrs)

        node.text = text
        return XmlCommandElement(node)

    def set_attribute(self, name, value):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        self._element.set(name, value)

    def append_xml_str(self, xml_text):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        node = secET.fromstring(xml_text)
        self._element.append(node)

    def to_string(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return etree.tostring(self._element).decode('utf-8')


    def __str__(self):
        return self.to_string()


class XmlCommand(XmlCommandElement):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

    def __init__(self, name):
        super().__init__(etree.Element(name))



class _GmpCommandFactory:


    """Factory to create gmp - Greenbone Manangement Protocol - commands
    """

    def create_agent_command(self, installer, signature, name, comment='',
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
                             copy='', howto_install='', howto_use=''):

        cmd = XmlCommand('create_agent')
        cmd.add_element('installer', installer)
        cmd.add_element('signature', signature)
        cmd.add_element('name', name)

        if comment:
            cmd.add_element('comment', comment)

        if copy:
            cmd.add_element('copy', copy)

        if howto_install:
            cmd.add_element('howto_install', howto_install)

        if howto_use:
            cmd.add_element('howto_use', howto_use)

        return cmd.to_string()

    def create_alert_command(self, name, condition, event, method, filter_id='',
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
                             copy='', comment=''):

        cmd = XmlCommand('create_alert')
        cmd.add_element('name', name)

        if len(condition) > 1:
            conditions = cmd.add_element('condition', condition[0])
            for value, key in condition[1].items():
                _data = conditions.add_element('data', value)
                _data.add_element('name', key)

        elif condition[0] == "Always":
            conditions = cmd.add_element('condition', condition[0])

        if len(event) > 1:
            events = cmd.add_element('event', event[0])
            for value, key in event[1].items():
                _data = events.add_element('data', value)
                _data.add_element('name', key)

        if len(method) > 1:
            methods = cmd.add_element('method', method[0])
            for value, key in method[1].items():
                _data = methods.add_element('data', value)
                _data.add_element('name', key)

        if filter_id:
            cmd.add_element('filter', attrs={'id': filter_id})

        if copy:
            cmd.add_element('copy', copy)

        if comment:
            cmd.add_element('comment', comment)

        return cmd.to_string()

    def create_asset_command(self, name, asset_type, comment=''):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if asset_type not in ('host', 'os'):
            raise ValueError('create_asset requires asset_type to be either '
                             'host or os')
        cmd = XmlCommand('create_asset')
        asset = cmd.add_element('asset')
        asset.add_element('type', asset_type)
        asset.add_element('name', name)

        if comment:
            asset.add_element('comment', comment)

        return cmd.to_string()

    def create_authenticate_command(self, username, password):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates string for authentification on gvmd

        Creates the gmp authentication xml string.
        Inserts the username and password into it.

        Keyword Arguments:
            username {str} -- Username for GVM User
            password {str} -- Password for GVM User
        """
        cmd = XmlCommand('authenticate')

        credentials = cmd.add_element('credentials')
        credentials.add_element('username', username)
        credentials.add_element('password', password)

        return cmd.to_string()

    def create_config_command(self, copy_id, name):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create config on gvmd."""
        cmd = XmlCommand('create_config')
        cmd.add_element('copy', copy_id)
        cmd.add_element('name', name)

        return cmd.to_string()

    def create_credential_command(self, name, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create credential on gvmd."""
        cmd = XmlCommand('create_credential')
        cmd.add_element('name', name)

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        allow_insecure = kwargs.get('allow_insecure', '')
        if allow_insecure:
            cmd.add_element('allow_insecure', allow_insecure)

        certificate = kwargs.get('certificate', '')
        if certificate:
            cmd.add_element('certificate', certificate)

        key = kwargs.get('key', '')
        if key:
            phrase = key['phrase']
            private = key['private']
            if not phrase:
                raise ValueError('create_credential requires a phrase element')
            if not private:
                raise ValueError('create_credential requires a '
                                 'private element')

            _xmlKey = cmd.add_element('key')

            _xmlKey.add_element('phrase', phrase)
            _xmlKey.add_element('private', private)

        login = kwargs.get('login', '')
        if login:
            cmd.add_element('login', login)

        password = kwargs.get('password', '')
        if password:
            cmd.add_element('password', password)

        auth_algorithm = kwargs.get('auth_algorithm', '')
        if auth_algorithm:
            if auth_algorithm not in ('md5', 'sha1'):
                raise ValueError('create_credential requires auth_algorithm '
                                 'to be either md5 or sha1')
            cmd.add_element('auth_algorithm', auth_algorithm)

        community = kwargs.get('community', '')
        if community:
            cmd.add_element('community', community)

        privacy = kwargs.get('privacy', '')
        if privacy:
            algorithm = privacy.algorithm
            if algorithm not in ('aes', 'des'):
                raise ValueError('create_credential requires algorithm '
                                 'to be either aes or des')
            p_password = privacy.password
            _xmlPrivacy = cmd.add_element('privacy')

            _xmlPrivacy.add_element('algorithm', algorithm)
            _xmlPrivacy.add_element('password', p_password)

        cred_type = kwargs.get('type', '')
        if cred_type:
            if cred_type not in ('cc', 'snmp', 'up', 'usk'):
                raise ValueError('create_credential requires type '
                                 'to be either cc, snmp, up or usk')
            cmd.add_element('type', cred_type)

        return cmd.to_string()

    def create_filter_command(self, name, make_unique, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create filter on gvmd."""

        cmd = XmlCommand('create_filter')
        _xmlName = cmd.add_element('name', name)

        if make_unique:
            _xmlName.add_element('make_unique', '1')
        else:
            _xmlName.add_element('make_unique', '0')

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        term = kwargs.get('term', '')
        if term:
            cmd.add_element('term', term)

        filter_type = kwargs.get('type', '')
        if filter_type:
            if filter_type not in FILTER_NAMES:
                raise ValueError('create_filter requires type '
                                 'to be either cc, snmp, up or usk')
            cmd.add_element('type', filter_type)

        return cmd.to_string()

    def create_group_command(self, name, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create group on gvmd."""

        cmd = XmlCommand('create_group')
        cmd.add_element('name', name)

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        special = kwargs.get('special', '')
        if special:
            _xmlSpecial = cmd.add_element('specials')

            _xmlSpecial.add_element('full')

        users = kwargs.get('users', '')
        if users:
            cmd.add_element('users', users)

        return cmd.to_string()

    def create_note_command(self, text, nvt_oid, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create note on gvmd."""

        cmd = XmlCommand('create_note')
        cmd.add_element('text', text)
        cmd.add_element('nvt', attrs={"oid": nvt_oid})

        active = kwargs.get('active', '')
        if active:
            cmd.add_element('active', active)

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        hosts = kwargs.get('hosts', '')
        if hosts:
            cmd.add_element('hosts', hosts)

        port = kwargs.get('port', '')
        if port:
           cmd.add_element('port', port)


        result_id = kwargs.get('result_id', '')
        if result_id:
            cmd.add_element('result', attrs={'id': result_id})

        severity = kwargs.get('severity', '')
        if severity:
            cmd.add_element('severity', severity)

        task_id = kwargs.get('task_id', '')
        if task_id:
            cmd.add_element('task', attrs={'id': task_id})

        threat = kwargs.get('threat', '')
        if threat:
            cmd.add_element('threat', threat)

        return cmd.to_string()

    def create_override_command(self, text, nvt_oid, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create override on gvmd."""

        cmd = XmlCommand('create_override')
        cmd.add_element('text', text)
        cmd.add_element('nvt', attrs={'oid': nvt_oid})

        active = kwargs.get('active', '')
        if active:
            cmd.add_element('active', active)

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        hosts = kwargs.get('hosts', '')
        if hosts:
            cmd.add_element('hosts', hosts)

        port = kwargs.get('port', '')
        if port:
            cmd.add_element('port', port)

        result_id = kwargs.get('result_id', '')
        if result_id:
            cmd.add_element('result', attrs={'id': result_id})

        severity = kwargs.get('severity', '')
        if severity:
            cmd.add_element('severity', severity)

        new_severity = kwargs.get('new_severity', '')
        if new_severity:
            cmd.add_element('new_severity', new_severity)

        task_id = kwargs.get('task_id', '')
        if task_id:
            cmd.add_element('task', attrs={'id': task_id})

        threat = kwargs.get('threat', '')
        if threat:
            cmd.add_element('threat', threat)

        new_threat = kwargs.get('new_threat', '')
        if new_threat:
            cmd.add_element('new_threat', new_threat)

        return cmd.to_string()

    def create_permission_command(self, name, subject_id, type, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        # pretty(gmp.create_permission('get_version',
        # 'cc9cac5e-39a3-11e4-abae-406186ea4fc5', 'role'))
        # libs.gvm_connection.GMPError: Error in NAME
        # TODO: Research why!!


        if not name:
            raise ValueError('create_permission requires a name element')
        if not subject_id:
            raise ValueError('create_permission requires a subject_id element')
        if type not in ('user', 'group', 'role'):
            raise ValueError('create_permission requires type '
                             'to be either user, group or role')

        cmd = XmlCommand('create_permission')
        cmd.add_element('name', name)
        _xmlSubject = cmd.add_element('subject', attrs={'id': subject_id})

        _xmlSubject.add_element('type', type)

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        resource = kwargs.get('resource', '')
        if resource:
            resource_id = resource.id
            resource_type = resource.type
            _xmlResource = cmd.add_element('resource',

                                            attrs={'id': resource_id})

            _xmlResource.add_element('type', resource_type)

        return cmd.to_string()

    def create_port_list_command(self, name, port_range, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create port list on gvmd."""
        if not name:
            raise ValueError('create_port_list requires a name element')
        if not port_range:
            raise ValueError('create_port_list requires a port_range element')

        cmd = XmlCommand('create_port_list')
        cmd.add_element('name', name)
        cmd.add_element('port_range', port_range)

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        return cmd.to_string()

    def create_port_range_command(self, port_list_id, start, end, type,
class Foo:
    def some_method(self, x, y):
        return x + y;
                                  comment=''):
        """Generates xml string for create port range on gvmd."""

        if not port_list_id:
            raise ValueError('create_port_range requires '
                             'a port_list_id element')
        if not type:
            raise ValueError('create_port_range requires a type element')

        cmd.add_element('create_port_range')

        cmd.add_element('port_list', attrs={'id': port_list_id})

        cmd.add_element('start', start)

        cmd.add_element('end', end)

        cmd.add_element('type', type)


        if len(comment):

            cmd.add_element('comment', comment)


        return cmd.to_string()


    def create_report_command(self, report_xml_string, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create report on gvmd."""

        if not report_xml_string:
            raise ValueError('create_report requires a report')

        task_id = kwargs.get('task_id', '')
        task_name = kwargs.get('task_name', '')

        cmd = XmlCommand('create_report')
        comment = kwargs.get('comment', '')
        if task_id:
            cmd.add_element('task', attrs={'id': task_id})
        elif task_name:
            _xmlTask = cmd.add_element('task')

            _xmlTask.add_element('name', task_name)
            if comment:
                _xmlTask.add_element('comment', comment)
        else:
            raise ValueError('create_report requires an id or name for a task')

        in_assets = kwargs.get('in_assets', '')
        if in_assets:
            cmd.add_element('in_assets', in_assets)

        cmd.append_xml_str(report_xml_string)

        return cmd.to_string()

    def create_role_command(self, name, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create role on gvmd."""

        if not name:
            raise ValueError('create_role requires a name element')

        cmd = XmlCommand('create_role')
        cmd.add_element('name', name)

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        users = kwargs.get('users', '')
        if users:
            cmd.add_element('users', users)

        return cmd.to_string()

    def create_scanner_command(self, name, host, port, type, ca_pub,
class Foo:
    def some_method(self, x, y):
        return x + y;
                               credential_id, kwargs):
        """Generates xml string for create scanner on gvmd."""
        if not name:
            raise ValueError('create_scanner requires a name element')
        if not host:
            raise ValueError('create_scanner requires a host element')
        if not port:
            raise ValueError('create_scanner requires a port element')
        if not type:
            raise ValueError('create_scanner requires a type element')
        if not ca_pub:
            raise ValueError('create_scanner requires a ca_pub element')
        if not credential_id:
            raise ValueError('create_scanner requires a credential_id element')

        cmd = XmlCommand('create_scanner')
        cmd.add_element('name', name)
        cmd.add_element('host', host)
        cmd.add_element('port', port)
        cmd.add_element('type', type)
        cmd.add_element('ca_pub', ca_pub)
        cmd.add_element('credential', attrs={'id': str(credential_id)})

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        return cmd.to_string()

    def create_schedule_command(self, name, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create schedule on gvmd."""
        if not name:
            raise ValueError('create_schedule requires a name element')

        cmd = XmlCommand('create_schedule')
        cmd.add_element('name', name)

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        first_time = kwargs.get('first_time', '')
        if first_time:
            first_time_minute = first_time['minute']
            first_time_hour = first_time['hour']
            first_time_day_of_month = first_time['day_of_month']
            first_time_month = first_time['month']
            first_time_year = first_time['year']

            _xmlFtime = cmd.add_element('first_time')

            _xmlFtime.add_element('minute', first_time_minute)
            _xmlFtime.add_element('hour', str(first_time_hour))
            _xmlFtime.add_element('day_of_month', str(first_time_day_of_month))
            _xmlFtime.add_element('month', str(first_time_month))
            _xmlFtime.add_element('year', str(first_time_year))

        duration = kwargs.get('duration', '')
        if len(duration) > 1:
            _xmlDuration = cmd.add_element('duration', str(duration[0]))

            _xmlDuration.add_element('unit', str(duration[1]))

        period = kwargs.get('period', '')
        if len(period) > 1:
            _xmlPeriod = cmd.add_element('period', str(period[0]))

            _xmlPeriod.add_element('unit', str(period[1]))

        timezone = kwargs.get('timezone', '')
        if timezone:
            cmd.add_element('timezone', str(timezone))

        return cmd.to_string()

    def create_tag_command(self, name, resource_id, resource_type, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create tag on gvmd."""

        cmd = XmlCommand('create_tag')
        _cmd.add_element('name', name)

        _xmlResource = cmd.add_element('resource',

                                       attrs={'id': str(resource_id)})
        _xmlResource.add_element('type', resource_type)

        comment = kwargs.get('comment', '')
        if comment:
            cmd.add_element('comment', comment)

        copy = kwargs.get('copy', '')
        if copy:
            cmd.add_element('copy', copy)

        value = kwargs.get('value', '')
        if value:
            cmd.add_element('value', value)

        active = kwargs.get('active', '')
        if active:
            cmd.add_element('active', active)

        return cmd.to_string()

    def create_target_command(self, name, make_unique, kwargs):
class Foo:
    def some_method(self, x, y):
        return x + y;
        """Generates xml string for create target on gvmd."""
        if not name:
            raise ValueError('create_target requires a name element')

        cmd = XmlCommand('create_target')
        _xmlName = cmd.add_element('name', name)

        if make_unique:
            _xmlName.add_element('make_unique', '1')
        else:
            _xmlName.add_element('make_unique', '0')

        if 'asset_hosts' in kwargs:
            hosts = kwargs.get('asset_hosts')
            filter = hosts['filter']

            cmd.add_element('asset_hosts', attrs={'filter': str(filter)})
        elif 'hosts' in kwargs:
            hosts = kwargs.get('hosts')
            cmd.add_element('hosts', hosts)
        else:
            raise ValueError('create_target requires either a hosts or '
                             'an asset_hosts element')

        if 'comment' in kwargs:
            cmd.add_element('comment', kwargs.get('comment'))

        if 'copy' in kwargs:
            # NOTE: It seems that hosts/asset_hosts is silently ignored by the
            # server when copy is supplied. But for specification conformance
            # we raise the ValueError above and consider copy optional.
            cmd.add_element('copy', kwargs.get('copy'))

        if 'exclude_hosts' in kwargs:
            cmd.add_element('exclude_hosts', kwargs.get('exclude_hosts'))

        if 'ssh_credential' in kwargs:
            ssh_credential = kwargs.get('ssh_credential')
            if 'id' in ssh_credential:
                _xmlSSH = cmd.add_element('ssh_credential', '',

                                          attrs={'id': ssh_credential['id']})
                if 'port' in ssh_credential:
                    _xmlSSH.add_element('port', ssh_credential['port'])
            else:
                raise ValueError('ssh_credential requires an id attribute')

        if 'smb_credential' in kwargs:
            smb_credential = kwargs.get('smb_credential')
            if 'id' in smb_credential:
                cmd.add_element('smb_credential',
                                attrs={'id': smb_credential['id']})
            else:
                raise ValueError('smb_credential requires an id attribute')

        if 'esxi_credential' in kwargs:
            esxi_credential = kwargs.get('esxi_credential')
            if 'id' in esxi_credential:
                cmd.add_element('esxi_credential',
                                attrs={'id': esxi_credential['id']})
            else:
                raise ValueError('esxi_credential requires an id attribute')

        if 'snmp_credential' in kwargs:
            snmp_credential = kwargs.get('snmp_credential')
            if 'id' in snmp_credential:
                cmd.add_element('snmp_credential',
                                attrs={'id': snmp_credential['id']})
            else:
                raise ValueError('snmp_credential requires an id attribute')

        if 'alive_tests' in kwargs:
            # NOTE: As the alive_tests are referenced by their name and some
            # names contain ampersand ('&') characters it should be considered
            # replacing any characters special to XML in the variable with
            # their corresponding entities.
            cmd.add_element('alive_tests', kwargs.get('alive_tests'))

        if 'reverse_lookup_only' in kwargs:
            reverse_lookup_only = kwargs.get('reverse_lookup_only')
            if reverse_lookup_only:
                cmd.add_element('reverse_lookup_only', '1')
            else:
                cmd.add_element('reverse_lookup_only', '0')

        if 'reverse_lookup_unify' in kwargs:
            reverse_lookup_unify = kwargs.get('reverse_lookup_unify')
            if reverse_lookup_unify:
                cmd.add_element('reverse_lookup_unify', '1')
            else:
                cmd.add_element('reverse_lookup_unify', '0')

        if 'port_range' in kwargs:
            cmd.add_element('port_range', kwargs.get('port_range'))

        if 'port_list' in kwargs:
            port_list = kwargs.get('port_list')
            if 'id' in port_list:
                cmd.add_element('port_list',
                                attrs={'id': str(port_list['id'])})
            else:
                raise ValueError('port_list requires an id attribute')

        return cmd.to_string()

    def create_task_command(self, name, config_id, target_id, scanner_id,
class Foo:
    def some_method(self, x, y):
        return x + y;
                            alert_ids=None, comment=''):
        """Generates xml string for create task on gvmd."""

        if alert_ids is None:
            alert_ids = []
        cmd = XmlCommand('create_task')
        cmd.add_element('name', name)
        cmd.add_element('comment', comment)
        cmd.add_element('config', attrs={'id': config_id})
        cmd.add_element('target', attrs={'id': target_id})
        cmd.add_element('scanner', attrs={'id': scanner_id})

        #if given the alert_id is wrapped and integrated suitably as xml
        if len(alert_ids) > 0:

            if isinstance(alert_ids, str):
                #if a single id is given as a string wrap it into a list
                alert_ids = [alert_ids]
            if isinstance(alert_ids, list):
                #parse all given alert id's
                for alert in alert_ids:
                    cmd.add_element('alert', attrs={'id': str(alert)})

        return cmd.to_string()

    def create_user_command(self, name, password, copy='', hosts_allow='0',
class Foo:
    def some_method(self, x, y):
        return x + y;
                            ifaces_allow='0', role_ids=(), hosts=None,
                            ifaces=None):
        """Generates xml string for create user on gvmd."""
        cmd = XmlCommand('create_user')
        cmd.add_element('name', name)

        if copy:
            cmd.add_element('copy', copy)

        if password:
            cmd.add_element('password', password)

        if hosts is not None:
            cmd.add_element('hosts', hosts, attrs={'allow': str(hosts_allow)})

        if ifaces is not None:
            cmd.add_element('ifaces', ifaces,
                            attrs={'allow': str(ifaces_allow)})

        if len(role_ids) > 0:

            for role in role_ids:
                cmd.add_element('role', attrs={'allow': str(role)})

        return cmd.to_string()

    def modify_agent_command(self, agent_id, name='', comment=''):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not agent_id:
            raise ValueError('modify_agent requires an agent_id element')

        xmlRoot = etree.Element('modify_agent', agent_id=str(agent_id))

        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_alert_command(self, alert_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not alert_id:
            raise ValueError('modify_alert requires an agent_id element')

        xmlRoot = etree.Element('modify_alert', alert_id=str(alert_id))


        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        filter_id = kwargs.get('filter_id', '')
        if filter_id:
            _xmlFilter = etree.SubElement(xmlRoot, 'filter', id=filter_id)


        event = kwargs.get('event', '')
        if len(event) > 1:
            _xmlEvent = etree.SubElement(xmlRoot, 'event')

            _xmlEvent.text = event[0]
            for value, key in event[1].items():
                _xmlData = etree.SubElement(_xmlEvent, 'data')

                _xmlData.text = value
                _xmlDName = etree.SubElement(_xmlData, 'name')

                _xmlDName.text = key

        condition = kwargs.get('condition', '')
        if len(condition) > 1:
            _xmlCond = etree.SubElement(xmlRoot, 'condition')

            _xmlCond.text = condition[0]
            for value, key in condition[1].items():
                _xmlData = etree.SubElement(_xmlCond, 'data')

                _xmlData.text = value
                _xmlDName = etree.SubElement(_xmlData, 'name')

                _xmlDName.text = key

        method = kwargs.get('method', '')
        if len(method) > 1:
            _xmlMethod = etree.SubElement(xmlRoot, 'method')

            _xmlMethod.text = method[0]
            for value, key in method[1].items():
                _xmlData = etree.SubElement(_xmlMethod, 'data')

                _xmlData.text = value
                _xmlDName = etree.SubElement(_xmlData, 'name')

                _xmlDName.text = key

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_auth_command(self, group_name, auth_conf_settings):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not group_name:
            raise ValueError('modify_auth requires a group element '
                             'with a name attribute')
        if not auth_conf_settings:
            raise ValueError('modify_auth requires '
                             'an auth_conf_settings element')

        xmlRoot = etree.Element('modify_auth')

        _xmlGroup = etree.SubElement(xmlRoot, 'group', name=str(group_name))


        for key, value in auth_conf_settings.items():
            _xmlAuthConf = etree.SubElement(_xmlGroup, 'auth_conf_setting')

            _xmlKey = etree.SubElement(_xmlAuthConf, 'key')

            _xmlKey.text = key
            _xmlValue = etree.SubElement(_xmlAuthConf, 'value')

            _xmlValue.text = value

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_config_command(self, selection, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if selection not in ('nvt_pref', 'sca_pref',
                             'family_selection', 'nvt_selection'):
            raise ValueError('selection must be one of nvt_pref, sca_pref, '
                             'family_selection or nvt_selection')
        config_id = kwargs.get('config_id')

        xmlRoot = etree.Element('modify_config', config_id=str(config_id))


        if selection in 'nvt_pref':
            nvt_oid = kwargs.get('nvt_oid')
            name = kwargs.get('name')
            value = kwargs.get('value')
            _xmlPref = etree.SubElement(xmlRoot, 'preference')

            _xmlNvt = etree.SubElement(_xmlPref, 'nvt', oid=nvt_oid)

            _xmlName = etree.SubElement(_xmlPref, 'name')

            _xmlName.text = name
            _xmlValue = etree.SubElement(_xmlPref, 'value')

            _xmlValue.text = value

        elif selection in 'nvt_selection':
            nvt_oid = kwargs.get('nvt_oid')
            family = kwargs.get('family')
            _xmlNvtSel = etree.SubElement(xmlRoot, 'nvt_selection')

            _xmlFamily = etree.SubElement(_xmlNvtSel, 'family')

            _xmlFamily.text = family

            if isinstance(nvt_oid, list):
                for nvt in nvt_oid:
                    _xmlNvt = etree.SubElement(_xmlNvtSel, 'nvt', oid=nvt)

            else:
                _xmlNvt = etree.SubElement(_xmlNvtSel, 'nvt', oid=nvt)


        elif selection in 'family_selection':
            family = kwargs.get('family')
            _xmlFamSel = etree.SubElement(xmlRoot, 'family_selection')

            _xmlGrow = etree.SubElement(_xmlFamSel, 'growing')

            _xmlGrow.text = '1'
            _xmlFamily = etree.SubElement(_xmlFamSel, 'family')

            _xmlName = etree.SubElement(_xmlFamily, 'name')

            _xmlName.text = family
            _xmlAll = etree.SubElement(_xmlFamily, 'all')

            _xmlAll.text = '1'
            _xmlGrowI = etree.SubElement(_xmlFamily, 'growing')

            _xmlGrowI.text = '1'
        else:
            raise NotImplementedError

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_credential_command(self, credential_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not credential_id:
            raise ValueError('modify_credential requires '
                             'a credential_id attribute')

        xmlRoot = etree.Element('modify_credential',

                                credential_id=credential_id)

        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        allow_insecure = kwargs.get('allow_insecure', '')
        if allow_insecure:
            _xmlAllowinsecure = etree.SubElement(xmlRoot, 'allow_insecure')

            _xmlAllowinsecure.text = allow_insecure

        certificate = kwargs.get('certificate', '')
        if certificate:
            _xmlCertificate = etree.SubElement(xmlRoot, 'certificate')

            _xmlCertificate.text = certificate

        key = kwargs.get('key', '')
        if key:
            phrase = key['phrase']
            private = key['private']
            if not phrase:
                raise ValueError('modify_credential requires a phrase element')
            if not private:
                raise ValueError('modify_credential requires '
                                 'a private element')
            _xmlKey = etree.SubElement(xmlRoot, 'key')

            _xmlKeyphrase = etree.SubElement(_xmlKey, 'phrase')

            _xmlKeyphrase.text = phrase
            _xmlKeyprivate = etree.SubElement(_xmlKey, 'private')

            _xmlKeyprivate.text = private

        login = kwargs.get('login', '')
        if login:
            _xmlLogin = etree.SubElement(xmlRoot, 'login')

            _xmlLogin.text = login

        password = kwargs.get('password', '')
        if password:
            _xmlPass = etree.SubElement(xmlRoot, 'password')

            _xmlPass.text = password

        auth_algorithm = kwargs.get('auth_algorithm', '')
        if auth_algorithm:
            if auth_algorithm not in ('md5', 'sha1'):
                raise ValueError('modify_credential requires auth_algorithm '
                                 'to be either md5 or sha1')
            _xmlAuthalg = etree.SubElement(xmlRoot, 'auth_algorithm')

            _xmlAuthalg.text = auth_algorithm

        community = kwargs.get('community', '')
        if community:
            _xmlCommunity = etree.SubElement(xmlRoot, 'community')

            _xmlCommunity.text = community

        privacy = kwargs.get('privacy', '')
        if privacy:
            algorithm = privacy.algorithm
            if algorithm not in ('aes', 'des'):
                raise ValueError('modify_credential requires algorithm '
                                 'to be either aes or des')
            p_password = privacy.password
            _xmlPrivacy = etree.SubElement(xmlRoot, 'privacy')

            _xmlAlgorithm = etree.SubElement(_xmlPrivacy, 'algorithm')

            _xmlAlgorithm.text = algorithm
            _xmlPpass = etree.SubElement(_xmlPrivacy, 'password')

            _xmlPpass.text = p_password

        cred_type = kwargs.get('type', '')
        if cred_type:
            if cred_type not in ('cc', 'snmp', 'up', 'usk'):
                raise ValueError('modify_credential requires type '
                                 'to be either cc, snmp, up or usk')
            _xmlCredtype = etree.SubElement(xmlRoot, 'type')

            _xmlCredtype.text = cred_type

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_filter_command(self, filter_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not filter_id:
            raise ValueError('modify_filter requires a filter_id attribute')

        xmlRoot = etree.Element('modify_filter', filter_id=filter_id)


        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        copy = kwargs.get('copy', '')
        if copy:
            _xmlCopy = etree.SubElement(xmlRoot, 'copy')

            _xmlCopy.text = copy

        term = kwargs.get('term', '')
        if term:
            _xmlTerm = etree.SubElement(xmlRoot, 'term')

            _xmlTerm.text = term

        filter_type = kwargs.get('type', '')
        if filter_type:
            if filter_type not in ('cc', 'snmp', 'up', 'usk'):
                raise ValueError('modify_filter requires type '
                                 'to be either cc, snmp, up or usk')
            _xmlFiltertype = etree.SubElement(xmlRoot, 'type')

            _xmlFiltertype.text = filter_type

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_group_command(self, group_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not group_id:
            raise ValueError('modify_group requires a group_id attribute')

        xmlRoot = etree.Element('modify_group', group_id=group_id)


        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        users = kwargs.get('users', '')
        if users:
            _xmlUser = etree.SubElement(xmlRoot, 'users')

            _xmlUser.text = users

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_note_command(self, note_id, text, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not note_id:
            raise ValueError('modify_note requires a note_id attribute')
        if not text:
            raise ValueError('modify_note requires a text element')

        xmlRoot = etree.Element('modify_note', note_id=note_id)

        _xmlText = etree.SubElement(xmlRoot, 'text')

        _xmlText.text = text

        active = kwargs.get('active', '')
        if active:
            _xmlActive = etree.SubElement(xmlRoot, 'active')

            _xmlActive.text = active

        hosts = kwargs.get('hosts', '')
        if hosts:
            _xmlHosts = etree.SubElement(xmlRoot, 'hosts')

            _xmlHosts.text = hosts

        port = kwargs.get('port', '')
        if port:
            _xmlPort = etree.SubElement(xmlRoot, 'port')

            _xmlPort.text = port

        result_id = kwargs.get('result_id', '')
        if result_id:
            _xmlResultid = etree.SubElement(xmlRoot, 'result', id=result_id)


        severity = kwargs.get('severity', '')
        if severity:
            _xmlSeverity = etree.SubElement(xmlRoot, 'severity')

            _xmlSeverity.text = severity

        task_id = kwargs.get('task_id', '')
        if task_id:
            _xmlTaskid = etree.SubElement(xmlRoot, 'task', id=task_id)


        threat = kwargs.get('threat', '')
        if threat:
            _xmlThreat = etree.SubElement(xmlRoot, 'threat')

            _xmlThreat.text = threat

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_override_command(self, override_id, text, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        xmlRoot = etree.Element('modify_override',

                                override_id=override_id)
        _xmlText = etree.SubElement(xmlRoot, 'text')

        _xmlText.text = text

        active = kwargs.get('active', '')
        if active:
            _xmlActive = etree.SubElement(xmlRoot, 'active')

            _xmlActive.text = active

        hosts = kwargs.get('hosts', '')
        if hosts:
            _xmlHosts = etree.SubElement(xmlRoot, 'hosts')

            _xmlHosts.text = hosts

        port = kwargs.get('port', '')
        if port:
            _xmlPort = etree.SubElement(xmlRoot, 'port')

            _xmlPort.text = port

        result_id = kwargs.get('result_id', '')
        if result_id:
            _xmlResultid = etree.SubElement(xmlRoot, 'result', id=result_id)


        severity = kwargs.get('severity', '')
        if severity:
            _xmlSeverity = etree.SubElement(xmlRoot, 'severity')

            _xmlSeverity.text = severity

        new_severity = kwargs.get('new_severity', '')
        if new_severity:
            _xmlNSeverity = etree.SubElement(xmlRoot, 'new_severity')

            _xmlNSeverity.text = new_severity

        task_id = kwargs.get('task_id', '')
        if task_id:
            _xmlTaskid = etree.SubElement(xmlRoot, 'task', id=task_id)


        threat = kwargs.get('threat', '')
        if threat:
            _xmlThreat = etree.SubElement(xmlRoot, 'threat')

            _xmlThreat.text = threat

        new_threat = kwargs.get('new_threat', '')
        if new_threat:
            _xmlNThreat = etree.SubElement(xmlRoot, 'new_threat')

            _xmlNThreat.text = new_threat

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_permission_command(self, permission_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not permission_id:
            raise ValueError('modify_permission requires '
                             'a permission_id element')

        xmlRoot = etree.Element('modify_permission',

                                permission_id=permission_id)

        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        resource = kwargs.get('resource', '')
        if resource:
            resource_id = resource['id']
            resource_type = resource['type']
            _xmlResource = etree.SubElement(xmlRoot, 'resource',

                                            id=resource_id)
            _xmlRType = etree.SubElement(_xmlResource, 'type')

            _xmlRType.text = resource_type

        subject = kwargs.get('subject', '')
        if subject:
            subject_id = subject['id']
            subject_type = subject['type']
            _xmlSubject = etree.SubElement(xmlRoot, 'subject', id=subject_id)

            _xmlType = etree.SubElement(_xmlSubject, 'type')

            _xmlType.text = subject_type

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_port_list_command(self, port_list_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not port_list_id:
            raise ValueError('modify_port_list requires '
                             'a port_list_id attribute')
        xmlRoot = etree.Element('modify_port_list',

                                port_list_id=port_list_id)

        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_report_format_command(self, report_format_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if len(kwargs) < 1:
            raise Exception('modify_report_format: Missing parameter')

        xmlRoot = etree.Element('modify_report_format',

                                report_format_id=report_format_id)

        active = kwargs.get('active', '')
        if active:
            _xmlActive = etree.SubElement(xmlRoot, 'active')

            _xmlActive.text = active

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

            summary = kwargs.get('summary', '')
        if summary:

            _xmlSummary = etree.SubElement(xmlRoot, 'summary')

            _xmlSummary.text = summary

        param = kwargs.get('param', '')
        if param:
            p_name = param[0]
            p_value = param[1]
            _xmlParam = etree.SubElement(xmlRoot, 'param')

            _xmlPname = etree.SubElement(_xmlParam, 'name')

            _xmlPname.text = p_name
            _xmlValue = etree.SubElement(_xmlParam, 'value')

            _xmlValue.text = p_value

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_role_command(self, role_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not role_id:
            raise ValueError('modify_role requires a role_id element')

        xmlRoot = etree.Element('modify_role',

                                role_id=role_id)

        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        users = kwargs.get('users', '')
        if users:
            _xmlUser = etree.SubElement(xmlRoot, 'users')

            _xmlUser.text = users

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_scanner_command(self, scanner_id, host, port, scanner_type,
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
                               kwargs):
        if not scanner_id:
            raise ValueError('modify_scanner requires a scanner_id element')
        if not host:
            raise ValueError('modify_scanner requires a host element')
        if not port:
            raise ValueError('modify_scanner requires a port element')
        if not scanner_type:
            raise ValueError('modify_scanner requires a type element')

        xmlRoot = etree.Element('modify_scanner', scanner_id=scanner_id)

        _xmlHost = etree.SubElement(xmlRoot, 'host')

        _xmlHost.text = host
        _xmlPort = etree.SubElement(xmlRoot, 'port')

        _xmlPort.text = port
        _xmlType = etree.SubElement(xmlRoot, 'type')

        _xmlType.text = scanner_type

        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        ca_pub = kwargs.get('ca_pub', '')
        if ca_pub:
            _xmlCAPub = etree.SubElement(xmlRoot, 'ca_pub')

            _xmlCAPub.text = ca_pub

        credential_id = kwargs.get('credential_id', '')
        if credential_id:
            _xmlCred = etree.SubElement(xmlRoot, 'credential',

                                        id=str(credential_id))

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_schedule_command(self, schedule_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not schedule_id:
            raise ValueError('modify_schedule requires a schedule_id element')

        xmlRoot = etree.Element('modify_schedule', schedule_id=schedule_id)

        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        first_time = kwargs.get('first_time', '')
        if first_time:
            first_time_minute = first_time['minute']
            first_time_hour = first_time['hour']
            first_time_day_of_month = first_time['day_of_month']
            first_time_month = first_time['month']
            first_time_year = first_time['year']

            _xmlFtime = etree.SubElement(xmlRoot, 'first_time')

            _xmlMinute = etree.SubElement(_xmlFtime, 'minute')

            _xmlMinute.text = str(first_time_minute)
            _xmlHour = etree.SubElement(_xmlFtime, 'hour')

            _xmlHour.text = str(first_time_hour)
            _xmlDay = etree.SubElement(_xmlFtime, 'day_of_month')

            _xmlDay.text = str(first_time_day_of_month)
            _xmlMonth = etree.SubElement(_xmlFtime, 'month')

            _xmlMonth.text = str(first_time_month)
            _xmlYear = etree.SubElement(_xmlFtime, 'year')

            _xmlYear.text = str(first_time_year)

        duration = kwargs.get('duration', '')
        if len(duration) > 1:
            _xmlDuration = etree.SubElement(xmlRoot, 'duration')

            _xmlDuration.text = str(duration[0])
            _xmlUnit = etree.SubElement(_xmlDuration, 'unit')

            _xmlUnit.text = str(duration[1])

        period = kwargs.get('period', '')
        if len(period) > 1:
            _xmlPeriod = etree.SubElement(xmlRoot, 'period')

            _xmlPeriod.text = str(period[0])
            _xmlPUnit = etree.SubElement(_xmlPeriod, 'unit')

            _xmlPUnit.text = str(period[1])

        timezone = kwargs.get('timezone', '')
        if timezone:
            _xmlTimezone = etree.SubElement(xmlRoot, 'timezone')

            _xmlTimezone.text = str(timezone)

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_tag_command(self, tag_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not tag_id:
            raise ValueError('modify_tag requires a tag_id element')

        xmlRoot = etree.Element('modify_tag', tag_id=str(tag_id))


        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        value = kwargs.get('value', '')
        if value:
            _xmlValue = etree.SubElement(xmlRoot, 'value')

            _xmlValue.text = value

        active = kwargs.get('active', '')
        if active:
            _xmlActive = etree.SubElement(xmlRoot, 'active')

            _xmlActive.text = value

        resource = kwargs.get('resource', '')
        if resource:
            resource_id = resource['id']
            resource_type = resource['type']
            _xmlResource = etree.SubElement(xmlRoot, 'resource',

                                            resource_id=resource_id)
            _xmlRType = etree.SubElement(_xmlResource, 'type')

            _xmlRType.text = resource_type

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_target_command(self, target_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not target_id:
            raise ValueError('modify_target requires a target_id element')

        xmlRoot = etree.Element('modify_target', target_id=target_id)


        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        hosts = kwargs.get('hosts', '')
        if hosts:
            _xmlHosts = etree.SubElement(xmlRoot, 'hosts')

            _xmlHosts.text = hosts

        copy = kwargs.get('copy', '')
        if copy:
            _xmlCopy = etree.SubElement(xmlRoot, 'copy')

            _xmlCopy.text = kwargs.get('copy')

        exclude_hosts = kwargs.get('exclude_hosts', '')
        if exclude_hosts:
            _xmlExHosts = etree.SubElement(xmlRoot, 'exclude_hosts')

            _xmlExHosts.text = kwargs.get('exclude_hosts')

        alive_tests = kwargs.get('alive_tests', '')
        if alive_tests:
            _xmlAlive = etree.SubElement(xmlRoot, 'alive_tests')

            _xmlAlive.text = kwargs.get('alive_tests')

        reverse_lookup_only = kwargs.get('reverse_lookup_only', '')
        if reverse_lookup_only:
            _xmlLookup = etree.SubElement(xmlRoot, 'reverse_lookup_only')

            _xmlLookup.text = reverse_lookup_only

        reverse_lookup_unify = kwargs.get('reverse_lookup_unify', '')
        if reverse_lookup_unify:
            _xmlLookupU = etree.SubElement(xmlRoot, 'reverse_lookup_unify')

            _xmlLookupU.text = reverse_lookup_unify

        port_range = kwargs.get('port_range', '')
        if port_range:
            _xmlPortR = etree.SubElement(xmlRoot, 'port_range')

            _xmlPortR.text = kwargs.get('port_range')

        port_list = kwargs.get('port_list', '')
        if port_list:
            _xmlPortL = etree.SubElement(xmlRoot, 'port_list',

                                         id=str(port_list))

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_task_command(self, task_id, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if not task_id:
            raise ValueError('modify_task requires a task_id element')

        xmlRoot = etree.Element('modify_task', task_id=task_id)


        name = kwargs.get('name', '')
        if name:
            _xmlName = etree.SubElement(xmlRoot, 'name')

            _xmlName.text = name

        comment = kwargs.get('comment', '')
        if comment:
            _xmlComment = etree.SubElement(xmlRoot, 'comment')

            _xmlComment.text = comment

        target_id = kwargs.get('target_id', '')
        if target_id:
            _xmlTarget = etree.SubElement(xmlRoot, 'target', id=target_id)


        scanner = kwargs.get('scanner', '')
        if scanner:
            _xmlScanner = etree.SubElement(xmlRoot, 'scanner', id=scanner)


        schedule_periods = kwargs.get('schedule_periods', '')
        if schedule_periods:
            _xmlPeriod = etree.SubElement(xmlRoot, 'schedule_periods')

            _xmlPeriod.text = str(schedule_periods)

        schedule = kwargs.get('schedule', '')
        if schedule:
            _xmlSched = etree.SubElement(xmlRoot, 'schedule', id=str(schedule))


        alert = kwargs.get('alert', '')
        if alert:
            _xmlAlert = etree.SubElement(xmlRoot, 'alert', id=str(alert))


        observers = kwargs.get('observers', '')
        if observers:
            _xmlObserver = etree.SubElement(xmlRoot, 'observers')

            _xmlObserver.text = str(observers)

        preferences = kwargs.get('preferences', '')
        if preferences:
            _xmlPrefs = etree.SubElement(xmlRoot, 'preferences')

            for n in range(len(preferences["scanner_name"])):

                preferences_scanner_name = preferences["scanner_name"][n]
                preferences_value = preferences["value"][n]
                _xmlPref = etree.SubElement(_xmlPrefs, 'preference')

                _xmlScan = etree.SubElement(_xmlPref, 'scanner_name')

                _xmlScan.text = preferences_scanner_name
                _xmlVal = etree.SubElement(_xmlPref, 'value')

                _xmlVal.text = preferences_value

        file = kwargs.get('file', '')
        if file:
            file_name = file['name']
            file_action = file['action']
            if file_action != "update" and file_action != "remove":
                raise ValueError('action can only be "update" or "remove"!')
            _xmlFile = etree.SubElement(xmlRoot, 'file', name=file_name,

                                        action=file_action)

        return etree.tostring(xmlRoot).decode('utf-8')


    def modify_user_command(self, kwargs):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        user_id = kwargs.get('user_id', '')
        name = kwargs.get('name', '')

        if not user_id and not name:
            raise ValueError('modify_user requires '
                             'either a user_id or a name element')

        xmlRoot = etree.Element('modify_user', user_id=str(user_id))


        new_name = kwargs.get('new_name', '')
        if new_name:
            _xmlName = etree.SubElement(xmlRoot, 'new_name')

            _xmlName.text = new_name

        password = kwargs.get('password', '')
        if password:
            _xmlPass = etree.SubElement(xmlRoot, 'password')

            _xmlPass.text = password

        role_ids = kwargs.get('role_ids', '')
        if len(role_ids) > 0:

            for role in role_ids:
                _xmlRole = etree.SubElement(xmlRoot, 'role',

                                            id=str(role))
        hosts = kwargs.get('hosts', '')
        hosts_allow = kwargs.get('hosts_allow', '')
        if hosts or hosts_allow:
            _xmlHosts = etree.SubElement(xmlRoot, 'hosts',

                                         allow=str(hosts_allow))
            _xmlHosts.text = hosts

        ifaces = kwargs.get('ifaces', '')
        ifaces_allow = kwargs.get('ifaces_allow', '')
        if ifaces or ifaces_allow:
            _xmlIFaces = etree.SubElement(xmlRoot, 'ifaces',

                                          allow=str(ifaces_allow))
            _xmlIFaces.text = ifaces

        sources = kwargs.get('sources', '')
        if sources:
            _xmlSource = etree.SubElement(xmlRoot, 'sources')

            _xmlSource.text = sources

        return etree.tostring(xmlRoot).decode('utf-8')


1			# -- coding: utf-8 --
			0 ignored issues – show Coding Style introduced 2018-10-23 11:01 UTC by Report Bug Copy Issue Report This module should have a docstring. The coding style of this project requires that you add a docstring to this code element. Below, you find an example for methods: class SomeClass: def some_method(self): """Do x and return foo.""" If you would like to know more about docstrings, we recommend to read PEP-257: Docstring Conventions. Loading history... coding-style introduced 2018-10-30 07:17 UTC by Report Bug Copy Issue Report Too many lines in module (1599/1000) Loading history...
2			# Copyright (C) 2018 Greenbone Networks GmbH
3			#
4			# SPDX-License-Identifier: GPL-3.0-or-later
5			#
6			# This program is free software: you can redistribute it and/or modify
7			# it under the terms of the GNU General Public License as published by
8			# the Free Software Foundation, either version 3 of the License, or
9			# (at your option) any later version.
10			#
11			# This program is distributed in the hope that it will be useful,
12			# but WITHOUT ANY WARRANTY; without even the implied warranty of
13			# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			# GNU General Public License for more details.
15			#
16			# You should have received a copy of the GNU General Public License
17			# along with this program. If not, see <http://www.gnu.org/licenses/>.
18
19			import defusedxml.lxml as secET
20
21			from lxml import etree
22
23			FILTER_NAMES = [
24			'Agent',
25			'Alert',
26			'Asset',
27			'Config',
28			'Credential',
29			'Filter',
30			'Group',
31			'Note',
32			'Override',
33			'Permission',
34			'Port List',
35			'Report',
36			'Report Format',
37			'Result',
38			'Role',
39			'Schedule',
greenbone / gvm-tools

Pull Request — master (#116)

_GmpCommandFactory.create_schedule_command() B

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like
