Simplify import of services

corbado · Oct 14, 2024 · 58726c5 · 58726c5
1 parent 50ebd60
commit 58726c5
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 11 deletions.
diff --git a/src/corbado_python_sdk/__init__.py b/src/corbado_python_sdk/__init__.py
@@ -12,6 +12,7 @@
     UserStatus,
     UserUpdateReq,
 )
+from .services import IdentifierService, SessionService, UserService
 
 __all__ = [
     "SessionValidationResult",
@@ -26,4 +27,7 @@
     "UserEntity",
     "CorbadoSDK",
     "Config",
+    "IdentifierService",
+    "SessionService",
+    "UserService",
 ]
diff --git a/tests/integration/test_identifier_service.py b/tests/integration/test_identifier_service.py
@@ -3,12 +3,15 @@
 
 from typing_extensions import List
 
-from corbado_python_sdk import CorbadoSDK, Identifier, IdentifierStatus, IdentifierType
-from corbado_python_sdk.exceptions import ServerException
-from corbado_python_sdk.generated.models.identifier_list import IdentifierList
-from corbado_python_sdk.services.implementation.identifier_service import (
+from corbado_python_sdk import (
+    CorbadoSDK,
+    Identifier,
     IdentifierService,
+    IdentifierStatus,
+    IdentifierType,
 )
+from corbado_python_sdk.exceptions import ServerException
+from corbado_python_sdk.generated.models.identifier_list import IdentifierList
 from tests.utils import TestUtils
 
 CANNOT_BE_BLANK = "cannot be blank"

diff --git a/tests/integration/test_user_service.py b/tests/integration/test_user_service.py
@@ -1,8 +1,7 @@
 import unittest
 
-from corbado_python_sdk import CorbadoSDK, UserEntity, UserStatus
+from corbado_python_sdk import CorbadoSDK, UserEntity, UserService, UserStatus
 from corbado_python_sdk.exceptions import ServerException
-from corbado_python_sdk.services.implementation import UserService
 from tests.utils import TestUtils
 
 

diff --git a/tests/unit/test_session_service.py b/tests/unit/test_session_service.py
@@ -7,8 +7,12 @@
 from jwt import encode
 from pydantic import ValidationError
 
-from corbado_python_sdk import Config, CorbadoSDK, SessionValidationResult
-from corbado_python_sdk.services import SessionService
+from corbado_python_sdk import (
+    Config,
+    CorbadoSDK,
+    SessionService,
+    SessionValidationResult,
+)
 
 TEST_NAME = "Test Name"
 TEST_EMAIL = "[email protected]"
@@ -114,9 +118,7 @@ def _generate_jwt(cls, iss: str, exp: int, nbf: int) -> str:
 class TestSessionService(TestBase):
     def test_get_and_validate_short_session_value(self):
         for valid, token in self._provide_jwts():
-            result: SessionValidationResult = self.session_service.get_and_validate_short_session_value(
-                short_session=token
-            )
+            result: SessionValidationResult = self.session_service.get_and_validate_short_session_value(short_session=token)
 
             self.assertEqual(first=valid, second=result.authenticated)
             self.assertEqual(first=valid, second=result.error is None)