From 67209252d32026d0b8225d2bd5c58b534fe5844c Mon Sep 17 00:00:00 2001 From: John Clow Date: Mon, 3 Apr 2023 19:01:16 -0700 Subject: [PATCH] Adding typing to Postgres dialect file. --- lib/sqlalchemy/dialects/postgresql/types.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/lib/sqlalchemy/dialects/postgresql/types.py b/lib/sqlalchemy/dialects/postgresql/types.py index a6b82044b3..5341aa0eaa 100644 --- a/lib/sqlalchemy/dialects/postgresql/types.py +++ b/lib/sqlalchemy/dialects/postgresql/types.py @@ -6,10 +6,10 @@ # mypy: ignore-errors import datetime as dt +from typing import Optional from ...sql import sqltypes - _DECIMAL_TYPES = (1231, 1700) _FLOAT_TYPES = (700, 701, 1021, 1022) _INT_TYPES = (20, 21, 23, 26, 1005, 1007, 1016) @@ -150,7 +150,7 @@ class TIMESTAMP(sqltypes.TIMESTAMP): __visit_name__ = "TIMESTAMP" - def __init__(self, timezone=False, precision=None): + def __init__(self, timezone=False, precision: Optional[int] = None): """Construct a TIMESTAMP. :param timezone: boolean value if timezone present, default False @@ -169,7 +169,7 @@ class TIME(sqltypes.TIME): __visit_name__ = "TIME" - def __init__(self, timezone=False, precision=None): + def __init__(self, timezone=False, precision: Optional[int] = None): """Construct a TIME. :param timezone: boolean value if timezone present, default False @@ -189,7 +189,7 @@ class INTERVAL(sqltypes.NativeForEmulated, sqltypes._AbstractInterval): __visit_name__ = "INTERVAL" native = True - def __init__(self, precision=None, fields=None): + def __init__(self, precision: Optional[int] = None, fields: Optional[str] = None): """Construct an INTERVAL. :param precision: optional integer precision value @@ -225,7 +225,7 @@ PGInterval = INTERVAL class BIT(sqltypes.TypeEngine[int]): __visit_name__ = "BIT" - def __init__(self, length=None, varying=False): + def __init__(self, length: Optional[int] = None, varying=False): if not varying: # BIT without VARYING defaults to length 1 self.length = length or 1 -- 2.47.3