feat: Better package structure

Make the client class usable as impact_stack.rest.Client. Use __all__ to take more control over the externally visible symbols.
moreonion · Oct 9, 2023 · b76d7c1 · b76d7c1
1 parent 058fa25
commit b76d7c1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/impact_stack/rest/__init__.py b/impact_stack/rest/__init__.py
@@ -107,3 +107,7 @@ def get_client(self, app_slug, api_version=None, needs_auth=True):
             auth=self.auth_middleware if needs_auth else None,
             request_timeout=config["timeout"],
         )
+
+
+Client = rest.Client
+__all__ = ["AuthMiddleware", "ClientFactory", "Client"]
diff --git a/tests/rest_test.py b/tests/rest_test.py
@@ -6,7 +6,7 @@
 import requests
 import requests.exceptions
 
-from impact_stack.rest import rest
+from impact_stack import rest
 
 
 class ClientTest: