gvmtools.config

Complexity

Total Complexity

9

Size/Duplication

Total Lines	84
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	Config.__init__()	0	13	1
A	Config.get()	0	5	2
A	Config.defaults()	0	2	1
A	Config.load()	0	26	5

# -*- coding: utf-8 -*-
# Copyright (C) 2019 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/>.
"""
Module to store gvm-tools configuration settings
"""

import configparser
import logging

from pathlib import Path

from gvm.connections import DEFAULT_UNIX_SOCKET_PATH, DEFAULT_GVM_PORT

logger = logging.getLogger(__name__)

DEFAULT_SSH_PORT = 22


class Config:
    def __init__(self):
        self._config = configparser.ConfigParser(default_section='main')

        self._config = {}

        self._config['gmp'] = dict(username='', password='')
        self._config['ssh'] = dict(
            username='gmp', password='gmp', port=DEFAULT_SSH_PORT
        )
        self._config['unixsocket'] = dict(socketpath=DEFAULT_UNIX_SOCKET_PATH)
        self._config['tls'] = dict(port=DEFAULT_GVM_PORT)

        self._defaults = dict()

    def load(self, filename):
        path = Path(filename).expanduser()

        config = configparser.ConfigParser(default_section='main')

        config.read_file(path.open())

        if 'Auth' in config:
            logger.warning(
                "Warning: Loaded config file %s contains deprecated 'Auth' "
                "section. This section will be ignored in future.",
                filename,
            )
            gmp_username = config.get('Auth', 'gmp_username', fallback='')
            gmp_password = config.get('Auth', 'gmp_password', fallback='')
            self._config['gmp']['username'] = gmp_username
            self._config['gmp']['password'] = gmp_password

        self._defaults.update(config.defaults())

        for section in config.sections():
            if section == 'Auth':
                continue

            for key, value in config.items(section):
                self._config.setdefault(section, dict())[key] = value

The variable dict does not seem to be defined.

    def defaults(self):
        return self._defaults

    def get(self, section, name):
        if not section in self._config:
            return None

        return self._config[section].get(name)