bullishpy 0.4.0__py3-none-any.whl → 0.6.0__py3-none-any.whl

bullish/database/alembic/versions/11d35a452b40_.py

@@ -0,0 +1,368 @@
+"""
+
+Revision ID: 11d35a452b40
+Revises: 73564b60fe24
+Create Date: 2025-06-23 06:02:07.122505
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+# revision identifiers, used by Alembic.
+revision: str = "11d35a452b40"
+down_revision: Union[str, None] = "73564b60fe24"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("analysis", schema=None) as batch_op:
+        batch_op.alter_column(
+            "quarterly_positive_free_cash_flow",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_growing_operating_cash_flow",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_operating_cash_flow_is_higher_than_net_income",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_net_income",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_operating_income",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_growing_net_income",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_growing_operating_income",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_diluted_eps",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_basic_eps",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_growing_basic_eps",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_growing_diluted_eps",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_debt_to_equity",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_return_on_assets",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_return_on_equity",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_free_cash_flow",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "growing_operating_cash_flow",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "operating_cash_flow_is_higher_than_net_income",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_net_income",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_operating_income",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "growing_net_income",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "growing_operating_income",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_diluted_eps",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_basic_eps",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "growing_basic_eps",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "growing_diluted_eps",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_debt_to_equity",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_return_on_assets",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_return_on_equity",
+            existing_type=sa.FLOAT(),
+            type_=sa.Boolean(),
+            existing_nullable=True,
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("analysis", schema=None) as batch_op:
+        batch_op.alter_column(
+            "positive_return_on_equity",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_return_on_assets",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_debt_to_equity",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "growing_diluted_eps",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "growing_basic_eps",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_basic_eps",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_diluted_eps",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "growing_operating_income",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "growing_net_income",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_operating_income",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_net_income",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "operating_cash_flow_is_higher_than_net_income",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "growing_operating_cash_flow",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "positive_free_cash_flow",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_return_on_equity",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_return_on_assets",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_debt_to_equity",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_growing_diluted_eps",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_growing_basic_eps",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_basic_eps",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_diluted_eps",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_growing_operating_income",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_growing_net_income",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_operating_income",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_net_income",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_operating_cash_flow_is_higher_than_net_income",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_growing_operating_cash_flow",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+        batch_op.alter_column(
+            "quarterly_positive_free_cash_flow",
+            existing_type=sa.Boolean(),
+            type_=sa.FLOAT(),
+            existing_nullable=True,
+        )
+
+    # ### end Alembic commands ###

bullish/database/alembic/versions/49c83f9eb5ac_.py

@@ -0,0 +1,103 @@
+"""
+
+Revision ID: 49c83f9eb5ac
+Revises: ee5baabb35f8
+Create Date: 2025-07-06 07:29:51.181393
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import sqlite
+
+# revision identifiers, used by Alembic.
+revision: str = "49c83f9eb5ac"
+down_revision: Union[str, None] = "ee5baabb35f8"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("analysis", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column("median_rate_of_change_1", sa.Float(), nullable=True)
+        )
+        batch_op.add_column(
+            sa.Column("median_rate_of_change_7_4", sa.Float(), nullable=True)
+        )
+        batch_op.add_column(
+            sa.Column("median_rate_of_change_7_12", sa.Float(), nullable=True)
+        )
+        batch_op.add_column(
+            sa.Column("median_rate_of_change_30", sa.Float(), nullable=True)
+        )
+        batch_op.add_column(sa.Column("rate_of_change_7", sa.Float(), nullable=True))
+        batch_op.drop_index(batch_op.f("ix_analysis_rate_of_change_1"))
+        batch_op.drop_index(batch_op.f("ix_analysis_rate_of_change_7_12"))
+        batch_op.drop_index(batch_op.f("ix_analysis_rate_of_change_7_4"))
+        batch_op.create_index(
+            "ix_analysis_median_rate_of_change_1",
+            ["median_rate_of_change_1"],
+            unique=False,
+        )
+        batch_op.create_index(
+            "ix_analysis_median_rate_of_change_30",
+            ["median_rate_of_change_30"],
+            unique=False,
+        )
+        batch_op.create_index(
+            "ix_analysis_median_rate_of_change_7_12",
+            ["median_rate_of_change_7_12"],
+            unique=False,
+        )
+        batch_op.create_index(
+            "ix_analysis_median_rate_of_change_7_4",
+            ["median_rate_of_change_7_4"],
+            unique=False,
+        )
+        batch_op.create_index(
+            "ix_analysis_rate_of_change_7", ["rate_of_change_7"], unique=False
+        )
+        batch_op.drop_column("rate_of_change_7_4")
+        batch_op.drop_column("rate_of_change_7_12")
+        batch_op.drop_column("rate_of_change_1")
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("analysis", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("rate_of_change_1", sa.FLOAT(), nullable=True))
+        batch_op.add_column(sa.Column("rate_of_change_7_12", sa.FLOAT(), nullable=True))
+        batch_op.add_column(sa.Column("rate_of_change_7_4", sa.FLOAT(), nullable=True))
+        batch_op.drop_index("ix_analysis_rate_of_change_7")
+        batch_op.drop_index("ix_analysis_median_rate_of_change_7_4")
+        batch_op.drop_index("ix_analysis_median_rate_of_change_7_12")
+        batch_op.drop_index("ix_analysis_median_rate_of_change_30")
+        batch_op.drop_index("ix_analysis_median_rate_of_change_1")
+        batch_op.create_index(
+            batch_op.f("ix_analysis_rate_of_change_7_4"),
+            ["rate_of_change_7_4"],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f("ix_analysis_rate_of_change_7_12"),
+            ["rate_of_change_7_12"],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f("ix_analysis_rate_of_change_1"),
+            ["rate_of_change_1"],
+            unique=False,
+        )
+        batch_op.drop_column("rate_of_change_7")
+        batch_op.drop_column("median_rate_of_change_30")
+        batch_op.drop_column("median_rate_of_change_7_12")
+        batch_op.drop_column("median_rate_of_change_7_4")
+        batch_op.drop_column("median_rate_of_change_1")
+
+    # ### end Alembic commands ###

