Passed
Pull Request — master (#246)
by
unknown
01:17
created

gvmtools.get_version_from_pyproject_toml()   A

Complexity

Conditions 4

Size

Total Lines 10
Code Lines 8

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 4
eloc 8
nop 0
dl 0
loc 10
rs 10
c 0
b 0
f 0
1
# -*- coding: utf-8 -*-
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
Main module of gvm-tools.
20
"""
21
22
from pathlib import Path
23
24
from pkg_resources import safe_version
25
26
import toml
27
28
29
def get_version_from_pyproject_toml() -> str:
30
    path = Path(__file__)
31
    pyproject_toml_path = path.parent.parent / 'pyproject.toml'
32
33
    if pyproject_toml_path.exists():
34
        pyproject_toml = toml.loads(pyproject_toml_path.read_text())
35
        if 'tool' in pyproject_toml and 'poetry' in pyproject_toml['tool']:
36
            return pyproject_toml['tool']['poetry']['version']
37
38
    raise RuntimeError('Version information not found in pyproject.toml file.')
39
40
41
def get_version() -> str:
42
    """Returns the version of gvm-tools as a string in `PEP440`_ compliant
43
    format.
44
45
    Returns:
46
        str: Current version of gvm-tools
47
48
    .. _PEP440:
49
       https://www.python.org/dev/peps/pep-0440
50
    """
51
    str_version = get_version_from_pyproject_toml()
52
    return safe_version(str_version)
53