sibi-dst 2025.8.7__py3-none-any.whl → 2025.8.8__py3-none-any.whl

sibi_dst/utils/storage_hive.py

@@ -1,4 +1,7 @@
 from __future__ import annotations
+
+import asyncio
+
 import pandas as pd
 import dask.dataframe as dd
 from typing import Iterable, Optional, List, Tuple, Union
@@ -192,4 +195,38 @@ class HiveDatePartitionedStore:
             clauses.append([("yyyy","==",y)])
         for m in range(1, eM):
             clauses.append([("yyyy","==",eY),("mm","==",m)])
-        return clauses
+        return clauses
+
+    async def write_async(
+            self,
+            df: dd.DataFrame,
+            *,
+            repartition: int | None = None,
+            overwrite: bool = False,
+            timeout: float | None = None,
+    ) -> None:
+        async def _run():
+            return await asyncio.to_thread(self.write, df, repartition=repartition, overwrite=overwrite)
+
+        return await (asyncio.wait_for(_run(), timeout) if timeout else _run())
+
+    async def read_range_async(
+            self,
+            start, end, *, columns: Iterable[str] | None = None, timeout: float | None = None
+    ) -> dd.DataFrame:
+        async def _run():
+            return await asyncio.to_thread(self.read_range, start, end, columns=columns)
+
+        return await (asyncio.wait_for(_run(), timeout) if timeout else _run())
+
+    async def read_month_async(self, year: int, month: int, *, columns=None, timeout: float | None = None):
+        async def _run():
+            return await asyncio.to_thread(self.read_month, year, month, columns=columns)
+
+        return await (asyncio.wait_for(_run(), timeout) if timeout else _run())
+
+    async def read_day_async(self, year: int, month: int, day: int, *, columns=None, timeout: float | None = None):
+        async def _run():
+            return await asyncio.to_thread(self.read_day, year, month, day, columns=columns)
+
+        return await (asyncio.wait_for(_run(), timeout) if timeout else _run())