]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
Add public package attributes
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Wed, 24 Feb 2021 02:14:10 +0000 (03:14 +0100)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Wed, 24 Feb 2021 02:14:10 +0000 (03:14 +0100)
They are useful to allow people to annotate programs with psycopg3 types.

psycopg3/psycopg3/__init__.py

index a8dfce76b82fef12ad14c160e71b8f934db3bb3a..2888d7e39b641812ce84aa29ed5185f5c3a10ebb 100644 (file)
@@ -8,12 +8,12 @@ from . import pq
 from . import types
 from .copy import Copy, AsyncCopy
 from .adapt import global_adapters
-from .cursor import AsyncCursor, Cursor
+from .cursor import AsyncCursor, Cursor, BaseCursor
 from .errors import Warning, Error, InterfaceError, DatabaseError
 from .errors import DataError, OperationalError, IntegrityError
 from .errors import InternalError, ProgrammingError, NotSupportedError
 from ._column import Column
-from .connection import AsyncConnection, Connection, Notify
+from .connection import BaseConnection, AsyncConnection, Connection, Notify
 from .transaction import Rollback, Transaction, AsyncTransaction
 from .server_cursor import AsyncServerCursor, ServerCursor
 
@@ -38,11 +38,14 @@ BinaryDumper.register(Binary, global_adapters)  # dbapi20
 # this is the canonical place to obtain them and should be used by MyPy too,
 # so that function signatures are consistent with the documentation.
 __all__ = [
+    "__version__",
     "AsyncConnection",
     "AsyncCopy",
     "AsyncCursor",
     "AsyncServerCursor",
     "AsyncTransaction",
+    "BaseConnection",
+    "BaseCursor",
     "Column",
     "Connection",
     "Copy",
@@ -51,4 +54,19 @@ __all__ = [
     "Rollback",
     "ServerCursor",
     "Transaction",
+    # DBAPI exports
+    "connect",
+    "apilevel",
+    "threadsafety",
+    "paramstyle",
+    "Warning",
+    "Error",
+    "InterfaceError",
+    "DatabaseError",
+    "DataError",
+    "OperationalError",
+    "IntegrityError",
+    "InternalError",
+    "ProgrammingError",
+    "NotSupportedError",
 ]