PyAthena 3.30.1__tar.gz → 3.31.1__tar.gz

{pyathena-3.30.1 → pyathena-3.31.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyAthena
-Version: 3.30.1
+Version: 3.31.1
 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/
@@ -38,8 +38,8 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Database :: Front-Ends
 Requires-Python: >=3.10
-Requires-Dist: boto3>=1.26.4
-Requires-Dist: botocore>=1.29.4
+Requires-Dist: boto3>=1.38.2
+Requires-Dist: botocore>=1.41.2
 Requires-Dist: fsspec
 Requires-Dist: python-dateutil
 Requires-Dist: tenacity>=4.1.0
@@ -80,7 +80,7 @@ PyAthena is a Python [DB API 2.0 (PEP 249)](https://www.python.org/dev/peps/pep-
 
 ## Requirements
 
-* Python
+- Python
 
   - CPython 3.10, 3.11, 3.12, 3.13, 3.14
 
@@ -136,10 +136,10 @@ Many of the implementations in this library are based on [PyHive](https://github
 
 ## Links
 
-- Documentation: https://pyathena.dev/
-- PyPI Releases: https://pypi.org/project/PyAthena/
-- Source Code: https://github.com/pyathena-dev/PyAthena/
-- Issue Tracker: https://github.com/pyathena-dev/PyAthena/issues
+- Documentation: <https://pyathena.dev/>
+- PyPI Releases: <https://pypi.org/project/PyAthena/>
+- Source Code: <https://github.com/pyathena-dev/PyAthena/>
+- Issue Tracker: <https://github.com/pyathena-dev/PyAthena/issues>
 
 ## Logo
 

{pyathena-3.30.1 → pyathena-3.31.1}/README.md

@@ -21,7 +21,7 @@ PyAthena is a Python [DB API 2.0 (PEP 249)](https://www.python.org/dev/peps/pep-
 
 ## Requirements
 
-* Python
+- Python
 
   - CPython 3.10, 3.11, 3.12, 3.13, 3.14
 
@@ -77,10 +77,10 @@ Many of the implementations in this library are based on [PyHive](https://github
 
 ## Links
 
-- Documentation: https://pyathena.dev/
-- PyPI Releases: https://pypi.org/project/PyAthena/
-- Source Code: https://github.com/pyathena-dev/PyAthena/
-- Issue Tracker: https://github.com/pyathena-dev/PyAthena/issues
+- Documentation: <https://pyathena.dev/>
+- PyPI Releases: <https://pypi.org/project/PyAthena/>
+- Source Code: <https://github.com/pyathena-dev/PyAthena/>
+- Issue Tracker: <https://github.com/pyathena-dev/PyAthena/issues>
 
 ## Logo
 

{pyathena-3.30.1 → pyathena-3.31.1}/pyathena/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '3.30.1'
-__version_tuple__ = version_tuple = (3, 30, 1)
+__version__ = version = '3.31.1'
+__version_tuple__ = version_tuple = (3, 31, 1)
 
 __commit_id__ = commit_id = None

{pyathena-3.30.1 → pyathena-3.31.1}/pyathena/filesystem/s3.py

@@ -1409,11 +1409,18 @@ class S3File(AbstractBufferedFile):
                 )
 
     def _upload_chunk(self, final: bool = False) -> bool:
+        # The return value controls whether fsspec's flush() resets self.buffer
+        # afterwards: it does so only when this returns a value other than False.
+        # Returning ``not final`` keeps the buffer intact on the final flush so a
+        # deferred commit() (autocommit=False, i.e. inside an fsspec transaction)
+        # can still read the bytes; resetting it there would upload an empty
+        # object for small files. Mid-stream chunks (final=False) return True so
+        # fsspec clears the already-uploaded buffer between parts.
         if self.tell() < self.blocksize:
             # Files smaller than block size in size cannot be multipart uploaded.
             if self.autocommit and final:
                 self.commit()
-            return True
+            return not final
 
         if not self.multipart_upload:
             raise RuntimeError("Multipart upload is not initialized.")
@@ -1456,7 +1463,7 @@ class S3File(AbstractBufferedFile):
 
         if self.autocommit and final:
             self.commit()
-        return True
+        return not final
 
     def commit(self) -> None:
         if self.tell() == 0:

{pyathena-3.30.1 → pyathena-3.31.1}/pyathena/model.py

@@ -111,9 +111,13 @@ class AthenaQueryExecution:
         self._query_planning_time_in_millis: int | None = statistics.get(
             "QueryPlanningTimeInMillis", None
         )
+        self._service_pre_processing_time_in_millis: int | None = statistics.get(
+            "ServicePreProcessingTimeInMillis", None
+        )
         self._service_processing_time_in_millis: int | None = statistics.get(
             "ServiceProcessingTimeInMillis", None
         )
+        self._dpu_count: float | None = statistics.get("DpuCount")
         self._data_manifest_location: str | None = statistics.get("DataManifestLocation")
         reuse_info = statistics.get("ResultReuseInformation", {})
         self._reused_previous_result: bool | None = reuse_info.get("ReusedPreviousResult")
@@ -127,6 +131,24 @@ class AthenaQueryExecution:
         acl_conf = result_conf.get("AclConfiguration", {})
         self._s3_acl_option: str | None = acl_conf.get("S3AclOption")
 
+        managed_results_conf = query_execution.get("ManagedQueryResultsConfiguration", {})
+        self._managed_query_results_enabled: bool | None = managed_results_conf.get("Enabled")
+        managed_results_encryption_conf = managed_results_conf.get("EncryptionConfiguration", {})
+        self._managed_query_results_kms_key: str | None = managed_results_encryption_conf.get(
+            "KmsKey"
+        )
+
+        s3_access_grants_conf = query_execution.get("QueryResultsS3AccessGrantsConfiguration", {})
+        self._enable_s3_access_grants: bool | None = s3_access_grants_conf.get(
+            "EnableS3AccessGrants"
+        )
+        self._create_user_level_prefix: bool | None = s3_access_grants_conf.get(
+            "CreateUserLevelPrefix"
+        )
+        self._s3_access_grants_authentication_type: str | None = s3_access_grants_conf.get(
+            "AuthenticationType"
+        )
+
         engine_version = query_execution.get("EngineVersion", {})
         self._selected_engine_version: str | None = engine_version.get(
             "SelectedEngineVersion", None
@@ -224,10 +246,18 @@ class AthenaQueryExecution:
     def query_planning_time_in_millis(self) -> int | None:
         return self._query_planning_time_in_millis
 
+    @property
+    def service_pre_processing_time_in_millis(self) -> int | None:
+        return self._service_pre_processing_time_in_millis
+
     @property
     def service_processing_time_in_millis(self) -> int | None:
         return self._service_processing_time_in_millis
 
+    @property
+    def dpu_count(self) -> float | None:
+        return self._dpu_count
+
     @property
     def output_location(self) -> str | None:
         return self._output_location
@@ -272,6 +302,26 @@ class AthenaQueryExecution:
     def result_reuse_minutes(self) -> int | None:
         return self._result_reuse_minutes
 
+    @property
+    def managed_query_results_enabled(self) -> bool | None:
+        return self._managed_query_results_enabled
+
+    @property
+    def managed_query_results_kms_key(self) -> str | None:
+        return self._managed_query_results_kms_key
+
+    @property
+    def enable_s3_access_grants(self) -> bool | None:
+        return self._enable_s3_access_grants
+
+    @property
+    def create_user_level_prefix(self) -> bool | None:
+        return self._create_user_level_prefix
+
+    @property
+    def s3_access_grants_authentication_type(self) -> str | None:
+        return self._s3_access_grants_authentication_type
+
 
 class AthenaCalculationExecutionStatus:
     """Status information for an Athena calculation execution.

{pyathena-3.30.1 → pyathena-3.31.1}/pyproject.toml

@@ -5,8 +5,8 @@ authors = [
     {name = "laughingman7743", email = "laughingman7743@gmail.com"},
 ]
 dependencies = [
-    "boto3>=1.26.4",
-    "botocore>=1.29.4",
+    "boto3>=1.38.2",
+    "botocore>=1.41.2",
     "tenacity>=4.1.0",
     "fsspec",
     "python-dateutil",
@@ -226,9 +226,9 @@ allowlist_externals =
     make
 commands =
     uv sync --group dev
-    pyathena: make test
-    sqla: make test-sqla
-    sqla_async: make test-sqla-async
+    pyathena: make test/pyathena
+    sqla: make test/sqla
+    sqla_async: make test/sqla-async
 passenv =
     TOXENV
     AWS_*