Skip to content

Commit f187643

Browse files
bjoernricksgreenbonebot
authored andcommitted
Change: Refactor module layout
Move generic protocol implementation into gvm.protocols.core and GMP request classes to gvm.protocols.gmp.requests.
1 parent 66362ed commit f187643

18 files changed

+18
-23
lines changed

Diff for: gvm/protocols/_protocol.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
from gvm.connections import GvmConnection
1111

12-
from .gmp.core import Connection, Request, Response
12+
from .core import Connection, Request, Response
1313

1414
T = TypeVar("T", default=str)
1515

@@ -34,7 +34,7 @@ def __init__(
3434
self,
3535
connection: GvmConnection,
3636
*,
37-
transform: Callable[[Response], T] = str_transform, # type: ignore[assignment] # this should work with mypy 1.9.0 without an ignore
37+
transform: Callable[[Response], T] = str_transform, # type: ignore[assignment]
3838
):
3939
self._connection = connection
4040
self._protocol = Connection()
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

Diff for: gvm/protocols/gmp/_gmp.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,12 @@
77

88
from gvm.connections import GvmConnection
99
from gvm.errors import GvmError
10+
from gvm.protocols.core import Response
1011

1112
from .._protocol import GvmProtocol, T, str_transform
1213
from ._gmp224 import GMPv224
1314
from ._gmp225 import GMPv225
14-
from .core import Response
15-
from .core.requests import Version
15+
from .requests import Version
1616

1717
SUPPORTED_GMP_VERSIONS = Union[GMPv224, GMPv225]
1818

@@ -53,7 +53,7 @@ def __init__(
5353
self,
5454
connection: GvmConnection,
5555
*,
56-
transform: Callable[[Response], T] = str_transform, # type: ignore[assignment] # this should work with mypy 1.9.0 without an ignore
56+
transform: Callable[[Response], T] = str_transform, # type: ignore[assignment]
5757
):
5858
super().__init__(connection, transform=transform)
5959

@@ -91,7 +91,7 @@ def determine_supported_gmp(self) -> SUPPORTED_GMP_VERSIONS:
9191
f"The GMP version was {major_version}.{minor_version}"
9292
)
9393

94-
return gmp_class(self._connection, transform=self._transform_callable)
94+
return gmp_class(self._connection, transform=self._transform_callable) # type: ignore[arg-type]
9595

9696
def __enter__(self):
9797
self._gmp = self.determine_supported_gmp()

Diff for: gvm/protocols/gmp/_gmp224.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from typing import Optional, Union
66

77
from .._protocol import GvmProtocol, T
8-
from .core.requests import (
8+
from .requests import (
99
Authentication,
1010
PortList,
1111
PortRangeType,

Diff for: gvm/protocols/gmp/_gmp225.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from .._protocol import T
88
from ._gmp224 import GMPv224
9-
from .core.requests import (
9+
from .requests import (
1010
ResourceNames,
1111
ResourceType,
1212
)
File renamed without changes.

Diff for: gvm/protocols/gmp/core/requests/_auth.py renamed to gvm/protocols/gmp/requests/_auth.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,9 @@
33
# SPDX-License-Identifier: GPL-3.0-or-later
44

55
from gvm.errors import RequiredArgument
6+
from gvm.protocols.core import Request
67
from gvm.xml import XmlCommand
78

8-
from .._request import Request
9-
109

1110
class Authentication:
1211

Diff for: gvm/protocols/gmp/core/requests/_port_list.py renamed to gvm/protocols/gmp/requests/_port_list.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,10 @@
66

77
from gvm._enum import Enum
88
from gvm.errors import RequiredArgument
9+
from gvm.protocols.core import Request
910
from gvm.utils import to_bool
1011
from gvm.xml import XmlCommand
1112

12-
from .._request import Request
13-
1413

1514
class PortRangeType(Enum):
1615
"""Enum for port range type"""

Diff for: gvm/protocols/gmp/core/requests/_resource_names.py renamed to gvm/protocols/gmp/requests/_resource_names.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,9 @@
66

77
from gvm._enum import Enum
88
from gvm.errors import RequiredArgument
9+
from gvm.protocols.core import Request
910
from gvm.xml import XmlCommand
1011

11-
from .._request import Request
12-
1312

1413
class ResourceType(Enum):
1514
"""Enum for resource types"""

Diff for: gvm/protocols/gmp/core/requests/_version.py renamed to gvm/protocols/gmp/requests/_version.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,9 @@
22
#
33
# SPDX-License-Identifier: GPL-3.0-or-later
44

5+
from gvm.protocols.core import Request
56
from gvm.xml import XmlCommand
67

7-
from .._request import Request
8-
98

109
class Version:
1110
@staticmethod

Diff for: tests/protocols/gmp/core/test_connection.py renamed to tests/protocols/core/test_connection.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import unittest
66

77
from gvm.errors import GvmError
8-
from gvm.protocols.gmp.core import Connection, InvalidStateError
8+
from gvm.protocols.core import Connection, InvalidStateError
99

1010

1111
class RequestMock:

Diff for: tests/protocols/gmp/core/test_request.py renamed to tests/protocols/core/test_request.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import unittest
66

7-
from gvm.protocols.gmp.core import Request
7+
from gvm.protocols.core import Request
88

99

1010
class RequestMock:

Diff for: tests/protocols/gmp/core/test_response.py renamed to tests/protocols/core/test_response.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import unittest
66

7-
from gvm.protocols.gmp.core import Response, StatusError
7+
from gvm.protocols.core import Response, StatusError
88
from gvm.xml import Element, XmlError
99

1010

Diff for: tests/protocols/gmp/core/test_states.py renamed to tests/protocols/core/test_states.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,12 @@
55
import unittest
66
from unittest.mock import MagicMock
77

8-
from gvm.protocols.gmp.core import Response
9-
from gvm.protocols.gmp.core._connection import (
8+
from gvm.protocols.core import InvalidStateError, Response
9+
from gvm.protocols.core._connection import (
1010
AwaitingResponseState,
1111
Context,
1212
ErrorState,
1313
InitialState,
14-
InvalidStateError,
1514
ReceivingDataState,
1615
)
1716

Diff for: tests/protocols/gmp/core/test_xml_reader.py renamed to tests/protocols/core/test_xml_reader.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import unittest
66

7-
from gvm.protocols.gmp.core._connection import XmlReader
7+
from gvm.protocols.core._connection import XmlReader
88

99

1010
class XmlReaderTestCase(unittest.TestCase):

0 commit comments

Comments
 (0)