bullishpy 0.4.0__py3-none-any.whl

bullish/interface/interface.py

@@ -0,0 +1,96 @@
+import abc
+import logging
+from typing import List, Optional
+
+import pandas as pd
+from bearish.interface.interface import BearishDbBase  # type: ignore
+from bearish.models.base import Ticker  # type: ignore
+
+
+from bullish.analysis.analysis import Analysis, AnalysisView
+from bullish.analysis.filter import FilterQuery, FilteredResults
+from bullish.jobs.models import JobTracker, JobTrackerStatus, add_icons
+
+logger = logging.getLogger(__name__)
+
+
+class BullishDbBase(BearishDbBase):  # type: ignore
+    def write_analysis(self, analysis: "Analysis") -> None:
+        return self._write_analysis(analysis)
+
+    def read_analysis(self, ticker: Ticker) -> Optional["Analysis"]:
+        return self._read_analysis(ticker)
+
+    def read_filter_query(self, query: FilterQuery) -> pd.DataFrame:
+
+        if not set(query.query_parameters).issubset(set(Analysis.model_fields)):
+            raise ValueError(
+                f"Query parameters {query.query_parameters} are not a "
+                f"subset of Analysis model fields {Analysis.model_fields}"
+            )
+        query_ = query.to_query()
+        fields = ",".join(list(AnalysisView.model_fields))
+        query_str: str = f""" 
+        SELECT {fields} FROM analysis WHERE {query_} LIMIT 1000
+        """  # noqa: S608
+        return self._read_filter_query(query_str)
+
+    def read_analysis_data(
+        self, columns: Optional[List[str]] = None, symbols: Optional[List[str]] = None
+    ) -> pd.DataFrame:
+        columns = columns or list(AnalysisView.model_fields)
+        data = self._read_analysis_data(columns, symbols=symbols)
+        if set(data.columns) != set(columns):
+            raise ValueError(
+                f"Expected columns {columns}, but got {data.columns.tolist()}"
+            )
+        return data
+
+    def read_job_trackers(self) -> pd.DataFrame:
+        return add_icons(self._read_job_trackers())
+
+    @abc.abstractmethod
+    def _read_job_trackers(self) -> pd.DataFrame:
+        ...
+
+    @abc.abstractmethod
+    def write_job_tracker(self, job_tracker: JobTracker) -> None:
+        ...
+
+    @abc.abstractmethod
+    def delete_job_trackers(self, job_ids: List[str]) -> None:
+        ...
+
+    @abc.abstractmethod
+    def update_job_tracker_status(self, job_tracker_status: JobTrackerStatus) -> None:
+        ...
+
+    @abc.abstractmethod
+    def _write_analysis(self, analysis: "Analysis") -> None:
+        ...
+
+    @abc.abstractmethod
+    def _read_analysis(self, ticker: Ticker) -> Optional["Analysis"]:
+        ...
+
+    @abc.abstractmethod
+    def _read_filter_query(self, query: str) -> pd.DataFrame:
+        ...
+
+    @abc.abstractmethod
+    def _read_analysis_data(
+        self, columns: List[str], symbols: Optional[List[str]] = None
+    ) -> pd.DataFrame:
+        ...
+
+    @abc.abstractmethod
+    def read_filtered_results(self, name: str) -> Optional[FilteredResults]:
+        ...
+
+    @abc.abstractmethod
+    def read_list_filtered_results(self) -> List[str]:
+        ...
+
+    @abc.abstractmethod
+    def write_filtered_results(self, filtered_results: FilteredResults) -> None:
+        ...

bullish/jobs/app.py

@@ -0,0 +1,3 @@
+from huey import SqliteHuey  # type: ignore
+
+huey = SqliteHuey(name="queue")

bullish/jobs/models.py

@@ -0,0 +1,25 @@
+from datetime import datetime
+from typing import Literal, get_args
+
+import pandas as pd
+from pydantic import BaseModel, Field
+
+JobType = Literal["Update data", "Update analysis", "Fetching news"]
+JobStatus = Literal["Completed", "Failed", "Running", "Started"]
+StatusIcon = ["✅ Completed", "❌ Failed", "🔄 Running", "🚀 Started"]
+
+
+class JobTrackerStatus(BaseModel):
+    job_id: str
+    status: JobStatus = "Started"
+
+
+class JobTracker(JobTrackerStatus):
+    type: JobType
+    started_at: datetime = Field(default_factory=datetime.now)
+
+
+def add_icons(data: pd.DataFrame) -> pd.DataFrame:
+    status_map = dict(zip(list(get_args(JobStatus)), StatusIcon, strict=True))
+    data["status"] = data["status"].map(status_map)
+    return data

bullish/jobs/tasks.py

