Skip to content

Commit 61fc565

Browse files
committed
refactor(Connection, handle_PARAMETER_STATUS): comply with PEP 632
1 parent dc5d889 commit 61fc565

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

redshift_connector/core.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,14 @@
77
from datetime import datetime as Datetime
88
from datetime import timedelta as Timedelta
99
from decimal import Decimal
10-
from distutils.version import LooseVersion
1110
from hashlib import md5
1211
from itertools import count
1312
from os import getpid
1413
from struct import pack
1514
from typing import TYPE_CHECKING
1615
from warnings import warn
1716

17+
from packaging import version
1818
from scramp import ScramClient # type: ignore
1919

2020
from redshift_connector.config import (
@@ -2111,10 +2111,12 @@ def handle_PARAMETER_STATUS(self: "Connection", data: bytes, ps) -> None:
21112111
self._client_protocol_version = int(value)
21122112
self._enable_protocol_based_conversion_funcs()
21132113
elif key == b"server_version":
2114-
self._server_version: LooseVersion = LooseVersion(value.decode("ascii"))
2115-
if self._server_version < LooseVersion("8.2.0"):
2114+
self._server_version: typing.Union[version.LegacyVersion, version.Version] = version.parse(
2115+
(value.decode("ascii"))
2116+
)
2117+
if self._server_version < version.parse("8.2.0"):
21162118
self._commands_with_count = (b"INSERT", b"DELETE", b"UPDATE", b"MOVE")
2117-
elif self._server_version < LooseVersion("9.0.0"):
2119+
elif self._server_version < version.parse("9.0.0"):
21182120
self._commands_with_count = (b"INSERT", b"DELETE", b"UPDATE", b"MOVE", b"FETCH", b"COPY")
21192121

21202122
def array_inspect(self: "Connection", value):

0 commit comments

Comments
 (0)