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

bullish/figures/figures.py

@@ -1,17 +1,19 @@
-from typing import Optional
+import datetime
+from typing import Optional, List
 
 import pandas as pd
 import plotly.graph_objects as go
 from plotly.subplots import make_subplots
 
 from bullish.analysis.functions import add_indicators
+from datetime import date
 
 
 def plot(
     data: pd.DataFrame,
     symbol: str,
     name: Optional[str] = None,
-    dates: Optional[pd.Series] = None,
+    dates: Optional[List[date]] = None,
 ) -> go.Figure:
     data = add_indicators(data)
     fig = make_subplots(
@@ -125,10 +127,21 @@ def plot(
         row=7,
         col=1,
     )
-    if dates is not None and not dates.empty:
+    if dates is not None and dates:
         for date in dates:
+            if (
+                data.first_valid_index().date() > date  # type: ignore
+                or data.last_valid_index().date() + datetime.timedelta(days=31 * 3)  # type: ignore
+                < date
+            ):
+                continue
             fig.add_vline(
-                x=date, line_dash="dashdot", line_color="MediumPurple", line_width=3
+                x=date,
+                line_dash="dashdot",
+                line_color="MediumPurple",
+                line_width=1,
+                row=1,
+                col=1,
             )
 
     # Layout tweaks

bullish/interface/interface.py

@@ -1,5 +1,6 @@
 import abc
 import logging
+from datetime import date
 from typing import List, Optional
 
 import pandas as pd
@@ -83,3 +84,9 @@ class BullishDbBase(BearishDbBase):  # type: ignore
 
     @abc.abstractmethod
     def read_symbols(self) -> List[str]: ...
+
+    @abc.abstractmethod
+    def read_job_tracker(self, task_id: str) -> Optional[JobTracker]: ...
+
+    @abc.abstractmethod
+    def read_dates(self, symbol: str) -> List[date]: ...

bullish/jobs/tasks.py

@@ -10,7 +10,7 @@ from .app import huey
 from pathlib import Path
 from huey.api import Task  # type: ignore
 
-from .models import JobTrackerStatus
+from .models import JobTrackerStatus, JobTracker, JobType
 from ..analysis.analysis import run_analysis
 from ..database.crud import BullishDb
 from bullish.analysis.filter import FilterUpdate
@@ -21,16 +21,22 @@ logger = logging.getLogger(__name__)
 def job_tracker(func: Callable[..., Any]) -> Callable[..., Any]:
     @functools.wraps(func)
     def wrapper(
-        database_path: Path, *args: Any, task: Optional[Task] = None, **kwargs: Any
+        database_path: Path,
+        job_type: JobType,
+        *args: Any,
+        task: Optional[Task] = None,
+        **kwargs: Any,
     ) -> None:
         bullish_db = BullishDb(database_path=database_path)
         if task is None:
             raise ValueError("Task must be provided for job tracking.")
+        if bullish_db.read_job_tracker(task.id) is None:
+            bullish_db.write_job_tracker(JobTracker(job_id=str(task.id), type=job_type))
         bullish_db.update_job_tracker_status(
             JobTrackerStatus(job_id=task.id, status="Running")
         )
         try:
-            func(database_path, *args, task=task, **kwargs)
+            func(database_path, job_type, *args, task=task, **kwargs)
             bullish_db.update_job_tracker_status(
                 JobTrackerStatus(job_id=task.id, status="Completed")
             )
@@ -47,6 +53,7 @@ def job_tracker(func: Callable[..., Any]) -> Callable[..., Any]:
 @job_tracker
 def update(
     database_path: Path,
+    job_type: JobType,
     symbols: List[str],
     update_query: FilterUpdate,
     task: Optional[Task] = None,
@@ -69,6 +76,7 @@ def update(
 @job_tracker
 def analysis(
     database_path: Path,
+    job_type: JobType,
     task: Optional[Task] = None,
 ) -> None:
     bullish_db = BullishDb(database_path=database_path)
@@ -79,6 +87,7 @@ def analysis(
 @job_tracker
 def news(
     database_path: Path,
+    job_type: JobType,
     symbols: List[str],
     headless: bool = True,
     task: Optional[Task] = None,

bullish/utils/checks.py

@@ -40,6 +40,8 @@ def get_table_names_from_path(database_path: Path) -> List[str]:
 
 
 def empty_analysis_table(database_path: Path) -> bool:
+    if "analysis" not in get_table_names_from_path(database_path):
+        return True
     with get_sqlite_connection(database_path) as conn:
         cursor = conn.cursor()
 

{bullishpy-0.9.0.dist-info → bullishpy-0.11.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: bullishpy
-Version: 0.9.0
+Version: 0.11.0
 Summary: 
 Author: aan
 Author-email: andoludovic.andriamamonjy@gmail.com
@@ -8,6 +8,7 @@ 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: click (>=7.0,<=8.1)
 Requires-Dist: huey (>=2.5.3,<3.0.0)
 Requires-Dist: pandas-ta (>=0.3.14b0,<0.4.0)
 Requires-Dist: plotly (>=6.1.2,<7.0.0)

{bullishpy-0.9.0.dist-info → bullishpy-0.11.0.dist-info}/RECORD

@@ -1,13 +1,13 @@
 bullish/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bullish/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bullish/analysis/analysis.py,sha256=ZqIMdAq1NJYgoQZ52PsV724VsXK8Jtno4alFFd_7_XU,17842
-bullish/analysis/filter.py,sha256=DzhA-cuKzZhIL1kwi61PTz77BBo8B6G_7EI4OVkPrww,17123
-bullish/analysis/functions.py,sha256=yM4d7C_2gwEnV_t1-TWvYOzpeKx36WhupjPZxBSsOYg,12317
-bullish/analysis/indicators.py,sha256=kOvtnrz-9-_fCUUFE842ijwjDbF9X2O-W4YoPkSDisk,18117
-bullish/analysis/predefined_filters.py,sha256=g4vScDC4qiEzKb6gVhvxSFriw1fbi71k1xV3nNS7VS4,4715
+bullish/analysis/analysis.py,sha256=9ugJp2fGuA6xFUlGIJVIHWZ8d8E6TdbQHWjg5zWPIwY,18445
+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
+bullish/analysis/predefined_filters.py,sha256=3O244FCZkbnTSaReh6w6cwEIXdLzQZmbkBS0uRW-Y0M,12391
 bullish/app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bullish/app/app.py,sha256=Es244AnLce32d-MUYbKXZeEPE8GlfCUO4SQ7aiqooGc,12539
-bullish/cli.py,sha256=C31Pj7XGzdLz2Y3nIPQ7CF1DvyGVU5EyLvzj423QbwQ,1915
+bullish/app/app.py,sha256=E0H78LOODl1H6s308jXpQGTUoFPoLOJkPBXOLQGLCeA,13331
+bullish/cli.py,sha256=uYLZmGDAolZKWzduZ58bP-xul1adg0oKfeUQtZMXTvA,1958
 bullish/database/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bullish/database/alembic/README,sha256=heMzebYwlGhnE8_4CWJ4LS74WoEZjBy-S-mIJRxAEKI,39
 bullish/database/alembic/alembic.ini,sha256=VuwqBJV5ObTyyRNrqv8Xr-TDIRfqPjP9R1mqewYM_xE,3695
@@ -16,12 +16,14 @@ bullish/database/alembic/script.py.mako,sha256=MEqL-2qATlST9TAOeYgscMn1uy6HUS9NF
 bullish/database/alembic/versions/037dbd721317_.py,sha256=U7EA4odH3t9w0-J4FmvBUt8HOuGDMn0rEAu_0vPUYaI,8595
 bullish/database/alembic/versions/08ac1116e055_.py,sha256=zMEiCbraMEAZItT4ibc3evAH7-7mkXpdgnZy4tPVYeg,27263
 bullish/database/alembic/versions/11d35a452b40_.py,sha256=j2PaU1RssLQ20OevGmBC7S9E9ocWiXpBue9SOS4AQoY,11521
+bullish/database/alembic/versions/17e51420e7ad_.py,sha256=xeiVIm1YUZb08opE9rocHZP1__9WQWXsKsXgeFV9cvs,2960
 bullish/database/alembic/versions/49c83f9eb5ac_.py,sha256=kCBItp7KmqpJ03roy5ikQjhefZia1oKgfZwournQDq8,3890
 bullish/database/alembic/versions/4b0a2f40b7d3_.py,sha256=G0K7w7pOPYjPZkXTB8LWhxoxuWBPcPwOfnubTBtdeEY,1827
 bullish/database/alembic/versions/73564b60fe24_.py,sha256=MTlDRDNHj3E9gK7IMeAzv2UxxxYtWiu3gI_9xTLE-wg,1008
+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=2D0uDxJlDv1eJFRmgipiCQxCvlC5ILB2OR4OORlIJf4,5967
+bullish/database/crud.py,sha256=6-Fb1AjGZqsrmwwl2Qay_leqQ9_-RAIjZ8D0efe8nKA,7022
 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
@@ -29,16 +31,16 @@ bullish/database/scripts/upgrade.py,sha256=-Gz7aFNPEt9y9e1kltqXE76-j_8QeNtet_Vlw
 bullish/database/settings.py,sha256=nMudufmF7iC_62_PHrGSMjlqDLN2I0qTbtz9JKZHSko,164
 bullish/exceptions.py,sha256=4z_i-dD-CDz1bkGmZH9DOf1L_awlCPCgdUDPF7dhWAI,106
 bullish/figures/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bullish/figures/figures.py,sha256=W4XJIs5wFtpX75OgoocVcuuA8Hdb_SNN58VW0LYI5oI,3808
+bullish/figures/figures.py,sha256=SWTTiEoVyWMZeIIxg0ERi23v7s4tySB5BLKyPu12jC4,4193
 bullish/interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bullish/interface/interface.py,sha256=T3MdJbX2ZF7hiMhnZ00ppXSXbDwyxaqD69a3M4bPswU,2806
+bullish/interface/interface.py,sha256=-3V4M1J2VVn0ugwbaCKzMuBYRHH7eFmz9gij0hUTNUw,3013
 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=gJEB342nCTw2KD3YgDXpQhGGITcXd14GAMiqRRVbZ-A,2757
+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=rUyFNkx9F5DuOSxjldXymNHwRxIddN7PYbqPICnz1uM,2101
-bullishpy-0.9.0.dist-info/METADATA,sha256=_8wyV_EDN-HtaPLN38wq7U9r8_1sBi5yFnIimjkny5s,709
-bullishpy-0.9.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-bullishpy-0.9.0.dist-info/entry_points.txt,sha256=eaPpmL6vmSBFo0FBtwibCXGqAW4LFJ83whJzT1VjD-0,43
-bullishpy-0.9.0.dist-info/RECORD,,
+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,,