gvm.xml._GmpCommandFactory.create_target_command()

Complexity

Conditions

25

Size

Total Lines	102
Code Lines	69

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

# -*- coding: utf-8 -*-
    def create_task_command(self, name, config_id, target_id, scanner_id,

This method could be written as a function/class method.

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

This method could be written as a function/class method.

                            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=''):

This method could be written as a function/class method.

        """Generates xml string for modify agent on gvmd."""
        if not agent_id:
            raise ValueError('modify_agent requires an agent_id element')

        cmd = XmlCommand('modify_agent')
        cmd.set_attribute('agent_id', str(agent_id))
        if name:
            cmd.add_element('name', name)
        if comment:
            cmd.add_element('comment', comment)

        return cmd.to_string()

    def modify_alert_command(self, alert_id, kwargs):

This function exceeds the maximum number of variables (17/15).

This method could be written as a function/class method.

        """Generates xml string for modify alert on gvmd."""

        if not alert_id:
            raise ValueError('modify_alert requires an agent_id element')

        cmd = XmlCommand('modify_alert')
        cmd.set_attribute('alert_id', str(alert_id))

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

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

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

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

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

        method = kwargs.get('method', '')
        if len(method) > 1:
            _xmlmethod = cmd.add_element('method', method[0])