Improve stubs for psycopg2.

Author: andrew-chang-dewitt

stubs/psycopg2/__init__.pyi

@@ -1,3 +1,19 @@
+# pylint: disable=missing-module-docstring
+# pylint: disable=missing-class-docstring
+# pylint: disable=missing-function-docstring
+# pylint: disable=no-name-in-module
+# pylint: disable=unused-import
+# pylint: disable=unused-argument
+# pylint: disable=multiple-statements
+# pylint: disable=invalid-name
+# pylint: disable=invalid-length-returned
+# pylint: disable=too-few-public-methods
+# pylint: disable=too-many-public-methods
+# pylint: disable=no-self-use
+# pylint: disable=redefined-builtin
+# pylint: disable=super-init-not-called
+
+from typing import Any, Optional
 from psycopg2._psycopg import (
     BINARY,
     Binary,
@@ -27,7 +43,6 @@ from psycopg2._psycopg import (
     paramstyle,
     threadsafety
 )
-from typing import Any, Optional
 
 connection = connection
 OperationalError = OperationalError

stubs/psycopg2/_ipaddress.pyi

@@ -1,9 +1,33 @@
-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
+# pylint: disable=missing-function-docstring
+# pylint: disable=missing-class-docstring
+# pylint: disable=missing-module-docstring
+# pylint: disable=unused-argument
+# pylint: disable=multiple-statements
+# pylint: disable=invalid-name
+# pylint: disable=invalid-length-returned
+# pylint: disable=too-few-public-methods
+# pylint: disable=too-many-public-methods
+# pylint: disable=no-self-use
+# pylint: disable=redefined-builtin
+# pylint: disable=super-init-not-called
+# pylint: disable=unused-import
+# pylint: disable=useless-import-alias
+# pylint: disable=line-too-long
+
 from typing import Any, Optional
 
+from psycopg2.extensions import (
+    QuotedString,
+    new_array_type,
+    new_type,
+    register_adapter,
+    register_type,
+)
+
 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): ...
+def cast_interface(s: Any, cur: Optional[Any] = ...) -> Any: ...
+def cast_network(s: Any, cur: Optional[Any] = ...) -> Any: ...
+def adapt_ipaddress(obj: Any) -> Any: ...

stubs/psycopg2/_json.pyi

@@ -1,3 +1,19 @@
+# pylint: disable=missing-module-docstring
+# pylint: disable=missing-class-docstring
+# pylint: disable=missing-function-docstring
+# pylint: disable=no-name-in-module
+# pylint: disable=unused-import
+# pylint: disable=unused-argument
+# pylint: disable=multiple-statements
+# pylint: disable=invalid-name
+# pylint: disable=invalid-length-returned
+# pylint: disable=too-few-public-methods
+# pylint: disable=too-many-public-methods
+# pylint: disable=too-many-arguments
+# pylint: disable=no-self-use
+# pylint: disable=redefined-builtin
+# pylint: disable=super-init-not-called
+
 from typing import Any, Optional
 
 from psycopg2._psycopg import (
@@ -23,13 +39,23 @@ class Json:
     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_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: ...
+    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: ...
+    conn_or_curs: Optional[Any] = ...,
+    globally: bool = ...,
+    loads: Optional[Any] = ...) -> Any: ...