@@ -0,0 +1,77 @@
+import functools
+import logging
+from typing import Optional, Any, Callable, List
+
+from bearish.main import Bearish  # type: ignore
+from tickermood.main import get_news  # type: ignore
+from tickermood.types import DatabaseConfig  # type: ignore
+
+from .app import huey
+from pathlib import Path
+from huey.api import Task  # type: ignore
+
+from .models import JobTrackerStatus
+from ..analysis.analysis import run_analysis
+from ..database.crud import BullishDb
+from bullish.analysis.filter import FilterUpdate
+
+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
+    ) -> None:
+        bullish_db = BullishDb(database_path=database_path)
+        if task is None:
+            raise ValueError("Task must be provided for job tracking.")
+        bullish_db.update_job_tracker_status(
+            JobTrackerStatus(job_id=task.id, status="Running")
+        )
+        try:
+            func(database_path, *args, task=task, **kwargs)
+            bullish_db.update_job_tracker_status(
+                JobTrackerStatus(job_id=task.id, status="Completed")
+            )
+        except Exception as e:
+            logger.exception(f"Fail to complete job {func.__name__}: {e}")
+            bullish_db.update_job_tracker_status(
+                JobTrackerStatus(job_id=task.id, status="Failed")
+            )
+
+    return wrapper
+
+
+@huey.task(context=True)  # type: ignore
+@job_tracker
+def update(
+    database_path: Path,
+    symbols: List[str],
+    update_query: FilterUpdate,
+    task: Optional[Task] = None,
+) -> None:
+    logger.debug(f"Running update task for {len(symbols)} tickers.")
+    if not update_query.update_analysis_only:
+        bearish = Bearish(path=database_path, auto_migration=False)
+        bearish.update_prices(
+            symbols,
+            series_length=update_query.window_size,
+            delay=update_query.data_age_in_days,
+        )
+        if update_query.update_financials:
+            bearish.update_financials(symbols)
+    bullish_db = BullishDb(database_path=database_path)
+    run_analysis(bullish_db)
+
+
+@huey.task(context=True)  # type: ignore
+@job_tracker
+def news(
+    database_path: Path,
+    symbols: List[str],
+    headless: bool = True,
+    task: Optional[Task] = None,
+) -> None:
+    database_config = DatabaseConfig(database_path=database_path, no_migration=True)
+    get_news(symbols, database_config, headless=headless)

bullishpy-0.4.0.dist-info/METADATA

@@ -0,0 +1,22 @@
+Metadata-Version: 2.3
+Name: bullishpy
+Version: 0.4.0
+Summary: 
+Author: aan
+Author-email: andoludovic.andriamamonjy@gmail.com
+Requires-Python: >=3.10,<3.13
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: bearishpy (>=0.19.0,<0.20.0)
+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)
+Requires-Dist: streamlit (>=1.45.1,<2.0.0)
+Requires-Dist: streamlit-file-browser (>=3.2.22,<4.0.0)
+Requires-Dist: streamlit-pydantic (>=v0.6.1-rc.3,<0.7.0)
+Requires-Dist: tickermood (>=0.4.0,<0.5.0)
+Description-Content-Type: text/markdown
+
+## Bullish

bullishpy-0.4.0.dist-info/RECORD

@@ -0,0 +1,34 @@
+bullish/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bullish/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bullish/analysis/analysis.py,sha256=JqaFzWfZ_AxF-7qa7BPmSWESnrcVdSEJKmUMRFGHzew,22168
+bullish/analysis/filter.py,sha256=rNqGGTIlaOq8Gw8mblX09bQOfR98by-NY8VY2TwP5oQ,3783
+bullish/app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bullish/app/app.py,sha256=Ef9IfrOJelV5_oymPFtEhyW438LDw0mlK6EhZR9Yqq8,7656
+bullish/cli.py,sha256=mxobC_49Zpn_qgXThVGhtpexjKLJS2tZZbOlo_wMLfg,1893
+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
+bullish/database/alembic/env.py,sha256=TBsN4TyVppyc2QpWqViebd4-xxUT7Cs3GDYXQdKiAMs,2260
+bullish/database/alembic/script.py.mako,sha256=MEqL-2qATlST9TAOeYgscMn1uy6HUS9NFvDgl93dMj8,635
+bullish/database/alembic/versions/037dbd721317_.py,sha256=pbmfCcJBFwy0SH42lfYVXawDlM72bR0oUUO-0RgFmUM,8594
+bullish/database/alembic/versions/4b0a2f40b7d3_.py,sha256=iyLvZY1Wiy7NucqxPi0TCAYD0Q8zmuFweMRZsAYx6o4,1826
+bullish/database/alembic/versions/73564b60fe24_.py,sha256=zp9Hvr8e3FWc9SLNqiQpbA8lsd0cU8DKDScKQdQwHYA,1007
+bullish/database/crud.py,sha256=xUSSzhHGp-KKBlG4iCNVtAEs9U5BNv4VhGSrLIR4fmM,5611
+bullish/database/schemas.py,sha256=FjFhCa_VijAu2EEWsQ6rShSRzdH6FS9IM66ibh2kx3g,1104
+bullish/database/scripts/create_revision.py,sha256=rggIf-3koPqJNth8FIg89EOfnIM7a9QrvL8X7UJsP0g,628
+bullish/database/scripts/stamp.py,sha256=PWgVUEBumjNUMjTnGw46qmU3p221LeN-KspnW_gFuu4,839
+bullish/database/scripts/upgrade.py,sha256=-Gz7aFNPEt9y9e1kltqXE76-j_8QeNtet_VlwY5AWjo,806
+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=3Ifrnl4I7gq6DiMEWZs2P0589nEsaGKYNiq-Cxj35_g,2771
+bullish/interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bullish/interface/interface.py,sha256=1LiTFaMI1WXgUHLf73_KPiYKbv4RFcctyPxIcvzkkf0,3082
+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=vRjbCBcQciTC9283El_ji7BKxx40IbLcJkMbMVoE5wA,2533
+bullishpy-0.4.0.dist-info/METADATA,sha256=EVIlGsSkq8hAovop7OSQ_tNy2csZzF6QysWCrwXElNg,772
+bullishpy-0.4.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+bullishpy-0.4.0.dist-info/entry_points.txt,sha256=eaPpmL6vmSBFo0FBtwibCXGqAW4LFJ83whJzT1VjD-0,43
+bullishpy-0.4.0.dist-info/RECORD,,

bullishpy-0.4.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 2.1.3
+Root-Is-Purelib: true
+Tag: py3-none-any

bullishpy-0.4.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+bullish=bullish.cli:app
+