kabukit 0.7.1__py3-none-any.whl → 0.7.2__py3-none-any.whl

kabukit/core/base.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import datetime
 from typing import TYPE_CHECKING
+from zoneinfo import ZoneInfo
 
 import polars as pl
 
@@ -30,7 +31,7 @@ class Base:
     def write(self) -> Path:
         data_dir = self.data_dir()
         data_dir.mkdir(parents=True, exist_ok=True)
-        path = datetime.datetime.today().strftime("%Y%m%d")  # noqa: DTZ002
+        path = datetime.datetime.now(ZoneInfo("Asia/Tokyo")).strftime("%Y%m%d")
         filename = data_dir / f"{path}.parquet"
         self.data.write_parquet(filename)
         return filename

kabukit/edinet/client.py

@@ -6,7 +6,9 @@ import zipfile
 from enum import StrEnum
 from typing import TYPE_CHECKING
 
+import httpx
 from polars import DataFrame
+from tenacity import retry, retry_if_exception, stop_after_attempt, wait_exponential
 
 from kabukit.core.client import Client
 from kabukit.utils.config import load_dotenv
@@ -24,6 +26,11 @@ API_VERSION = "v2"
 BASE_URL = f"https://api.edinet-fsa.go.jp/api/{API_VERSION}"
 
 
+def is_retryable(e: BaseException) -> bool:
+    """Return True if the exception is a retryable network error."""
+    return isinstance(e, (httpx.ConnectTimeout, httpx.ReadTimeout, httpx.ConnectError))
+
+
 class AuthKey(StrEnum):
     """Environment variable keys for EDINET authentication."""
 
@@ -43,6 +50,12 @@ class EdinetClient(Client):
         if api_key:
             self.client.params = {"Subscription-Key": api_key}
 
+    @retry(
+        reraise=True,
+        stop=stop_after_attempt(3),
+        wait=wait_exponential(multiplier=1, min=2, max=10),
+        retry=retry_if_exception(is_retryable),
+    )
     async def get(self, url: str, params: QueryParamTypes) -> Response:
         resp = await self.client.get(url, params=params)
         resp.raise_for_status()

kabukit/edinet/doc.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import datetime
 from typing import TYPE_CHECKING
+from zoneinfo import ZoneInfo
 
 import polars as pl
 
@@ -11,7 +12,11 @@ if TYPE_CHECKING:
 
 def clean_list(df: DataFrame, date: str | datetime.date) -> DataFrame:
     if isinstance(date, str):
-        date = datetime.datetime.strptime(date, "%Y-%m-%d").date()  # noqa: DTZ007
+        date = (
+            datetime.datetime.strptime(date, "%Y-%m-%d")
+            .replace(tzinfo=ZoneInfo("Asia/Tokyo"))
+            .date()
+        )
 
     null_columns = [c for c in df.columns if df[c].dtype == pl.Null]
 
@@ -21,7 +26,11 @@ def clean_list(df: DataFrame, date: str | datetime.date) -> DataFrame:
         )
         .with_columns(
             pl.lit(date).alias("Date"),
-            pl.col("^.+DateTime$").str.to_datetime("%Y-%m-%d %H:%M", strict=False),
+            pl.col("^.+DateTime$").str.to_datetime(
+                "%Y-%m-%d %H:%M",
+                strict=False,
+                time_zone="Asia/Tokyo",
+            ),
             pl.col("^period.+$").str.to_date("%Y-%m-%d", strict=False),
             pl.col("^.+Flag$").cast(pl.Int8).cast(pl.Boolean),
             pl.col("^.+Code$").cast(pl.String),

kabukit/jquants/client.py

@@ -5,6 +5,7 @@ import datetime
 import os
 from enum import StrEnum
 from typing import TYPE_CHECKING, final
+from zoneinfo import ZoneInfo
 
 import polars as pl
 from polars import DataFrame
@@ -32,10 +33,9 @@ class _CalendarCache:
     async def get_holidays(self, client: JQuantsClient) -> list[datetime.date]:
         async with self._lock:
             if self._holidays is None:
-                calendar_df = await client.get_calendar()
-                self._holidays = (
-                    calendar_df.filter(pl.col("IsHoliday")).get_column("Date").to_list()
-                )
+                df = await client.get_calendar()
+                holidays = df.filter(pl.col("IsHoliday"))["Date"]
+                self._holidays = holidays.to_list()
             return self._holidays
 
 
@@ -261,7 +261,7 @@ class JQuantsClient(Client):
         clean: bool = True,
     ) -> DataFrame:
         """直近利用可能な日付の株価を取得する。"""
-        today = datetime.date.today()  # noqa: DTZ011
+        today = datetime.datetime.now(ZoneInfo("Asia/Tokyo")).date()
 
         for days in range(num_days):
             date = today - datetime.timedelta(days)

kabukit/jquants/statements.py

