Skip to content

Commit d27cbcd

Browse files
committed
Refactor SSLMode.parse so type checkers undertstand it
1 parent ea9cb59 commit d27cbcd

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

asyncpg/connect_utils.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -41,10 +41,10 @@ class SSLMode(enum.IntEnum):
4141
verify_full = 5
4242

4343
@classmethod
44-
def parse(cls, sslmode):
45-
if isinstance(sslmode, cls):
46-
return sslmode
47-
return getattr(cls, sslmode.replace('-', '_'))
44+
def parse(cls, sslmode: typing.Union[str, "SSLMode"]) -> "SSLMode":
45+
if isinstance(sslmode, str):
46+
return getattr(cls, sslmode.replace('-', '_'))
47+
return sslmode
4848

4949

5050
class SSLNegotiation(compat.StrEnum):

0 commit comments

Comments
 (0)