bullish/database/alembic/versions/4b0a2f40b7d3_.py

@@ -5,6 +5,7 @@ Revises: 037dbd721317
 Create Date: 2025-06-20 09:17:53.566652
 
 """
+
 from typing import Sequence, Union
 
 from alembic import op

bullish/database/alembic/versions/73564b60fe24_.py

@@ -5,6 +5,7 @@ Revises: 4b0a2f40b7d3
 Create Date: 2025-06-20 17:08:28.818293
 
 """
+
 from typing import Sequence, Union
 
 from alembic import op

bullish/database/alembic/versions/ee5baabb35f8_.py

@@ -0,0 +1,51 @@
+"""
+
+Revision ID: ee5baabb35f8
+Revises: 08ac1116e055
+Create Date: 2025-07-02 05:16:42.862366
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import sqlite
+
+# revision identifiers, used by Alembic.
+revision: str = "ee5baabb35f8"
+down_revision: Union[str, None] = "08ac1116e055"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("analysis", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column("adosc_crosses_above_0", sa.Date(), nullable=True)
+        )
+        batch_op.add_column(
+            sa.Column("positive_adosc_20_day_breakout", sa.Date(), nullable=True)
+        )
+        batch_op.create_index(
+            "ix_analysis_adosc_crosses_above_0", ["adosc_crosses_above_0"], unique=False
+        )
+        batch_op.create_index(
+            "ix_analysis_positive_adosc_20_day_breakout",
+            ["positive_adosc_20_day_breakout"],
+            unique=False,
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("analysis", schema=None) as batch_op:
+        batch_op.drop_index("ix_analysis_positive_adosc_20_day_breakout")
+        batch_op.drop_index("ix_analysis_adosc_crosses_above_0")
+        batch_op.drop_column("positive_adosc_20_day_breakout")
+        batch_op.drop_column("adosc_crosses_above_0")
+
+    # ### end Alembic commands ###

bullish/database/crud.py

@@ -1,3 +1,4 @@
+import json
 import logging
 from functools import cached_property
 from pathlib import Path
@@ -64,6 +65,11 @@ class BullishDb(BearishDb, BullishDbBase):  # type: ignore
                 return None
             return Analysis.model_validate(analysis)
 
+    def read_symbols(self) -> List[str]:
+        query = "SELECT DISTINCT symbol FROM analysis"
+        data = pd.read_sql_query(query, self._engine)
+        return data["symbol"].tolist()
+
     def _read_analysis_data(
         self, columns: List[str], symbols: Optional[List[str]] = None
     ) -> pd.DataFrame:
@@ -140,7 +146,13 @@ class BullishDb(BearishDb, BullishDbBase):  # type: ignore
 
     def write_filtered_results(self, filtered_results: FilteredResults) -> None:
         with Session(self._engine) as session:
-            data = filtered_results.model_dump()
-            stmt = insert(FilteredResultsORM).prefix_with("OR REPLACE").values(data)
+            data = filtered_results.model_dump_json(
+                exclude_unset=True, exclude_defaults=True
+            )
+            stmt = (
+                insert(FilteredResultsORM)
+                .prefix_with("OR REPLACE")
+                .values(json.loads(data))
+            )
             session.exec(stmt)  # type: ignore
             session.commit()

bullish/database/schemas.py

@@ -5,6 +5,7 @@ from sqlalchemy import Column, JSON
 from bullish.analysis.analysis import Analysis
 from bullish.analysis.filter import FilteredResults
 from bullish.jobs.models import JobTracker
+from sqlalchemy import Index
 
 
 class BaseTable(SQLModel):
@@ -12,11 +13,23 @@ class BaseTable(SQLModel):
     source: str = Field(primary_key=True)
 
 
+dynamic_indexes = tuple(
+    Index(f"ix_analysis_{col}", col) for col in Analysis.model_fields
+)
+
+
 class AnalysisORM(BaseTable, Analysis, table=True):
     __tablename__ = "analysis"
     __table_args__ = {"extend_existing": True}  # noqa:RUF012
 
 
+AnalysisORM.__table_args__ = tuple(  # type: ignore # noqa: RUF005
+    Index(f"ix_{AnalysisORM.__tablename__}_{col.name}", col)
+    for col in AnalysisORM.__table__.columns
+    if not col.primary_key and not col.index and col.name != "id"
+) + (AnalysisORM.__table_args__,)
+
+
 class JobTrackerORM(SQLModel, JobTracker, table=True):
     __tablename__ = "jobtracker"
     __table_args__ = {"extend_existing": True}  # noqa:RUF012