bullishpy 0.11.0__py3-none-any.whl → 0.12.0__py3-none-any.whl

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."
+        )

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)

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.dist-info → bullishpy-0.12.0.dist-info}/METADATA

@@ -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.dist-info → bullishpy-0.12.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 bullish/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bullish/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bullish/analysis/analysis.py,sha256=9ugJp2fGuA6xFUlGIJVIHWZ8d8E6TdbQHWjg5zWPIwY,18445
+bullish/analysis/analysis.py,sha256=v5MFGhQUpMwGAVE0dg2293ldRSgQmqy5CiF8MQFTibM,19165
 bullish/analysis/filter.py,sha256=S8TuxoTAUY0U8ARPjNHE0tSSE_ToWkfZazAgnfgswk4,18136
 bullish/analysis/functions.py,sha256=KKz_0C7maQmcGu2tGwZvioxzmh-JcB-YNpPQGjyyheA,13825
 bullish/analysis/indicators.py,sha256=hZgzTq-80XPP6x7dXGhxd-Zzgra-6D-g3pVxUBYOW44,20167
@@ -23,7 +23,7 @@ bullish/database/alembic/versions/73564b60fe24_.py,sha256=MTlDRDNHj3E9gK7IMeAzv2
 bullish/database/alembic/versions/d663166c531d_.py,sha256=U92l6QXqPniAYrPeu2Bt77ReDbXveLj4aGXtgd806JY,1915
 bullish/database/alembic/versions/ee5baabb35f8_.py,sha256=nBMEY-_C8AsSXVPyaDdUkwrFFo2gxShzJhmrjejDwtc,1632
 bullish/database/alembic/versions/fc191121f522_.py,sha256=0sstF6TpAJ09-Mt-Vek9SdSWksvi4C58a5D92rBtuY8,1894
-bullish/database/crud.py,sha256=6-Fb1AjGZqsrmwwl2Qay_leqQ9_-RAIjZ8D0efe8nKA,7022
+bullish/database/crud.py,sha256=0AMrs07MMeDq5AVft_pf43KWPFOusMNzUmxuXwMX6K4,7400
 bullish/database/schemas.py,sha256=bU-DW49NqpBp--1VN486LUdDmLeScrI8TF69afzjoTc,1507
 bullish/database/scripts/create_revision.py,sha256=rggIf-3koPqJNth8FIg89EOfnIM7a9QrvL8X7UJsP0g,628
 bullish/database/scripts/stamp.py,sha256=PWgVUEBumjNUMjTnGw46qmU3p221LeN-KspnW_gFuu4,839
@@ -33,14 +33,14 @@ bullish/exceptions.py,sha256=4z_i-dD-CDz1bkGmZH9DOf1L_awlCPCgdUDPF7dhWAI,106
 bullish/figures/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bullish/figures/figures.py,sha256=SWTTiEoVyWMZeIIxg0ERi23v7s4tySB5BLKyPu12jC4,4193
 bullish/interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bullish/interface/interface.py,sha256=-3V4M1J2VVn0ugwbaCKzMuBYRHH7eFmz9gij0hUTNUw,3013
+bullish/interface/interface.py,sha256=6HXSK-N-vOuOGVserizknP4CeUGbUhB8P642oZmgpk0,3252
 bullish/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bullish/jobs/app.py,sha256=5MJ5KXUo7JSNAvOPgkpIMasD11VTrjQvGzM7vmCY65E,77
 bullish/jobs/models.py,sha256=ndrGTMP08S57yGLGEG9TQt8Uw2slc4HvbG-TZtEEuN0,744
 bullish/jobs/tasks.py,sha256=V_b0c8_GQC0-KIxaHDlLFhtkclQJOsck0gXaW6OlC_w,3055
 bullish/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bullish/utils/checks.py,sha256=Va10_xDVVnxYkOD2hafvyQ-TFV8FQpOkr4huJ7XgpDM,2188
-bullishpy-0.11.0.dist-info/METADATA,sha256=2eQfm9prQaZjTnRzLUDvRd0jjE8EapQ1RMhJtQ0CXjQ,745
-bullishpy-0.11.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-bullishpy-0.11.0.dist-info/entry_points.txt,sha256=eaPpmL6vmSBFo0FBtwibCXGqAW4LFJ83whJzT1VjD-0,43
-bullishpy-0.11.0.dist-info/RECORD,,
+bullishpy-0.12.0.dist-info/METADATA,sha256=hmEQ5ZWQ8ROXE5_Vm0l3npGSM6BfuBQD1-glStH16us,784
+bullishpy-0.12.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+bullishpy-0.12.0.dist-info/entry_points.txt,sha256=eaPpmL6vmSBFo0FBtwibCXGqAW4LFJ83whJzT1VjD-0,43
+bullishpy-0.12.0.dist-info/RECORD,,