mirror of
https://github.com/outbackdingo/patroni.git
synced 2026-08-25 14:53:37 +00:00
Alexander KukushkinandGitHub
76b3b99de2
- added pyrightconfig.json with typeCheckingMode=strict - added type hints to all files except api.py - added type stubs for dns, etcd, consul, kazoo, pysyncobj and other modules - added type stubs for psycopg2 and urllib3 with some little fixes - fixes most of the issues reported by pyright - remaining issues will be addressed later, along with enabling CI linting task
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
from collections.abc import Callable
|
|
from typing import Any, TypeVar, overload
|
|
|
|
# connection and cursor not available at runtime
|
|
from psycopg2._psycopg import (
|
|
BINARY as BINARY,
|
|
DATETIME as DATETIME,
|
|
NUMBER as NUMBER,
|
|
ROWID as ROWID,
|
|
STRING as STRING,
|
|
Binary as Binary,
|
|
DatabaseError as DatabaseError,
|
|
DataError as DataError,
|
|
Date as Date,
|
|
DateFromTicks as DateFromTicks,
|
|
Error as Error,
|
|
IntegrityError as IntegrityError,
|
|
InterfaceError as InterfaceError,
|
|
InternalError as InternalError,
|
|
NotSupportedError as NotSupportedError,
|
|
OperationalError as OperationalError,
|
|
ProgrammingError as ProgrammingError,
|
|
Time as Time,
|
|
TimeFromTicks as TimeFromTicks,
|
|
Timestamp as Timestamp,
|
|
TimestampFromTicks as TimestampFromTicks,
|
|
Warning as Warning,
|
|
__libpq_version__ as __libpq_version__,
|
|
apilevel as apilevel,
|
|
connection as connection,
|
|
cursor as cursor,
|
|
paramstyle as paramstyle,
|
|
threadsafety as threadsafety,
|
|
)
|
|
|
|
__version__: str
|
|
|
|
_T_conn = TypeVar("_T_conn", bound=connection)
|
|
|
|
@overload
|
|
def connect(dsn: str, connection_factory: Callable[..., _T_conn], cursor_factory: None = None, **kwargs: Any) -> _T_conn: ...
|
|
@overload
|
|
def connect(
|
|
dsn: str | None = None, *, connection_factory: Callable[..., _T_conn], cursor_factory: None = None, **kwargs: Any
|
|
) -> _T_conn: ...
|
|
@overload
|
|
def connect(
|
|
dsn: str | None = None,
|
|
connection_factory: Callable[..., connection] | None = None,
|
|
cursor_factory: Callable[..., cursor] | None = None,
|
|
**kwargs: Any,
|
|
) -> connection: ...
|