bullishpy 0.11.0__tar.gz → 0.12.0__tar.gz

{bullishpy-0.11.0 → bullishpy-0.12.0}/PKG-INFO

@@ -1,15 +1,16 @@
 Metadata-Version: 2.3
 Name: bullishpy
-Version: 0.11.0
+Version: 0.12.0
 Summary: 
 Author: aan
 Author-email: andoludovic.andriamamonjy@gmail.com
 Requires-Python: >=3.12,<3.13
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: bearishpy (>=0.20.0,<0.21.0)
+Requires-Dist: bearishpy (>=0.22.0,<0.23.0)
 Requires-Dist: click (>=7.0,<=8.1)
 Requires-Dist: huey (>=2.5.3,<3.0.0)
+Requires-Dist: joblib (>=1.5.1,<2.0.0)
 Requires-Dist: pandas-ta (>=0.3.14b0,<0.4.0)
 Requires-Dist: plotly (>=6.1.2,<7.0.0)
 Requires-Dist: streamlit (>=1.45.1,<2.0.0)

{bullishpy-0.11.0 → bullishpy-0.12.0}/bullish/analysis/analysis.py

@@ -1,4 +1,7 @@
 import logging
+import time
+from itertools import batched
+from pathlib import Path
 from typing import (
     Annotated,
     Any,
@@ -40,6 +43,7 @@ from bearish.types import TickerOnlySources  # type: ignore
 from pydantic import BaseModel, BeforeValidator, Field, create_model
 
 from bullish.analysis.indicators import Indicators, IndicatorModels
+from joblib import Parallel, delayed  # type: ignore
 
 if TYPE_CHECKING:
     from bullish.database.crud import BullishDb
@@ -482,10 +486,27 @@ class Analysis(AnalysisView, BaseEquity, TechnicalAnalysis, FundamentalAnalysis)
         )
 
 
+def compute_analysis(database_path: Path, ticker: Ticker) -> Analysis:
+    from bullish.database.crud import BullishDb
+
+    bullish_db = BullishDb(database_path=database_path)
+    return Analysis.from_ticker(bullish_db, ticker)
+
+
 def run_analysis(bullish_db: "BullishDb") -> None:
     price_trackers = set(bullish_db._read_tracker(TrackerQuery(), PriceTracker))
     finance_trackers = set(bullish_db._read_tracker(TrackerQuery(), FinancialsTracker))
     tickers = list(price_trackers.intersection(finance_trackers))
-    for ticker in tickers:
-        analysis = Analysis.from_ticker(bullish_db, ticker)
-        bullish_db.write_analysis(analysis)
+    parallel = Parallel(n_jobs=-1)
+
+    for batch_ticker in batched(tickers, 1000):
+        start = time.perf_counter()
+        many_analysis = parallel(
+            delayed(compute_analysis)(bullish_db.database_path, ticker)
+            for ticker in batch_ticker
+        )
+        bullish_db.write_many_analysis(many_analysis)
+        elapsed_time = time.perf_counter() - start
+        print(
+            f"Computed analysis for {len(batch_ticker)} tickers in {elapsed_time:.2f} seconds."
+        )

{bullishpy-0.11.0 → bullishpy-0.12.0}/bullish/database/crud.py

@@ -67,6 +67,16 @@ class BullishDb(BearishDb, BullishDbBase):  # type: ignore
             session.exec(stmt)  # type: ignore
             session.commit()
 
+    def _write_many_analysis(self, many_analysis: List[Analysis]) -> None:
+        with Session(self._engine) as session:
+            stmt = (
+                insert(AnalysisORM)
+                .prefix_with("OR REPLACE")
+                .values([a.model_dump() for a in many_analysis])
+            )
+            session.exec(stmt)  # type: ignore
+            session.commit()
+
     def _read_analysis(self, ticker: Ticker) -> Optional[Analysis]:
         with Session(self._engine) as session:
             query = select(AnalysisORM).where(AnalysisORM.symbol == ticker.symbol)

{bullishpy-0.11.0 → bullishpy-0.12.0}/bullish/interface/interface.py

@@ -19,6 +19,9 @@ class BullishDbBase(BearishDbBase):  # type: ignore
     def write_analysis(self, analysis: "Analysis") -> None:
         return self._write_analysis(analysis)
 
+    def write_many_analysis(self, many_analysis: List["Analysis"]) -> None:
+        return self._write_many_analysis(many_analysis)
+
     def read_analysis(self, ticker: Ticker) -> Optional["Analysis"]:
         return self._read_analysis(ticker)
 
@@ -62,6 +65,9 @@ class BullishDbBase(BearishDbBase):  # type: ignore
     @abc.abstractmethod
     def _write_analysis(self, analysis: "Analysis") -> None: ...
 
+    @abc.abstractmethod
+    def _write_many_analysis(self, many_analysis: List["Analysis"]) -> None: ...
+
     @abc.abstractmethod
     def _read_analysis(self, ticker: Ticker) -> Optional["Analysis"]: ...
 

{bullishpy-0.11.0 → bullishpy-0.12.0}/pyproject.toml

@@ -1,13 +1,13 @@
 [tool.poetry]
 name = "bullishpy"
-version = "0.11.0"
+version = "0.12.0"
 description = ""
 authors = ["aan <andoludovic.andriamamonjy@gmail.com>"]
 readme = "README.md"
 packages = [{ include = "bullish" }]
 [tool.poetry.dependencies]
 python = ">=3.12,<3.13"
-bearishpy = "^0.20.0"
+bearishpy = "^0.22.0"
 tickermood = "^0.4.0"
 streamlit = "^1.45.1"
 streamlit-pydantic = "^v0.6.1-rc.3"
@@ -17,6 +17,7 @@ pandas-ta = "^0.3.14b0"
 plotly = "^6.1.2"
 ta-lib = "^0.6.4"
 click = ">=7.0,<=8.1"
+joblib = "^1.5.1"
 
 [tool.poetry.scripts]
 bullish = "bullish.cli:app"