kabukit 0.5.0__tar.gz → 0.5.1__tar.gz

{kabukit-0.5.0 → kabukit-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kabukit
-Version: 0.5.0
+Version: 0.5.1
 Summary: A Python toolkit for Japanese financial market data, supporting J-Quants and EDINET APIs.
 Author: daizutabi
 Author-email: daizutabi <daizutabi@gmail.com>
@@ -28,13 +28,13 @@ License: MIT License
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: async-typer>=0.1
+Requires-Dist: async-typer>=0.1.10
 Requires-Dist: holidays>=0.81
-Requires-Dist: httpx>=0.28
-Requires-Dist: platformdirs>=4
-Requires-Dist: polars>=1
-Requires-Dist: python-dotenv>=1
-Requires-Dist: typer>=0.19
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: platformdirs>=4.4.0
+Requires-Dist: polars>=1.34.0
+Requires-Dist: python-dotenv>=1.1.1
+Requires-Dist: typer>=0.19.2
 Requires-Python: >=3.13
 Project-URL: Documentation, https://daizutabi.github.io/kabukit/
 Project-URL: Issues, https://github.com/daizutabi/kabukit/issues

{kabukit-0.5.0 → kabukit-0.5.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "kabukit"
-version = "0.5.0"
+version = "0.5.1"
 description = "A Python toolkit for Japanese financial market data, supporting J-Quants and EDINET APIs."
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -16,13 +16,13 @@ classifiers = [
 ]
 requires-python = ">=3.13"
 dependencies = [
-  "async-typer>=0.1",
+  "async-typer>=0.1.10",
   "holidays>=0.81",
-  "httpx>=0.28",
-  "platformdirs>=4",
-  "polars>=1",
-  "python-dotenv>=1",
-  "typer>=0.19",
+  "httpx>=0.28.1",
+  "platformdirs>=4.4.0",
+  "polars>=1.34.0",
+  "python-dotenv>=1.1.1",
+  "typer>=0.19.2",
 ]
 
 [project.scripts]
@@ -35,10 +35,10 @@ Issues = "https://github.com/daizutabi/kabukit/issues"
 
 [dependency-groups]
 dev = [
-  "altair>=5",
-  "basedpyright>=1.31.4",
-  "marimo[lsp]>=0.16",
-  "numpy>=2.3.3",                 # polars 1.33 type hinting workaround,
+  "altair>=5.5.0",
+  "basedpyright>=1.31.6",
+  "marimo[lsp]>=0.16.5",
+  "numpy>=2.3.3",                   # polars 1.33 type hinting workaround,
   "pytest-asyncio>=1.2.0",
   "pytest-clarity>=1.0.1",
   "pytest-cov>=7.0.0",
@@ -47,9 +47,9 @@ dev = [
   "pytest-xdist>=3.8.0",
   "rich>=14.1.0",
   "tqdm>=4.67.1",
-  "vegafusion-python-embed>=1.6",
-  "vegafusion>=2",
-  "vl-convert-python>=1.8",
+  "vegafusion-python-embed>=1.6.9",
+  "vegafusion>=2.0.3",
+  "vl-convert-python>=1.8.0",
 ]
 docs = ["mkapi>=4.4", "mkdocs-marimo", "mkdocs-material"]
 

{kabukit-0.5.0 → kabukit-0.5.1}/src/kabukit/core/prices.py

@@ -76,29 +76,32 @@ class Prices(Base):
                 .over("Code", "ReportDate")
                 .alias("CumulativeRatio"),
             )
-            .select(
-                "Date",
-                "Code",
+            .with_columns(
                 (pl.col("IssuedShares", "TreasuryShares") * pl.col("CumulativeRatio"))
                 .round(0)
                 .cast(pl.Int64)
                 .name.prefix("Adjusted"),
             )
+            .select("Date", "Code", "AdjustedIssuedShares", "AdjustedTreasuryShares")
         )
 
         data = self.data.join(adjusted, on=["Date", "Code"], how="left")
 
         return self.__class__(data)
 
-    # def with_yields(self, statements: Statements) -> Self:
-    #     """各種利回り指標（収益利回り、純資産利回り、配当利回り）を計算し、列として追加する。
+    def with_forecast_profit(self, statements: Statements) -> Self:
+        """時系列の予想純利益を列として追加する。
 
-    #     Args:
-    #         statements (Statements): 財務データを提供する`Statements`オブジェクト。
+        Args:
+            statements (Statements): 財務データを提供する`Statements`オブジェクト。
 
-    #     Returns:
-    #         Self: 各種利回り指標が追加された、新しいPricesオブジェクト。
-    #     """
-    #     prices_with_adjusted_shares = self.with_adjusted_shares(statements)
-    #     data_with_yields = calculate_all_yields(prices_with_adjusted_shares, statements)
-    #     return self.__class__(data_with_yields)
+        Returns:
+            Self: `ForecastProfit` 列が追加された、新しいPricesオブジェクト。
+        """
+        data = self.data.join_asof(
+            statements.forecast_profit(),
+            on="Date",
+            by="Code",
+            check_sortedness=False,
+        )
+        return self.__class__(data)

kabukit-0.5.1/src/kabukit/core/statements.py

@@ -0,0 +1,41 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import polars as pl
+
+from .base import Base
+
+if TYPE_CHECKING:
+    from polars import DataFrame
+
+
+class Statements(Base):
+    def number_of_shares(self) -> DataFrame:
+        """発行済株式数を取得する。"""
+        return self.data.filter(
+            pl.col("IssuedShares").is_not_null(),
+        ).select(
+            "Date",
+            "Code",
+            "IssuedShares",
+            "TreasuryShares",
+            "AverageOutstandingShares",
+        )
+
+    def forecast_profit(self) -> DataFrame:
+        """予想純利益を抽出する。
+
+        Returns:
+            DataFrame: Date, Code, ForecastProfit を含むDataFrame
+        """
+        return (
+            self.data.with_columns(
+                pl.when(pl.col("TypeOfDocument").str.starts_with("FY"))
+                .then(pl.col("NextYearForecastProfit"))
+                .otherwise(pl.col("ForecastProfit"))
+                .alias("ForecastProfit"),
+            )
+            .filter(pl.col("ForecastProfit").is_not_null())
+            .select("Date", "Code", "ForecastProfit")
+        )

{kabukit-0.5.0 → kabukit-0.5.1}/src/kabukit/jquants/concurrent.py

@@ -41,7 +41,7 @@ async def fetch(
         DataFrame:
             すべての銘柄の財務情報を含む単一のDataFrame。
     """
-    return await concurrent.fetch(
+    data = await concurrent.fetch(
         JQuantsClient,
         resource,
         codes,
@@ -49,6 +49,7 @@ async def fetch(
         progress=progress,
         callback=callback,
     )
+    return data.sort("Code", "Date")
 
 
 async def fetch_all(

kabukit-0.5.0/src/kabukit/core/statements.py

@@ -1,24 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import polars as pl
-
-from .base import Base
-
-if TYPE_CHECKING:
-    from polars import DataFrame
-
-
-class Statements(Base):
-    def number_of_shares(self) -> DataFrame:
-        """発行済株式数を取得する。"""
-        return self.data.filter(
-            pl.col("IssuedShares").is_not_null(),
-        ).select(
-            "Date",
-            "Code",
-            "IssuedShares",
-            "TreasuryShares",
-            "AverageOutstandingShares",
-        )