@@ -81,10 +81,10 @@ def _cast_bool(df: DataFrame) -> DataFrame:
 def with_date(df: DataFrame, holidays: list[datetime.date]) -> DataFrame:
     """`Date`列を追加する。
 
-    開示日が休日のとき、あるいは、開示時刻が15時以降の場合、Dateを開示日の翌営業日に設定する。
+    開示日が休日のとき、あるいは、開示時刻が15時30分以降の場合、Dateを開示日の翌営業日に設定する。
     """
     is_after_hours = pl.col("DisclosedTime").is_null() | (
-        pl.col("DisclosedTime") > datetime.time(15, 0)
+        pl.col("DisclosedTime") >= datetime.time(15, 30)
     )
 
     return df.select(

kabukit/utils/concurrent.py

@@ -150,5 +150,5 @@ async def fetch(
         if callback:
             stream = (x if (r := callback(x)) is None else r async for x in stream)
 
-        dfs = [df async for df in stream if not df.is_empty()]
+        dfs = [df async for df in stream if not df.is_empty()]  # ty: ignore[not-iterable]
         return pl.concat(dfs) if dfs else pl.DataFrame()

kabukit/utils/date.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import datetime
+from zoneinfo import ZoneInfo
 
 
 def get_dates(days: int | None = None, years: int | None = None) -> list[datetime.date]:
@@ -11,7 +12,7 @@ def get_dates(days: int | None = None, years: int | None = None) -> list[datetim
         years (int | None): 過去years年の日付リストを取得する。
             daysが指定されている場合は無視される。
     """
-    end_date = datetime.date.today()  # noqa: DTZ011
+    end_date = datetime.datetime.now(ZoneInfo("Asia/Tokyo")).date()
 
     if days is not None:
         start_date = end_date - datetime.timedelta(days=days)

{kabukit-0.7.1.dist-info → kabukit-0.7.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kabukit
-Version: 0.7.1
+Version: 0.7.2
 Summary: A Python toolkit for Japanese financial market data, supporting J-Quants and EDINET APIs.
 Author: daizutabi
 Author-email: daizutabi <daizutabi@gmail.com>
@@ -28,14 +28,17 @@ License: MIT License
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: async-typer>=0.1.10
 Requires-Dist: httpx>=0.28.1
-Requires-Dist: platformdirs>=4.4.0
+Requires-Dist: platformdirs>=4.5.0
 Requires-Dist: polars>=1.34.0
 Requires-Dist: python-dotenv>=1.1.1
-Requires-Dist: rich>=14.1.0
+Requires-Dist: rich>=14.2.0
+Requires-Dist: tenacity>=9.1.2
 Requires-Dist: tqdm>=4.67.1
 Requires-Dist: typer>=0.19.2
+Requires-Dist: tzdata ; sys_platform == 'win32'
 Requires-Python: >=3.13
 Project-URL: Documentation, https://daizutabi.github.io/kabukit/
 Project-URL: Issues, https://github.com/daizutabi/kabukit/issues

{kabukit-0.7.1.dist-info → kabukit-0.7.2.dist-info}/RECORD

@@ -12,7 +12,7 @@ kabukit/cli/auth.py,sha256=E4M9ChDrtCb1DvYrAds33k6eKlXZUQCL9yCNPyTSnUk,2727
 kabukit/cli/cache.py,sha256=cHjpNPi1BnaPwXXIFEpbmjCA8Cvtov9yNcbRm12109M,1745
 kabukit/cli/get.py,sha256=z_bo1EdABkO0T4Mpdmb0qSIMVdWRWyopN7IX9VoTxxE,4815
 kabukit/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kabukit/core/base.py,sha256=kHLTbqTX-ZFPpNjkIpe_fxsBzY4vD3Tjyf7zpyK20CY,1691
+kabukit/core/base.py,sha256=WVFvv70n99GTqLkhkRnibLRgYP-eMh6lYY3R__q_YXY,1725
 kabukit/core/client.py,sha256=tVq1r3zpOfjmOtnRI1KPZHgTgBZYIpJzfw15i2kAM48,676
 kabukit/core/info.py,sha256=knL0rX18RNfra7iBhMK9H0GuaIapzYbxMeizkfuM1QM,88
 kabukit/core/list.py,sha256=JQHE6aoYreE2Oik_LJ9x0xE7nmkGsx688XYASaVqooE,88
@@ -20,25 +20,25 @@ kabukit/core/prices.py,sha256=dPNgCTFf-eE7-C-e_2vy9uqVZT7O55k2JKUNG1CPFX0,17728
 kabukit/core/reports.py,sha256=G7p8IcUOHDopZJMxMLnYhss9hIq4gCKEXFixINQI_7w,91
 kabukit/core/statements.py,sha256=_n4-8G284e6y_MwhvUq4xBK4bapQHk-Zmwu09C_r7wU,2790
 kabukit/edinet/__init__.py,sha256=PKa4D-jVpeoOkdVp9NwwpgAiGEBjqvmJLmpzF-9SlVk,101
-kabukit/edinet/client.py,sha256=BlM7pjXpKweOo-ses41JiKinqRn4qerMN5NF7hgYKgg,3275
+kabukit/edinet/client.py,sha256=HbxUvl6vKmfJ4-quIwCGspVOuHTMFbSolZXbhfG891U,3760
 kabukit/edinet/concurrent.py,sha256=2YPzIFuuOB8-gL3CnHIlP486QH5d21qjKNKGCFK7Hzk,4707
-kabukit/edinet/doc.py,sha256=6ZDgmm8DHmEMOA4NjNz-dHLMc7IzzYn-nVyMQGLWb8I,1220
+kabukit/edinet/doc.py,sha256=plxJ-o5-8NzllHY7Sn0M8KbyIenDAfh5XISLfpq27Ag,1410
 kabukit/jquants/__init__.py,sha256=xY0H6x4_51ZBqbrT2TdmGGFbTD-hosZiDzVIz4UXPv0,112
 kabukit/jquants/calendar.py,sha256=Vz4Degedgx8qENHWri2MTkIbkuIRfO28CXRq7bZaHGE,333
-kabukit/jquants/client.py,sha256=lDqh8ysvzC9dKU9ervO7GOjDQuxtLOo5npTu7t32V0g,14229
+kabukit/jquants/client.py,sha256=kp3xr8LH8NxO4lerzVseFPBvnUZw4grrpULExN89UoU,14241
 kabukit/jquants/concurrent.py,sha256=86xYD_zPLnR24xZhfSS0mAcUM-dvUvpozzyKASSHiwo,3345
 kabukit/jquants/info.py,sha256=MZbtg0L-YIkegWnCd3AvTs1AopV3z58ImgOnxgJgJbw,997
 kabukit/jquants/prices.py,sha256=oApQpdgzHwPw11XHpdg8ccZS7kybGtV8APZlpD2L3Yw,882
 kabukit/jquants/schema.py,sha256=aILl9cp9JDOaT2o3UlfavPGQC5s7n0ZkVBGKiTzdogs,9768
-kabukit/jquants/statements.py,sha256=6O03xx4LJtCkk7LgEN1zHOADZzkJJDC-x3se6ICo4mw,2745
+kabukit/jquants/statements.py,sha256=CHJKwM44-TAUPMrieA1VVwHT89KpcXxEfJcbh9y-dWI,2752
 kabukit/jquants/topix.py,sha256=oU84WAbmd4U77fuKvj_EUpZv3Pu_Sf0HEW4Y0dB8508,326
 kabukit/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kabukit/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kabukit/utils/concurrent.py,sha256=aTiCS6nqQtWBy29xykLHSU1LTr6LuGraxEnZco99EIM,4692
+kabukit/utils/concurrent.py,sha256=qR3gvhUBYOjwtVT_Hluy1g-XaaFL9ysHvZz3WUE8G1k,4720
 kabukit/utils/config.py,sha256=fqUdlhVjgiWEcsLFmPckp-dXUleVz8Ypdy_NnrMIBfY,708
-kabukit/utils/date.py,sha256=DEC6Ac5LS8eiW6JtrmcD3U1pX4qzXtx4ale0swpO4Ag,937
+kabukit/utils/date.py,sha256=eB5ONCkqnvOiAg1Hvg1pN4OkrxLR47urvOAD5BF5yL0,982
 kabukit/utils/params.py,sha256=qcaJbf6CWPUoZAZsYDTaZSnBUWeAersbWnR_iiYW9GM,1108
-kabukit-0.7.1.dist-info/WHEEL,sha256=I8-bO5cg2sb8TH6ZM6EgCP87Y1cV_f9UGgWnfAhVOZI,78
-kabukit-0.7.1.dist-info/entry_points.txt,sha256=vvX771TemoM-35vVizW3JJ70HvRXnd2tX4P1Btzyoxs,46
-kabukit-0.7.1.dist-info/METADATA,sha256=l322FK4twD7OLOHI0NIdC3wX7dlR3Jyps_EQpcm6fI0,3233
-kabukit-0.7.1.dist-info/RECORD,,
+kabukit-0.7.2.dist-info/WHEEL,sha256=X16MKk8bp2DRsAuyteHJ-9qOjzmnY0x1aj0P1ftqqWA,78
+kabukit-0.7.2.dist-info/entry_points.txt,sha256=vvX771TemoM-35vVizW3JJ70HvRXnd2tX4P1Btzyoxs,46
+kabukit-0.7.2.dist-info/METADATA,sha256=bOioHG440DSt2ywamoIM3vqD87lp1qMUPgkZUZy1u30,3363
+kabukit-0.7.2.dist-info/RECORD,,

{kabukit-0.7.1.dist-info → kabukit-0.7.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.9.0
+Generator: uv 0.9.2
 Root-Is-Purelib: true
 Tag: py3-none-any