Remove exports from infrahub_sdk.__init__.py

Author: ogenstad

changelog/+8dff6891.removed.md

@@ -0,0 +1 @@
+Breaking change: Removed all exports from infrahub_sdk/__init__.py except InfrahubClient, InfrahubClientSync and Config. If you previously imported other classes such as InfrahubNode from the root level these need to change to instead be an absolute path.

infrahub_sdk/__init__.py

@@ -2,76 +2,13 @@
 
 import importlib.metadata
 
-from .analyzer import GraphQLOperation, GraphQLQueryAnalyzer, GraphQLQueryVariable
-from .batch import InfrahubBatch
-from .branch import InfrahubBranchManager, InfrahubBranchManagerSync
 from .client import InfrahubClient, InfrahubClientSync
 from .config import Config
-from .exceptions import (
-    AuthenticationError,
-    Error,
-    GraphQLError,
-    NodeNotFoundError,
-    ServerNotReachableError,
-    ServerNotResponsiveError,
-    ValidationError,
-)
-from .graphql import Mutation, Query
-from .node import InfrahubNode, InfrahubNodeSync
-from .schema import (
-    AttributeSchema,
-    GenericSchema,
-    InfrahubRepositoryConfig,
-    InfrahubSchema,
-    MainSchemaTypes,
-    NodeSchema,
-    ProfileSchema,
-    RelationshipCardinality,
-    RelationshipKind,
-    RelationshipSchema,
-    SchemaRoot,
-)
-from .store import NodeStore, NodeStoreSync
-from .timestamp import Timestamp
-from .uuidt import UUIDT, generate_uuid
 
 __all__ = [
-    "AttributeSchema",
-    "AuthenticationError",
     "Config",
-    "Error",
-    "InfrahubBatch",
-    "InfrahubBranchManager",
-    "InfrahubBranchManagerSync",
     "InfrahubClient",
     "InfrahubClientSync",
-    "InfrahubNode",
-    "InfrahubNodeSync",
-    "InfrahubRepositoryConfig",
-    "InfrahubSchema",
-    "generate_uuid",
-    "GenericSchema",
-    "GraphQLQueryAnalyzer",
-    "GraphQLQueryVariable",
-    "GraphQLError",
-    "GraphQLOperation",
-    "MainSchemaTypes",
-    "NodeNotFoundError",
-    "NodeSchema",
-    "Mutation",
-    "NodeStore",
-    "NodeStoreSync",
-    "ProfileSchema",
-    "Query",
-    "RelationshipSchema",
-    "RelationshipCardinality",
-    "RelationshipKind",
-    "SchemaRoot",
-    "ServerNotReachableError",
-    "ServerNotResponsiveError",
-    "Timestamp",
-    "UUIDT",
-    "ValidationError",
 ]
 
 try:

infrahub_sdk/ctl/generator.py

@@ -3,11 +3,11 @@
 
 from rich.console import Console
 
-from .. import InfrahubNode
 from ..ctl import config
 from ..ctl.client import initialize_client
 from ..ctl.repository import get_repository_config
 from ..ctl.utils import execute_graphql_query, parse_cli_vars
+from ..node import InfrahubNode
 from ..schema import InfrahubRepositoryConfig
 
 

tests/unit/sdk/test_schema.py

@@ -5,9 +5,9 @@
 import pytest
 from rich.console import Console
 
-from infrahub_sdk import Config, InfrahubClient, InfrahubClientSync, ValidationError
+from infrahub_sdk import Config, InfrahubClient, InfrahubClientSync
 from infrahub_sdk.ctl.schema import display_schema_load_errors
-from infrahub_sdk.exceptions import SchemaNotFoundError
+from infrahub_sdk.exceptions import SchemaNotFoundError, ValidationError
 from infrahub_sdk.schema import (
     InfrahubCheckDefinitionConfig,
     InfrahubJinja2TransformConfig,

tests/unit/sdk/test_store.py

@@ -1,6 +1,7 @@
 import pytest
 
-from infrahub_sdk import InfrahubNode, NodeStore
+from infrahub_sdk.node import InfrahubNode
+from infrahub_sdk.store import NodeStore
 
 client_types = ["standard", "sync"]