Merge pull request #244 from yassun7010/add_connection_enter

feat: connection に __enter__ を追加しました。

Author: ya7010

turu-core/src/turu/core/connection.py

@@ -1,10 +1,11 @@
 from abc import abstractmethod
 from typing import Any, Optional, Sequence, Tuple, Type
 
+from typing_extensions import Never, Self, override
+
 import turu.core.cursor
 from turu.core.protocols.connection import ConnectionProtocol
 from turu.core.protocols.cursor import Parameters
-from typing_extensions import Never, Self, override
 
 
 class Connection(ConnectionProtocol):
@@ -125,3 +126,9 @@ def executemany_map(
         """
 
         return self.cursor().executemany_map(row_type, operation, seq_of_parameters)
+
+    def __enter__(self) -> Self:
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback) -> None:
+        self.close()

turu-snowflake/src/turu/snowflake/features.py

@@ -1,8 +1,9 @@
 from typing import Generic, TypeVar
 
-from turu.core.features import _NotSupportFeature
 from typing_extensions import TypeAlias
 
+from turu.core.features import _NotSupportFeature
+
 T = TypeVar("T")
 
 
@@ -48,6 +49,7 @@ class _NotSupportFeatureT(Generic[T]):
     from typing import TypeVar
 
     import pandera  # type: ignore[import]  # noqa: F401
+    import pandera.typing  # type: ignore[import]  # noqa: F401
 
     USE_PANDERA = True
     PanderaDataFrame: TypeAlias = pandera.typing.DataFrame[T]  # type: ignore