Merge pull request #58 from modern-python/entering-scopes

add methods to enter and close scopes

Author: lesnik512

packages/modern-di/modern_di/container.py

@@ -119,16 +119,29 @@ def reset_override(self, provider_id: str | None = None) -> None:
         else:
             self._overrides.pop(provider_id, None)
 
-    async def __aenter__(self) -> "Container":
+    def async_enter(self) -> "Container":
         self._is_async = True
         return self
 
+    def sync_enter(self) -> "Container":
+        self._is_async = False
+        return self
+
     async def async_close(self) -> None:
         self._check_entered()
         for provider_state in reversed(self._provider_states.values()):
             await provider_state.async_tear_down()
         self._exit()
 
+    def sync_close(self) -> None:
+        self._check_entered()
+        for provider_state in reversed(self._provider_states.values()):
+            provider_state.sync_tear_down()
+        self._exit()
+
+    async def __aenter__(self) -> "Container":
+        return self.async_enter()
+
     async def __aexit__(
         self,
         exc_type: type[BaseException] | None,
@@ -138,16 +151,12 @@ async def __aexit__(
         await self.async_close()
 
     def __enter__(self) -> "Container":
-        self._is_async = False
-        return self
+        return self.sync_enter()
 
     def __exit__(
         self,
         exc_type: type[BaseException] | None,
         exc_value: BaseException | None,
         traceback: types.TracebackType | None,
     ) -> None:
-        self._check_entered()
-        for provider_state in reversed(self._provider_states.values()):
-            provider_state.sync_tear_down()
-        self._exit()
+        self.sync_close()