PyAthena 3.33.0__tar.gz → 3.34.0__tar.gz

{pyathena-3.33.0 → pyathena-3.34.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyAthena
-Version: 3.33.0
+Version: 3.34.0
 Summary: Python DB API 2.0 (PEP 249) client for Amazon Athena
 Project-URL: homepage, https://github.com/pyathena-dev/PyAthena/
 Project-URL: repository, https://github.com/pyathena-dev/PyAthena/

{pyathena-3.33.0 → pyathena-3.34.0}/pyathena/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '3.33.0'
-__version_tuple__ = version_tuple = (3, 33, 0)
+__version__ = version = '3.34.0'
+__version_tuple__ = version_tuple = (3, 34, 0)
 
 __commit_id__ = commit_id = None

{pyathena-3.33.0 → pyathena-3.34.0}/pyathena/arrow/result_set.py

@@ -315,7 +315,7 @@ class AthenaArrowResultSet(AthenaResultSet):
                 ),
             )
         except Exception as e:
-            _logger.exception("Failed to read %s/%s.", bucket, key)
+            _logger.exception(f"Failed to read {bucket}/{key}.")
             raise OperationalError(*e.args) from e
 
     def _read_parquet(self) -> Table:
@@ -333,7 +333,7 @@ class AthenaArrowResultSet(AthenaResultSet):
             dataset = parquet.ParquetDataset(f"{bucket}/{key}", filesystem=self._fs)
             return dataset.read(use_threads=True)
         except Exception as e:
-            _logger.exception("Failed to read %s/%s.", bucket, key)
+            _logger.exception(f"Failed to read {bucket}/{key}.")
             raise OperationalError(*e.args) from e
 
     def _as_arrow(self) -> Table:

pyathena-3.34.0/pyathena/filesystem/__init__.py

@@ -0,0 +1,35 @@
+import logging
+
+import fsspec
+
+from pyathena.filesystem.s3 import S3FileSystem
+
+_logger = logging.getLogger(__name__)
+
+
+def register_s3_filesystem() -> None:
+    """Register PyAthena's S3 filesystem as fsspec's "s3" / "s3a" protocols.
+
+    PyAthena registers its own filesystem so that the pandas/polars result
+    sets can read query results from S3 without depending on s3fs. The
+    registration replaces fsspec's default lazy mapping of the "s3" protocol
+    to s3fs, which means ``fsspec.filesystem("s3")`` returns PyAthena's
+    implementation and s3fs-specific settings (e.g., the ``S3FS_LOGGING_LEVEL``
+    environment variable) have no effect.
+
+    A filesystem class that has already been registered explicitly is also
+    overwritten, with a warning log. To restore another implementation,
+    re-register it after importing ``pyathena.pandas`` / ``pyathena.polars``::
+
+        fsspec.register_implementation("s3", s3fs.S3FileSystem, clobber=True)
+    """
+    for protocol in ("s3", "s3a"):
+        registered = fsspec.registry.get(protocol)
+        if registered is not None and registered is not S3FileSystem:
+            _logger.warning(
+                f"The fsspec {protocol!r} protocol is already registered as "
+                f"{registered.__module__}.{registered.__qualname__} and will be overwritten by "
+                f"{S3FileSystem.__module__}.{S3FileSystem.__qualname__}."
+            )
+        _logger.debug(f"Registering {S3FileSystem} as the fsspec {protocol!r} protocol.")
+        fsspec.register_implementation(protocol, S3FileSystem, clobber=True)