gmp.clients.helper

Complexity

Total Complexity

11

Size/Duplication

Total Lines	79
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	pretty_print()	0	18	5
B	authenticate()	0	33	6

# -*- coding: utf-8 -*-

This module should have a docstring.

# 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 getpass

from lxml import etree

from gmp.error import GmpError


def authenticate(gmp, args):
    """Authentication helper

    Tries to get authentication username and password from args and if not
    present asks the username and/or password from the terminal.

    Attributes:
        gmp: A protocol instance
        args: The parsed arguments

    Raises:
        GmpError: Raises GmpError if authentication fails.
    """
    if gmp.is_authenticated():
        return

    username = args.gmp_username
    password = args.gmp_password

    # Ask for login credentials if none are given.
    if not username:
        while len(username) == 0:

Do not use len(SEQUENCE) as condition value

            username = input('Enter username: ')

    if not password:
        password = getpass.getpass(
            'Enter password for {0}: '.format(username))

    try:
        gmp.authenticate(username, password)
    except GmpError as e:

The name e does not conform to the variable naming conventions ((([a-z][a-z0-9_]{2,30})|(_[a-z0-9_]*))$).

        print('Could not authenticate. Please check your credentials.')
        raise e


def pretty_print(xml):
    """Prints beautiful XML-Code

    This function gets an object of list<lxml.etree._Element>
    or directly a lxml element.
    Print it with good readable format.

    Arguments:
        xml: List<lxml.etree.Element> or directly a lxml element
    """
    if isinstance(xml, list):
        for item in xml:
            if etree.iselement(item):

The Module lxml.etree does not seem to have a member named iselement.

                print(etree.tostring(item, pretty_print=True).decode('utf-8'))

The Module lxml.etree does not seem to have a member named tostring.

            else:
                print(item)
    elif etree.iselement(xml):

The Module lxml.etree does not seem to have a member named iselement.

        print(etree.tostring(xml, pretty_print=True).decode('utf-8'))

The Module lxml.etree does not seem to have a member named tostring.