kabukit 0.7.0__tar.gz → 0.7.1__tar.gz

{kabukit-0.7.0 → kabukit-0.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kabukit
-Version: 0.7.0
+Version: 0.7.1
 Summary: A Python toolkit for Japanese financial market data, supporting J-Quants and EDINET APIs.
 Author: daizutabi
 Author-email: daizutabi <daizutabi@gmail.com>
@@ -29,7 +29,6 @@ Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: async-typer>=0.1.10
-Requires-Dist: holidays>=0.82
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: platformdirs>=4.4.0
 Requires-Dist: polars>=1.34.0

{kabukit-0.7.0 → kabukit-0.7.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "kabukit"
-version = "0.7.0"
+version = "0.7.1"
 description = "A Python toolkit for Japanese financial market data, supporting J-Quants and EDINET APIs."
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -17,7 +17,6 @@ classifiers = [
 requires-python = ">=3.13"
 dependencies = [
   "async-typer>=0.1.10",
-  "holidays>=0.82",
   "httpx>=0.28.1",
   "platformdirs>=4.4.0",
   "polars>=1.34.0",
@@ -69,7 +68,7 @@ skip_covered = true
 [tool.ruff]
 line-length = 88
 target-version = "py313"
-include = ["src", "tests"]
+include = ["src/**/*.py", "tests/**/*.py"]
 
 [tool.ruff.lint]
 select = ["ALL"]
@@ -91,9 +90,9 @@ ignore = [
 ]
 
 [tool.ruff.lint.per-file-ignores]
-"tests/*" = ["ANN", "FBT", "S101", "S607", "PLC2401"]
+"tests/**/*.py" = ["ANN", "FBT", "S101", "S607", "PLC2401"]
 "schema.py" = ["E501"]
-"notebooks/*" = ["F704", "PLE1142"]
+"notebooks/**/*.py" = ["F704", "PLE1142"]
 
 [tool.ruff.format]
 exclude = ["schema.py"]

{kabukit-0.7.0 → kabukit-0.7.1}/src/kabukit/analysis/visualization/__init__.py

@@ -4,4 +4,4 @@ from .prices import plot_prices
 __all__ = [
     "plot_prices",
     "plot_topix_timeseries",
-]
+]

{kabukit-0.7.0 → kabukit-0.7.1}/src/kabukit/core/info.py

@@ -1,12 +1,7 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
 from .base import Base
 
-if TYPE_CHECKING:
-    from polars import DataFrame
-
 
 class Info(Base):
     pass

{kabukit-0.7.0 → kabukit-0.7.1}/src/kabukit/core/list.py

@@ -1,12 +1,7 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
 from .base import Base
 
-if TYPE_CHECKING:
-    from polars import DataFrame
-
 
 class List(Base):
     pass

{kabukit-0.7.0 → kabukit-0.7.1}/src/kabukit/core/reports.py

@@ -1,12 +1,7 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
 from .base import Base
 
-if TYPE_CHECKING:
-    from polars import DataFrame
-
 
 class Reports(Base):
     pass

kabukit-0.7.1/src/kabukit/jquants/calendar.py

@@ -0,0 +1,12 @@
+from __future__ import annotations
+
+import polars as pl
+from polars import DataFrame
+
+
+def clean(df: DataFrame) -> DataFrame:
+    return df.with_columns(
+        pl.col("Date").str.to_date("%Y-%m-%d"),
+        pl.col("HolidayDivision").cast(pl.Categorical),
+        pl.col("HolidayDivision").eq("1").not_().alias("IsHoliday"),
+    )

{kabukit-0.7.0 → kabukit-0.7.1}/src/kabukit/jquants/client.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
+import asyncio
 import datetime
 import os
 from enum import StrEnum
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, final
 
 import polars as pl
 from polars import DataFrame
@@ -12,7 +13,7 @@ from kabukit.core.client import Client
 from kabukit.utils.config import load_dotenv, set_key
 from kabukit.utils.params import get_params
 
-from . import info, prices, statements, topix
+from . import calendar, info, prices, statements, topix
 
 if TYPE_CHECKING:
     from collections.abc import AsyncIterator
@@ -21,6 +22,26 @@ if TYPE_CHECKING:
     from httpx import HTTPStatusError  # noqa: F401
     from httpx._types import QueryParamTypes
 
+
+@final
+class _CalendarCache:
+    def __init__(self) -> None:
+        self._holidays: list[datetime.date] | None = None
+        self._lock = asyncio.Lock()
+
+    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()
+                )
+            return self._holidays
+
+
+_calendar_cache_manager = _CalendarCache()
+
+
 API_VERSION = "v1"
 BASE_URL = f"https://api.jquants.com/{API_VERSION}"
 
@@ -128,32 +149,6 @@ class JQuantsClient(Client):
         self.set_id_token(id_token)
         return self
 
-    async def get_info(
-        self,
-        code: str | None = None,
-        date: str | datetime.date | None = None,
-        *,
-        clean: bool = True,
-    ) -> DataFrame:
-        """銘柄情報を取得する。
-
-        Args:
-            code (str, optional): 情報を取得する銘柄のコード。
-            date (str | datetime.date, optional): 情報を取得する日付。
-            clean (bool, optional): 取得したデータをクリーンアップするかどうか。
-
-        Returns:
-            銘柄情報を含むDataFrame。
-
-        Raises:
-            HTTPStatusError: APIリクエストが失敗した場合。
-        """
-        params = get_params(code=code, date=date)
-        url = "/listed/info"
-        data = await self.get(url, params)
-        df = DataFrame(data["info"])
-        return info.clean(df) if clean else df
-
     async def iter_pages(
         self,
         url: str,
@@ -183,6 +178,32 @@ class JQuantsClient(Client):
             else:
                 break
 
+    async def get_info(
+        self,
+        code: str | None = None,
+        date: str | datetime.date | None = None,
+        *,
+        clean: bool = True,
+    ) -> DataFrame:
+        """銘柄情報を取得する。
+
+        Args:
+            code (str, optional): 情報を取得する銘柄のコード。
+            date (str | datetime.date, optional): 情報を取得する日付。
+            clean (bool, optional): 取得したデータをクリーンアップするかどうか。
+
+        Returns:
+            銘柄情報を含むDataFrame。
+
+        Raises:
+            HTTPStatusError: APIリクエストが失敗した場合。
+        """
+        params = get_params(code=code, date=date)
+        url = "/listed/info"
+        data = await self.get(url, params)
+        df = DataFrame(data["info"])
+        return info.clean(df) if clean else df
+
     async def get_prices(
         self,
         code: str | None = None,
@@ -291,7 +312,11 @@ class JQuantsClient(Client):
 
         df = statements.clean(df)
 
-        return statements.with_date(df) if with_date else df
+        if not with_date:
+            return df
+
+        holidays = await _calendar_cache_manager.get_holidays(self)
+        return statements.with_date(df, holidays=holidays)
 
     async def get_announcement(self) -> DataFrame:
         """翌日発表予定の決算情報を取得する。
@@ -310,7 +335,7 @@ class JQuantsClient(Client):
         if df.is_empty():
             return df
 
-        return df.with_columns(pl.col("Date").str.to_date("%Y-%m-%d", strict=False))
+        return df.with_columns(pl.col("Date").str.to_date("%Y-%m-%d"))
 
     async def get_trades_spec(
         self,
@@ -341,7 +366,7 @@ class JQuantsClient(Client):
         if df.is_empty():
             return df
 
-        return df.with_columns(pl.col("^.*Date$").str.to_date("%Y-%m-%d", strict=False))
+        return df.with_columns(pl.col("^.*Date$").str.to_date("%Y-%m-%d"))
 
     async def get_topix(
         self,
@@ -371,3 +396,38 @@ class JQuantsClient(Client):
             return df
 
         return topix.clean(df)
+
+    async def get_calendar(
+        self,
+        holidaydivision: str | None = None,
+        from_: str | datetime.date | None = None,
+        to: str | datetime.date | None = None,
+    ) -> DataFrame:
+        """東証およびOSEにおける営業日、休業日、ならびにOSEにおける祝日取引の有無の情報を取得する。
+
+        Args:
+            holidaydivision: 祝日区分。
+                - 非営業日: "0"
+                - 営業日: "1"
+                - 東証半日立会日: "2"
+                - 非営業日(祝日取引あり): "3"
+            from_: 取得期間の開始日。
+            to: 取得期間の終了日。
+
+        Returns:
+            営業日・非営業日データを含むPolars DataFrame。
+
+        Raises:
+            HTTPStatusError: APIリクエストが失敗した場合。
+        """
+        params = get_params(holidaydivision=holidaydivision, from_=from_, to=to)
+
+        url = "/markets/trading_calendar"
+        name = "trading_calendar"
+
+        dfs = [df async for df in self.iter_pages(url, params, name)]
+        df = pl.concat(dfs)
+        if df.is_empty():
+            return df
+
+        return calendar.clean(df)

{kabukit-0.7.0 → kabukit-0.7.1}/src/kabukit/jquants/statements.py

@@ -1,10 +1,8 @@
 from __future__ import annotations
 
 import datetime
-from functools import cache
 from typing import TYPE_CHECKING
 
-import holidays
 import polars as pl
 
 if TYPE_CHECKING:
@@ -80,17 +78,7 @@ def _cast_bool(df: DataFrame) -> DataFrame:
     )
 
 
-@cache
-def get_holidays(year: int | None = None, n: int = 10) -> list[datetime.date]:
-    """指定した過去年数の日本の祝日を取得する。"""
-    if year is None:
-        year = datetime.datetime.now().year  # noqa: DTZ005
-
-    dates = holidays.country_holidays("JP", years=range(year - n, year + 1))
-    return sorted(dates.keys())
-
-
-def with_date(df: DataFrame, year: int | None = None) -> DataFrame:
+def with_date(df: DataFrame, holidays: list[datetime.date]) -> DataFrame:
     """`Date`列を追加する。
 
     開示日が休日のとき、あるいは、開示時刻が15時以降の場合、Dateを開示日の翌営業日に設定する。
@@ -99,8 +87,6 @@ def with_date(df: DataFrame, year: int | None = None) -> DataFrame:
         pl.col("DisclosedTime") > datetime.time(15, 0)
     )
 
-    holidays = get_holidays(year=year)
-
     return df.select(
         pl.when(is_after_hours)
         .then(pl.col("DisclosedDate") + datetime.timedelta(days=1))