Add __all__ to edgedb/__init__.py

fixes #301

Author: fmoor

edgedb/__init__.py

@@ -44,6 +44,37 @@
 
 from .errors._base import EdgeDBError, EdgeDBMessage
 
+__all__ = [
+    "Array",
+    "AsyncIOClient",
+    "AsyncIOExecutor",
+    "AsyncIOReadOnlyExecutor",
+    "Client",
+    "ConfigMemory",
+    "DateDuration",
+    "EdgeDBError",
+    "EdgeDBMessage",
+    "EnumValue",
+    "Executor",
+    "IsolationLevel",
+    "Link",
+    "LinkSet",
+    "NamedTuple",
+    "Object",
+    "Range",
+    "ReadOnlyExecutor",
+    "RelativeDuration",
+    "RetryCondition",
+    "RetryOptions",
+    "Set",
+    "State",
+    "TransactionOptions",
+    "Tuple",
+    "create_async_client",
+    "create_client",
+    "default_backoff",
+]
+
 
 # The below is generated by `make gen-errors`.
 # DO NOT MODIFY BY HAND.
@@ -58,6 +89,8 @@
     TypeSpecNotFoundError,
     UnexpectedMessageError,
     InputDataError,
+    ParameterTypeMismatchError,
+    StateMismatchError,
     ResultCardinalityMismatchError,
     CapabilityError,
     UnsupportedCapabilityError,
@@ -102,11 +135,16 @@
     DuplicateFunctionDefinitionError,
     DuplicateConstraintDefinitionError,
     DuplicateCastDefinitionError,
+    SessionTimeoutError,
+    IdleSessionTimeoutError,
     QueryTimeoutError,
+    TransactionTimeoutError,
+    IdleTransactionTimeoutError,
     ExecutionError,
     InvalidValueError,
     DivisionByZeroError,
     NumericOutOfRangeError,
+    AccessPolicyError,
     IntegrityError,
     ConstraintViolationError,
     CardinalityViolationError,
@@ -118,6 +156,10 @@
     ConfigurationError,
     AccessError,
     AuthenticationError,
+    AvailabilityError,
+    BackendUnavailableError,
+    BackendError,
+    UnsupportedBackendFeatureError,
     LogMessage,
     WarningMessage,
     ClientError,
@@ -132,5 +174,103 @@
     UnknownArgumentError,
     InvalidArgumentError,
     NoDataError,
+    InternalClientError,
 )
+
+__all__.extend([
+    "InternalServerError",
+    "UnsupportedFeatureError",
+    "ProtocolError",
+    "BinaryProtocolError",
+    "UnsupportedProtocolVersionError",
+    "TypeSpecNotFoundError",
+    "UnexpectedMessageError",
+    "InputDataError",
+    "ParameterTypeMismatchError",
+    "StateMismatchError",
+    "ResultCardinalityMismatchError",
+    "CapabilityError",
+    "UnsupportedCapabilityError",
+    "DisabledCapabilityError",
+    "QueryError",
+    "InvalidSyntaxError",
+    "EdgeQLSyntaxError",
+    "SchemaSyntaxError",
+    "GraphQLSyntaxError",
+    "InvalidTypeError",
+    "InvalidTargetError",
+    "InvalidLinkTargetError",
+    "InvalidPropertyTargetError",
+    "InvalidReferenceError",
+    "UnknownModuleError",
+    "UnknownLinkError",
+    "UnknownPropertyError",
+    "UnknownUserError",
+    "UnknownDatabaseError",
+    "UnknownParameterError",
+    "SchemaError",
+    "SchemaDefinitionError",
+    "InvalidDefinitionError",
+    "InvalidModuleDefinitionError",
+    "InvalidLinkDefinitionError",
+    "InvalidPropertyDefinitionError",
+    "InvalidUserDefinitionError",
+    "InvalidDatabaseDefinitionError",
+    "InvalidOperatorDefinitionError",
+    "InvalidAliasDefinitionError",
+    "InvalidFunctionDefinitionError",
+    "InvalidConstraintDefinitionError",
+    "InvalidCastDefinitionError",
+    "DuplicateDefinitionError",
+    "DuplicateModuleDefinitionError",
+    "DuplicateLinkDefinitionError",
+    "DuplicatePropertyDefinitionError",
+    "DuplicateUserDefinitionError",
+    "DuplicateDatabaseDefinitionError",
+    "DuplicateOperatorDefinitionError",
+    "DuplicateViewDefinitionError",
+    "DuplicateFunctionDefinitionError",
+    "DuplicateConstraintDefinitionError",
+    "DuplicateCastDefinitionError",
+    "SessionTimeoutError",
+    "IdleSessionTimeoutError",
+    "QueryTimeoutError",
+    "TransactionTimeoutError",
+    "IdleTransactionTimeoutError",
+    "ExecutionError",
+    "InvalidValueError",
+    "DivisionByZeroError",
+    "NumericOutOfRangeError",
+    "AccessPolicyError",
+    "IntegrityError",
+    "ConstraintViolationError",
+    "CardinalityViolationError",
+    "MissingRequiredError",
+    "TransactionError",
+    "TransactionConflictError",
+    "TransactionSerializationError",
+    "TransactionDeadlockError",
+    "ConfigurationError",
+    "AccessError",
+    "AuthenticationError",
+    "AvailabilityError",
+    "BackendUnavailableError",
+    "BackendError",
+    "UnsupportedBackendFeatureError",
+    "LogMessage",
+    "WarningMessage",
+    "ClientError",
+    "ClientConnectionError",
+    "ClientConnectionFailedError",
+    "ClientConnectionFailedTemporarilyError",
+    "ClientConnectionTimeoutError",
+    "ClientConnectionClosedError",
+    "InterfaceError",
+    "QueryArgumentError",
+    "MissingArgumentError",
+    "UnknownArgumentError",
+    "InvalidArgumentError",
+    "NoDataError",
+    "InternalClientError",
+])
 # </ERRORS-AUTOGEN>

tools/gen_init.py

@@ -32,7 +32,11 @@
 
     names = re.findall(r'^class\s+(?P<name>\w+)', errors_txt, re.M)
     names_list = '\n'.join(f'    {name},' for name in names)
-    code = f'''from .errors import (\n{names_list}\n)\n'''.splitlines()
+    all_list = '\n'.join(f'    "{name}",' for name in names)
+    code = (
+        f'''from .errors import (\n{names_list}\n)\n'''
+        f'''\n__all__.extend([\n{all_list}\n])\n'''
+    ).splitlines()
 
     with open(init_fn, 'rt') as f:
         lines = f.read().splitlines()