kabukit 0.2.1__tar.gz → 0.3.1__tar.gz

{kabukit-0.2.1 → kabukit-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kabukit
-Version: 0.2.1
+Version: 0.3.1
 Summary: A Python toolkit for Japanese financial market data, supporting J-Quants and EDINET APIs.
 Author: daizutabi
 Author-email: daizutabi <daizutabi@gmail.com>
@@ -27,21 +27,16 @@ License: MIT License
          SOFTWARE.
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python
-Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: altair>=5
 Requires-Dist: async-typer>=0.1
 Requires-Dist: holidays>=0.81
 Requires-Dist: httpx>=0.28.1
-Requires-Dist: marimo[lsp]>=0.16
 Requires-Dist: platformdirs>=4
 Requires-Dist: polars>=1
 Requires-Dist: python-dotenv>=1
 Requires-Dist: typer>=0.19
-Requires-Dist: vegafusion-python-embed>=1.6
-Requires-Dist: vegafusion>=2
-Requires-Dist: vl-convert-python>=1.8
-Requires-Python: >=3.12
+Requires-Python: >=3.13
 Project-URL: Documentation, https://daizutabi.github.io/kabukit/
 Project-URL: Issues, https://github.com/daizutabi/kabukit/issues
 Project-URL: Source, https://github.com/daizutabi/kabukit
@@ -53,6 +48,9 @@ A Python toolkit for Japanese financial market data, supporting J-Quants and EDI
 
 [![PyPI Version][pypi-v-image]][pypi-v-link]
 [![Python Version][python-v-image]][python-v-link]
+[![Build Status][GHAction-image]][GHAction-link]
+[![Coverage Status][codecov-image]][codecov-link]
+[![Documentation Status][docs-image]][docs-link]
 
 ## Installation
 
@@ -61,7 +59,14 @@ pip install kabukit
 ```
 
 <!-- Badges -->
+
 [pypi-v-image]: https://img.shields.io/pypi/v/kabukit.svg
 [pypi-v-link]: https://pypi.org/project/kabukit/
 [python-v-image]: https://img.shields.io/pypi/pyversions/kabukit.svg
 [python-v-link]: https://pypi.org/project/kabukit
+[GHAction-image]: https://github.com/daizutabi/kabukit/actions/workflows/ci.yaml/badge.svg?branch=main&event=push
+[GHAction-link]: https://github.com/daizutabi/kabukit/actions?query=event%3Apush+branch%3Amain
+[codecov-image]: https://codecov.io/github/daizutabi/kabukit/graph/badge.svg?token=Yu6lAdVVnd
+[codecov-link]: https://codecov.io/github/daizutabi/kabukit?branch=main
+[docs-image]: https://img.shields.io/badge/docs-latest-blue.svg
+[docs-link]: https://daizutabi.github.io/kabukit/

kabukit-0.3.1/README.md

@@ -0,0 +1,28 @@
+# kabukit
+
+A Python toolkit for Japanese financial market data, supporting J-Quants and EDINET APIs.
+
+[![PyPI Version][pypi-v-image]][pypi-v-link]
+[![Python Version][python-v-image]][python-v-link]
+[![Build Status][GHAction-image]][GHAction-link]
+[![Coverage Status][codecov-image]][codecov-link]
+[![Documentation Status][docs-image]][docs-link]
+
+## Installation
+
+```bash
+pip install kabukit
+```
+
+<!-- Badges -->
+
+[pypi-v-image]: https://img.shields.io/pypi/v/kabukit.svg
+[pypi-v-link]: https://pypi.org/project/kabukit/
+[python-v-image]: https://img.shields.io/pypi/pyversions/kabukit.svg
+[python-v-link]: https://pypi.org/project/kabukit
+[GHAction-image]: https://github.com/daizutabi/kabukit/actions/workflows/ci.yaml/badge.svg?branch=main&event=push
+[GHAction-link]: https://github.com/daizutabi/kabukit/actions?query=event%3Apush+branch%3Amain
+[codecov-image]: https://codecov.io/github/daizutabi/kabukit/graph/badge.svg?token=Yu6lAdVVnd
+[codecov-link]: https://codecov.io/github/daizutabi/kabukit?branch=main
+[docs-image]: https://img.shields.io/badge/docs-latest-blue.svg
+[docs-link]: https://daizutabi.github.io/kabukit/

{kabukit-0.2.1 → kabukit-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "kabukit"
-version = "0.2.1"
+version = "0.3.1"
 description = "A Python toolkit for Japanese financial market data, supporting J-Quants and EDINET APIs."
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -12,23 +12,18 @@ authors = [{ name = "daizutabi", email = "daizutabi@gmail.com" }]
 classifiers = [
   "Development Status :: 4 - Beta",
   "Programming Language :: Python",
-  "Programming Language :: Python :: 3.12",
   "Programming Language :: Python :: 3.13",
 ]
-requires-python = ">=3.12"
+requires-python = ">=3.13"
 dependencies = [
   "altair>=5",
   "async-typer>=0.1",
   "holidays>=0.81",
   "httpx>=0.28.1",
-  "marimo[lsp]>=0.16",
   "platformdirs>=4",
   "polars>=1",
   "python-dotenv>=1",
   "typer>=0.19",
-  "vegafusion-python-embed>=1.6",
-  "vegafusion>=2",
-  "vl-convert-python>=1.8",
 ]
 
 [project.scripts]
@@ -42,20 +37,35 @@ Issues = "https://github.com/daizutabi/kabukit/issues"
 [dependency-groups]
 dev = [
   "basedpyright>=1.31.4",
-  "numpy>=2.3.3",           # polars 1.33 type hinting workaround,
+  "marimo[lsp]>=0.16",
+  "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",
+  "pytest-mock>=3.15.1",
   "pytest-randomly>=4.0.1",
   "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",
 ]
 docs = ["mkapi>=4.4", "mkdocs-marimo", "mkdocs-material"]
 
 [tool.pytest.ini_options]
-addopts = ["--cov=kabukit", "--cov-report=lcov:lcov.info", "--doctest-modules"]
-testpaths = ["src", "tests"]
+addopts = [
+  "--cov=kabukit",
+  "--cov-report=lcov:lcov.info",
+  "--doctest-modules",
+  "-m",
+  "not integration and not validation",
+]
+testpaths = ["tests/unit", "tests/integration", "tests/validation"]
+markers = [
+  "integration: marks tests as integration tests",
+  "validation: marks tests as data validation tests",
+]
 
 [tool.coverage.report]
 exclude_lines = ["no cov", "raise NotImplementedError", "if TYPE_CHECKING:"]
@@ -63,7 +73,7 @@ skip_covered = true
 
 [tool.ruff]
 line-length = 88
-target-version = "py312"
+target-version = "py313"
 include = ["src", "tests"]
 
 [tool.ruff.lint]
@@ -86,7 +96,7 @@ ignore = [
 ]
 
 [tool.ruff.lint.per-file-ignores]
-"tests/*" = ["ANN", "FBT", "S101", "S607"]
+"tests/*" = ["ANN", "FBT", "S101", "S607", "PLC2401"]
 "schema.py" = ["E501"]
 "notebooks/*" = ["F704", "PLE1142"]
 

kabukit-0.3.1/src/kabukit/__init__.py

@@ -0,0 +1,17 @@
+from .core.info import Info
+from .core.list import List
+from .core.prices import Prices
+from .core.reports import Reports
+from .core.statements import Statements
+from .edinet.client import EdinetClient
+from .jquants.client import JQuantsClient
+
+__all__ = [
+    "EdinetClient",
+    "Info",
+    "JQuantsClient",
+    "List",
+    "Prices",
+    "Reports",
+    "Statements",
+]

{kabukit-0.2.1 → kabukit-0.3.1}/src/kabukit/cli/app.py

@@ -5,13 +5,14 @@ from __future__ import annotations
 import typer
 from async_typer import AsyncTyper  # pyright: ignore[reportMissingTypeStubs]
 
-from . import auth
+from . import auth, get
 
 app = AsyncTyper(
     add_completion=False,
     help="J-Quants/EDINETデータツール",
 )
 app.add_typer(auth.app, name="auth")
+app.add_typer(get.app, name="get")
 
 
 @app.command()

kabukit-0.3.1/src/kabukit/cli/get.py

@@ -0,0 +1,154 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Annotated, Any
+
+import typer
+from async_typer import AsyncTyper  # pyright: ignore[reportMissingTypeStubs]
+from typer import Argument
+
+if TYPE_CHECKING:
+    from kabukit.core.base import Base
+
+app = AsyncTyper(
+    add_completion=False,
+    help="J-Quantsからデータを取得します。",
+)
+
+Code = Annotated[
+    str | None,
+    Argument(help="銘柄コード。指定しない場合は全銘柄の情報を取得します。"),
+]
+
+
+@app.async_command()  # pyright: ignore[reportUnknownMemberType]
+async def info(code: Code = None) -> None:
+    """上場銘柄一覧を取得します。"""
+    from kabukit.core.info import Info
+    from kabukit.jquants.client import JQuantsClient
+
+    async with JQuantsClient() as client:
+        df = await client.get_info(code)
+
+    typer.echo(df)
+
+    if code is None:
+        path = Info(df).write()
+        typer.echo(f"全銘柄の情報を '{path}' に保存しました。")
+
+
+async def _fetch(
+    code: str | None,
+    target: str,
+    cls: type[Base],
+    fetch_func_name: str,
+    message: str,
+    **kwargs: Any,
+) -> None:
+    """財務情報・株価情報を取得するための共通処理"""
+    from kabukit.jquants.client import JQuantsClient
+
+    if code is not None:
+        async with JQuantsClient() as client:
+            df = await getattr(client, fetch_func_name)(code)
+        typer.echo(df)
+        return
+
+    import tqdm.asyncio
+
+    from kabukit.jquants.concurrent import fetch_all
+
+    df = await fetch_all(target, progress=tqdm.asyncio.tqdm, **kwargs)
+    typer.echo(df)
+    path = cls(df).write()
+    typer.echo(f"全銘柄の{message}を '{path}' に保存しました。")
+
+
+@app.async_command()  # pyright: ignore[reportUnknownMemberType]
+async def statements(code: Code = None) -> None:
+    """財務情報を取得します。"""
+    from kabukit.core.statements import Statements
+
+    await _fetch(
+        code=code,
+        target="statements",
+        cls=Statements,
+        fetch_func_name="get_statements",
+        message="財務情報",
+    )
+
+
+@app.async_command()  # pyright: ignore[reportUnknownMemberType]
+async def prices(code: Code = None) -> None:
+    """株価を取得します。"""
+    from kabukit.core.prices import Prices
+
+    await _fetch(
+        code=code,
+        target="prices",
+        cls=Prices,
+        fetch_func_name="get_prices",
+        message="株価情報",
+        max_concurrency=8,
+    )
+
+
+@app.async_command(name="list")  # pyright: ignore[reportUnknownMemberType]
+async def list_() -> None:
+    """報告書一覧を取得します。"""
+    import tqdm.asyncio
+
+    from kabukit.core.list import List
+    from kabukit.edinet.concurrent import fetch_list
+
+    df = await fetch_list(years=10, progress=tqdm.asyncio.tqdm)
+    typer.echo(df)
+    path = List(df).write()
+    typer.echo(f"報告書一覧を '{path}' に保存しました。")
+
+
+@app.async_command()  # pyright: ignore[reportUnknownMemberType]
+async def reports() -> None:
+    """報告書を取得します。"""
+    import polars as pl
+    import tqdm.asyncio
+
+    from kabukit.core.list import List
+    from kabukit.core.reports import Reports
+    from kabukit.edinet.concurrent import fetch_csv
+
+    try:
+        df = List.read().data
+    except FileNotFoundError:
+        await list_()
+        df = List.read().data
+
+    lst = df.filter(pl.col("csvFlag"), pl.col("secCode").is_not_null())
+    doc_ids = lst["docID"].unique()
+
+    df = await fetch_csv(doc_ids, limit=1000, progress=tqdm.asyncio.tqdm)
+    typer.echo(df)
+    path = Reports(df).write()
+    typer.echo(f"報告書を '{path}' に保存しました。")
+
+
+@app.async_command(name="all")  # pyright: ignore[reportUnknownMemberType]
+async def all_(code: Code = None) -> None:
+    """上場銘柄一覧、財務情報、株価、報告書を連続して取得します。"""
+    typer.echo("上場銘柄一覧を取得します。")
+    await info(code)
+
+    typer.echo("---")
+    typer.echo("財務情報を取得します。")
+    await statements(code)
+
+    typer.echo("---")
+    typer.echo("株価を取得します。")
+    await prices(code)
+
+    if code is None:
+        typer.echo("---")
+        typer.echo("報告書一覧を取得します。")
+        await list_()
+        typer.echo("---")
+        typer.echo("報告書を取得します。")
+        await reports()

{kabukit-0.2.1 → kabukit-0.3.1}/src/kabukit/core/base.py

@@ -36,10 +36,15 @@ class Base:
     def read(cls, path: str | None = None) -> Self:
         data_dir = cls.data_dir()
 
-        if path is None:
-            filename = sorted(data_dir.glob("*.parquet"))[-1]
-        else:
+        if path:
             filename = data_dir / path
+        else:
+            filenames = sorted(data_dir.glob("*.parquet"))
+            if not filenames:
+                msg = f"No data found in {data_dir}"
+                raise FileNotFoundError(msg)
+
+            filename = filenames[-1]
 
         data = pl.read_parquet(filename)
         return cls(data)

kabukit-0.3.1/src/kabukit/core/list.py

@@ -0,0 +1,12 @@
+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.3.1/src/kabukit/core/reports.py

@@ -0,0 +1,12 @@
+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.3.1/src/kabukit/edinet/__init__.py

@@ -0,0 +1,3 @@
+from .concurrent import fetch, fetch_csv, fetch_list
+
+__all__ = ["fetch", "fetch_csv", "fetch_list"]

{kabukit-0.2.1 → kabukit-0.3.1}/src/kabukit/edinet/client.py

@@ -6,14 +6,13 @@ import zipfile
 from enum import StrEnum
 from typing import TYPE_CHECKING
 
-import polars as pl
 from polars import DataFrame
 
 from kabukit.core.client import Client
 from kabukit.utils.config import load_dotenv
 from kabukit.utils.params import get_params
 
-from .doc import clean_csv, clean_list
+from .doc import clean_csv, clean_list, read_csv
 
 if TYPE_CHECKING:
     import datetime
@@ -103,11 +102,8 @@ class EdinetClient(Client):
             for info in zf.infolist():
                 if info.filename.endswith(".csv"):
                     with zf.open(info) as f:
-                        return pl.read_csv(
-                            f.read(),
-                            separator="\t",
-                            encoding="utf-16-le",
-                        ).pipe(clean_csv, doc_id)
+                        df = read_csv(f.read())
+                        return clean_csv(df, doc_id)
 
         msg = "CSV is not available."
         raise ValueError(msg)

{kabukit-0.2.1 → kabukit-0.3.1}/src/kabukit/edinet/concurrent.py

@@ -1,13 +1,14 @@
 from __future__ import annotations
 
-import datetime
 from typing import TYPE_CHECKING
 
 from kabukit.utils import concurrent
+from kabukit.utils.date import get_dates
 
 from .client import EdinetClient
 
 if TYPE_CHECKING:
+    import datetime
     from collections.abc import Iterable
 
     from polars import DataFrame
@@ -15,30 +16,6 @@ if TYPE_CHECKING:
     from kabukit.utils.concurrent import Callback, Progress
 
 
-def get_dates(days: int | None = None, years: int | None = None) -> list[datetime.date]:
-    """過去days日またはyears年の日付リストを返す。
-
-    Args:
-        days (int | None): 過去days日の日付リストを取得する。
-        years (int | None): 過去years年の日付リストを取得する。
-            daysが指定されている場合は無視される。
-    """
-    end_date = datetime.date.today()  # noqa: DTZ011
-
-    if days is not None:
-        start_date = end_date - datetime.timedelta(days=days)
-    elif years is not None:
-        start_date = end_date.replace(year=end_date.year - years)
-    else:
-        msg = "daysまたはyearsのいずれかを指定してください。"
-        raise ValueError(msg)
-
-    return [
-        start_date + datetime.timedelta(days=i)
-        for i in range(1, (end_date - start_date).days + 1)
-    ]
-
-
 async def fetch(
     resource: str,
     args: Iterable[str | datetime.date],
@@ -106,13 +83,14 @@ async def fetch_list(
     if limit is not None:
         dates = dates[:limit]
 
-    return await fetch(
+    df = await fetch(
         "list",
         dates,
         max_concurrency=max_concurrency,
         progress=progress,
         callback=callback,
     )
+    return df.sort("Date")
 
 
 async def fetch_csv(
@@ -144,10 +122,11 @@ async def fetch_csv(
     if limit is not None:
         doc_ids = doc_ids[:limit]
 
-    return await fetch(
+    df = await fetch(
         "csv",
         doc_ids,
         max_concurrency=max_concurrency,
         progress=progress,
         callback=callback,
     )
+    return df.sort("docID")

kabukit-0.3.1/src/kabukit/edinet/doc.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+import datetime
+from typing import TYPE_CHECKING
+
+import polars as pl
+
+if TYPE_CHECKING:
+    from polars import DataFrame
+
+
+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
+
+    null_columns = [c for c in df.columns if df[c].dtype == pl.Null]
+
+    return (
+        df.with_columns(
+            pl.col(null_columns).cast(pl.String),
+        )
+        .with_columns(
+            pl.lit(date).alias("Date"),
+            pl.col("^.+DateTime$").str.to_datetime("%Y-%m-%d %H:%M", strict=False),
+            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),
+        )
+        .select("Date", pl.exclude("Date"))
+    )
+
+
+def read_csv(data: bytes) -> DataFrame:
+    return pl.read_csv(
+        data,
+        separator="\t",
+        encoding="utf-16-le",
+        infer_schema_length=None,
+    )
+
+
+def clean_csv(df: DataFrame, doc_id: str) -> DataFrame:
+    return df.select(
+        pl.lit(doc_id).alias("docID"),
+        pl.all(),
+    )

kabukit-0.3.1/src/kabukit/jquants/__init__.py

@@ -0,0 +1,4 @@
+from .concurrent import fetch, fetch_all
+from .schema import rename
+
+__all__ = ["fetch", "fetch_all", "rename"]

{kabukit-0.2.1 → kabukit-0.3.1}/src/kabukit/jquants/client.py

@@ -132,12 +132,15 @@ class JQuantsClient(Client):
         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。
@@ -148,7 +151,8 @@ class JQuantsClient(Client):
         params = get_params(code=code, date=date)
         url = "/listed/info"
         data = await self.get(url, params)
-        return DataFrame(data["info"]).pipe(info.clean)
+        df = DataFrame(data["info"])
+        return info.clean(df) if clean else df
 
     async def iter_pages(
         self,
@@ -185,6 +189,8 @@ class JQuantsClient(Client):
         date: str | datetime.date | None = None,
         from_: str | datetime.date | None = None,
         to: str | datetime.date | None = None,
+        *,
+        clean: bool = True,
     ) -> DataFrame:
         """日々の株価四本値を取得する。
 
@@ -198,6 +204,7 @@ class JQuantsClient(Client):
                 `date`とは併用不可。
             to (str | datetime.date, optional): 取得期間の終了日。
                 `date`とは併用不可。
+            clean (bool, optional): 取得したデータをクリーンアップするかどうか。
 
         Returns:
             日々の株価四本値を含むDataFrame。
@@ -207,7 +214,7 @@ class JQuantsClient(Client):
             HTTPStatusError: APIリクエストが失敗した場合。
         """
         if not date and not code:
-            return await self.get_latest_available_prices()
+            return await self.get_latest_available_prices(clean=clean)
 
         if date and (from_ or to):
             msg = "dateとfrom/toの両方を指定することはできません。"
@@ -224,15 +231,20 @@ class JQuantsClient(Client):
         if df.is_empty():
             return df
 
-        return prices.clean(df)
+        return prices.clean(df) if clean else df
 
-    async def get_latest_available_prices(self, num_days: int = 30) -> DataFrame:
+    async def get_latest_available_prices(
+        self,
+        num_days: int = 30,
+        *,
+        clean: bool = True,
+    ) -> DataFrame:
         """直近利用可能な日付の株価を取得する。"""
         today = datetime.date.today()  # noqa: DTZ011
 
         for days in range(num_days):
             date = today - datetime.timedelta(days)
-            df = await self.get_prices(date=date)
+            df = await self.get_prices(date=date, clean=clean)
 
             if not df.is_empty():
                 return df
@@ -243,12 +255,15 @@ class JQuantsClient(Client):
         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。
@@ -271,7 +286,7 @@ class JQuantsClient(Client):
         if df.is_empty():
             return df
 
-        return statements.clean(df)
+        return statements.clean(df) if clean else df
 
     async def get_announcement(self) -> DataFrame:
         """翌日発表予定の決算情報を取得する。

{kabukit-0.2.1 → kabukit-0.3.1}/src/kabukit/jquants/statements.py

@@ -2,25 +2,14 @@ from __future__ import annotations
 
 import datetime
 from functools import cache
-from typing import TYPE_CHECKING, Any, Protocol
+from typing import TYPE_CHECKING
 
 import holidays
 import polars as pl
 
 if TYPE_CHECKING:
-    from collections.abc import AsyncIterable, AsyncIterator
-
     from polars import DataFrame
 
-    class Progress(Protocol):
-        def __call__[T](
-            self,
-            async_iterable: AsyncIterable[T],
-            total: int | None = None,
-            *args: Any,
-            **kwargs: Any,
-        ) -> AsyncIterator[T]: ...
-
 
 def clean(df: DataFrame) -> DataFrame:
     return (

{kabukit-0.2.1 → kabukit-0.3.1}/src/kabukit/utils/concurrent.py

@@ -23,13 +23,14 @@ if TYPE_CHECKING:
     from kabukit.core.client import Client
 
     class _Progress(Protocol):
-        def __call__[T](
+        def __call__(
             self,
-            aiterable: AsyncIterable[T],
+            aiterable: AsyncIterable[Any],
+            /,
             total: int | None = None,
             *args: Any,
             **kwargs: Any,
-        ) -> AsyncIterator[T]: ...
+        ) -> AsyncIterator[Any]: ...
 
 
 MAX_CONCURRENCY = 12

{kabukit-0.2.1 → kabukit-0.3.1}/src/kabukit/utils/config.py

@@ -1,13 +1,11 @@
 from __future__ import annotations
 
-from functools import cache
 from pathlib import Path
 
 import dotenv
 from platformdirs import user_config_dir
 
 
-@cache
 def get_dotenv_path() -> Path:
     """Return the path to the .env file in the user config directory."""
     config_dir = Path(user_config_dir("kabukit"))
@@ -15,12 +13,11 @@ def get_dotenv_path() -> Path:
     return config_dir / ".env"
 
 
-@cache
-def load_dotenv() -> bool:
+def set_key(key: str, value: str) -> tuple[bool | None, str, str]:
     dotenv_path = get_dotenv_path()
-    return dotenv.load_dotenv(dotenv_path)
+    return dotenv.set_key(dotenv_path, key, value)
 
 
-def set_key(key: str, value: str) -> tuple[bool | None, str, str]:
+def load_dotenv() -> bool:
     dotenv_path = get_dotenv_path()
-    return dotenv.set_key(dotenv_path, key, value)
+    return dotenv.load_dotenv(dotenv_path)

kabukit-0.3.1/src/kabukit/utils/date.py

@@ -0,0 +1,27 @@
+from __future__ import annotations
+
+import datetime
+
+
+def get_dates(days: int | None = None, years: int | None = None) -> list[datetime.date]:
+    """過去days日またはyears年の日付リストを返す。
+
+    Args:
+        days (int | None): 過去days日の日付リストを取得する。
+        years (int | None): 過去years年の日付リストを取得する。
+            daysが指定されている場合は無視される。
+    """
+    end_date = datetime.date.today()  # noqa: DTZ011
+
+    if days is not None:
+        start_date = end_date - datetime.timedelta(days=days)
+    elif years is not None:
+        start_date = end_date.replace(year=end_date.year - years)
+    else:
+        msg = "daysまたはyearsのいずれかを指定してください。"
+        raise ValueError(msg)
+
+    return [
+        start_date + datetime.timedelta(days=i)
+        for i in range(1, (end_date - start_date).days + 1)
+    ]

{kabukit-0.2.1 → kabukit-0.3.1}/src/kabukit/utils/params.py

@@ -18,16 +18,16 @@ def iter_items(kwargs: dict[str, Any]) -> Iterator[tuple[str, Any]]:
             yield key, value
 
 
-def get_params(**kwargs: Any) -> dict[str, str]:
-    params: dict[str, str] = {}
+def get_params(**kwargs: Any) -> dict[str, Any]:
+    params: dict[str, Any] = {}
 
     for key, value in iter_items(kwargs):
         if isinstance(value, datetime.date):
             params[key] = date_to_str(value)
-        elif not isinstance(value, str):
-            params[key] = str(value)
-        else:
+        elif isinstance(value, str | int | float | bool):
             params[key] = value
+        else:
+            params[key] = str(value)
 
     return params
 

kabukit-0.2.1/README.md

@@ -1,18 +0,0 @@
-# kabukit
-
-A Python toolkit for Japanese financial market data, supporting J-Quants and EDINET APIs.
-
-[![PyPI Version][pypi-v-image]][pypi-v-link]
-[![Python Version][python-v-image]][python-v-link]
-
-## Installation
-
-```bash
-pip install kabukit
-```
-
-<!-- Badges -->
-[pypi-v-image]: https://img.shields.io/pypi/v/kabukit.svg
-[pypi-v-link]: https://pypi.org/project/kabukit/
-[python-v-image]: https://img.shields.io/pypi/pyversions/kabukit.svg
-[python-v-link]: https://pypi.org/project/kabukit

kabukit-0.2.1/src/kabukit/__init__.py

@@ -1,7 +0,0 @@
-from .core.info import Info
-from .core.prices import Prices
-from .core.statements import Statements
-from .edinet.client import EdinetClient
-from .jquants.client import JQuantsClient
-
-__all__ = ["EdinetClient", "Info", "JQuantsClient", "Prices", "Statements"]

kabukit-0.2.1/src/kabukit/edinet/__init__.py

@@ -1,3 +0,0 @@
-from .concurrent import fetch, fetch_list
-
-__all__ = ["fetch", "fetch_list"]

kabukit-0.2.1/src/kabukit/edinet/doc.py

@@ -1,32 +0,0 @@
-from __future__ import annotations
-
-import datetime
-from typing import TYPE_CHECKING
-
-import polars as pl
-
-if TYPE_CHECKING:
-    from polars import DataFrame
-
-
-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
-
-    return df.with_columns(
-        pl.lit(date).alias("Date"),
-        pl.col("submitDateTime").str.to_datetime("%Y-%m-%d %H:%M", strict=False),
-        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),
-        pl.col("opeDateTime")
-        .cast(pl.String)
-        .str.to_datetime("%Y-%m-%d %H:%M", strict=False),
-    ).select("Date", pl.exclude("Date"))
-
-
-def clean_csv(df: DataFrame, doc_id: str) -> DataFrame:
-    return df.select(
-        pl.lit(doc_id).alias("docID"),
-        pl.all(),
-    )

kabukit-0.2.1/src/kabukit/jquants/__init__.py

@@ -1,3 +0,0 @@
-from .concurrent import fetch, fetch_all
-
-__all__ = ["fetch", "fetch_all"]