Improve stubs for psycopg2

Author: andrew-chang-dewitt

stubs/psycopg2/_ipaddress.pyi

@@ -0,0 +1,9 @@
+from psycopg2.extensions import QuotedString as QuotedString, new_array_type as new_array_type, new_type as new_type, register_adapter as register_adapter, register_type as register_type
+from typing import Any, Optional
+
+ipaddress: Any
+
+def register_ipaddress(conn_or_curs: Optional[Any] = ...) -> None: ...
+def cast_interface(s: Any, cur: Optional[Any] = ...): ...
+def cast_network(s: Any, cur: Optional[Any] = ...): ...
+def adapt_ipaddress(obj: Any): ...

stubs/psycopg2/_json.pyi

@@ -0,0 +1,35 @@
+from typing import Any, Optional
+
+from psycopg2._psycopg import (
+    ISQLQuote,
+    QuotedString,
+    new_array_type,
+    new_type,
+    register_type
+)
+
+JSON_OID: int
+JSONARRAY_OID: int
+JSONB_OID: int
+JSONBARRAY_OID: int
+
+
+class Json:
+    adapted: Any = ...
+    def __init__(self, adapted: Any, dumps: Optional[Any] = ...) -> None: ...
+    def __conform__(self, proto: Any) -> Any: ...
+    def dumps(self, obj: Any) -> Any: ...
+    def prepare(self, conn: Any) -> None: ...
+    def getquoted(self) -> Any: ...
+
+
+def register_json(conn_or_curs: Optional[Any] = ..., globally: bool = ..., loads: Optional[Any]
+                  = ..., oid: Optional[Any] = ..., array_oid: Optional[Any] = ..., name: str = ...) -> Any: ...
+
+
+def register_default_json(
+    conn_or_curs: Optional[Any] = ..., globally: bool = ..., loads: Optional[Any] = ...) -> Any: ...
+
+
+def register_default_jsonb(
+    conn_or_curs: Optional[Any] = ..., globally: bool = ..., loads: Optional[Any] = ...) -> Any: ...

stubs/psycopg2/_range.pyi

@@ -0,0 +1,87 @@
+from typing import Any, Optional
+
+from psycopg2._psycopg import (
+    InterfaceError,
+    ProgrammingError
+)
+
+
+class Range:
+    def __init__(self, lower: Optional[Any] = ..., upper: Optional[Any]
+                 = ..., bounds: str = ..., empty: bool = ...) -> None: ...
+
+    @property
+    def lower(self) -> Any: ...
+    @property
+    def upper(self) -> Any: ...
+    @property
+    def isempty(self) -> Any: ...
+    @property
+    def lower_inf(self) -> Any: ...
+    @property
+    def upper_inf(self) -> Any: ...
+    @property
+    def lower_inc(self) -> Any: ...
+    @property
+    def upper_inc(self) -> Any: ...
+    def __contains__(self, x: Any) -> Any: ...
+    def __bool__(self) -> Any: ...
+    def __nonzero__(self) -> Any: ...
+    def __eq__(self, other: Any) -> Any: ...
+    def __ne__(self, other: Any) -> Any: ...
+    def __hash__(self) -> Any: ...
+    def __lt__(self, other: Any) -> Any: ...
+    def __le__(self, other: Any) -> Any: ...
+    def __gt__(self, other: Any) -> Any: ...
+    def __ge__(self, other: Any) -> Any: ...
+
+
+def register_range(pgrange: Any, pyrange: Any,
+                   conn_or_curs: Any, globally: bool = ...) -> Any: ...
+
+
+class RangeAdapter:
+    name: Any = ...
+    adapted: Any = ...
+    def __init__(self, adapted: Any) -> None: ...
+    def __conform__(self, proto: Any) -> Any: ...
+    def prepare(self, conn: Any) -> None: ...
+    def getquoted(self) -> Any: ...
+
+
+class RangeCaster:
+    subtype_oid: Any = ...
+    typecaster: Any = ...
+    array_typecaster: Any = ...
+    def __init__(self, pgrange: Any, pyrange: Any, oid: Any,
+                 subtype_oid: Any, array_oid: Optional[Any] = ...) -> None: ...
+
+    def parse(self, s: Any, cur: Optional[Any] = ...) -> Any: ...
+
+
+class NumericRange(Range):
+    ...
+
+
+class DateRange(Range):
+    ...
+
+
+class DateTimeRange(Range):
+    ...
+
+
+class DateTimeTZRange(Range):
+    ...
+
+
+class NumberRangeAdapter(RangeAdapter):
+    def getquoted(self) -> Any: ...
+
+
+int4range_caster: Any
+int8range_caster: Any
+numrange_caster: Any
+daterange_caster: Any
+tsrange_caster: Any
+tstzrange_caster: Any