bullishpy 0.13.0__py3-none-any.whl → 0.15.0__py3-none-any.whl

bullish/analysis/filter.py

@@ -1,6 +1,6 @@
 import datetime
 from datetime import date
-from typing import Literal, get_args, Any, Optional, List, Tuple, Type, Dict
+from typing import get_args, Any, Optional, List, Tuple, Type, Dict
 
 from bearish.types import SeriesLength  # type: ignore
 from pydantic import BaseModel, Field, ConfigDict
@@ -14,411 +14,8 @@ from bullish.analysis.analysis import (
     TechnicalAnalysis,
     AnalysisView,
 )
+from bullish.analysis.constants import Industry, IndustryGroup, Sector, Country
 
-Industry = Literal[
-    "Publishing",
-    "Internet Retail",
-    "Scientific & Technical Instruments",
-    "Engineering & Construction",
-    "Diagnostics & Research",
-    "Software - Infrastructure",
-    "Thermal Coal",
-    "Software - Application",
-    "Auto Manufacturers",
-    "Farm Products",
-    "Medical Devices",
-    "Education & Training Services",
-    "Auto Parts",
-    "Specialty Chemicals",
-    "Marine Shipping",
-    "Biotechnology",
-    "Real Estate Services",
-    "Gold",
-    "Entertainment",
-    "Specialty Retail",
-    "Utilities - Independent Power Producers",
-    "Steel",
-    "Mortgage Finance",
-    "Communication Equipment",
-    "Drug Manufacturers - Specialty & Generic",
-    "Electronic Gaming & Multimedia",
-    "Banks - Regional",
-    "Oil & Gas E&P",
-    "Travel Services",
-    "Real Estate - Diversified",
-    "Telecom Services",
-    "Uranium",
-    "Consulting Services",
-    "Waste Management",
-    "Agricultural Inputs",
-    "Utilities - Diversified",
-    "Auto & Truck Dealerships",
-    "Confectioners",
-    "Other Industrial Metals & Mining",
-    "Beverages - Wineries & Distilleries",
-    "Oil & Gas Midstream",
-    "Recreational Vehicles",
-    "Electrical Equipment & Parts",
-    "Household & Personal Products",
-    "Packaging & Containers",
-    "REIT - Specialty",
-    "Home Improvement Retail",
-    "Electronic Components",
-    "Asset Management",
-    "Consumer Electronics",
-    "Conglomerates",
-    "Health Information Services",
-    "Medical Instruments & Supplies",
-    "Building Products & Equipment",
-    "Information Technology Services",
-    "Specialty Industrial Machinery",
-    "Food Distribution",
-    "Packaged Foods",
-    "Rental & Leasing Services",
-    "Medical Distribution",
-    "Grocery Stores",
-    "Advertising Agencies",
-    "Beverages - Non - Alcoholic",
-    "Apparel Manufacturing",
-    "Oil & Gas Equipment & Services",
-    "Coking Coal",
-    "Industrial Distribution",
-    "Restaurants",
-    "Beverages - Brewers",
-    "Chemicals",
-    "Real Estate - Development",
-    "Credit Services",
-    "Tobacco",
-    "Metal Fabrication",
-    "Building Materials",
-    "Residential Construction",
-    "Specialty Business Services",
-    "REIT - Hotel & Motel",
-    "Internet Content & Information",
-    "Lodging",
-    "Furnishings, Fixtures & Appliances",
-    "Airlines",
-    "Computer Hardware",
-    "Integrated Freight & Logistics",
-    "Solar",
-    "Capital Markets",
-    "Leisure",
-    "Airports & Air Services",
-    "Aluminum",
-    "Insurance Brokers",
-    "Semiconductors",
-    "REIT - Retail",
-    "Luxury Goods",
-    "Lumber & Wood Production",
-    "REIT - Mortgage",
-    "Semiconductor Equipment & Materials",
-    "Aerospace & Defense",
-    "Security & Protection Services",
-    "Utilities - Renewable",
-    "Utilities - Regulated Gas",
-    "Apparel Retail",
-    "Pollution & Treatment Controls",
-    "Broadcasting",
-    "Resorts & Casinos",
-    "Other Precious Metals & Mining",
-    "Financial Data & Stock Exchanges",
-    "Footwear & Accessories",
-    "Medical Care Facilities",
-    "Electronics & Computer Distribution",
-    "Gambling",
-    "Tools & Accessories",
-    "Insurance - Property & Casualty",
-    "Utilities - Regulated Water",
-    "Insurance - Specialty",
-    "Personal Services",
-    "Pharmaceutical Retailers",
-    "Farm & Heavy Construction Machinery",
-    "Utilities - Regulated Electric",
-    "Department Stores",
-    "Staffing & Employment Services",
-    "Textile Manufacturing",
-    "Silver",
-    "REIT - Industrial",
-    "REIT - Diversified",
-    "Copper",
-    "Business Equipment & Supplies",
-    "Infrastructure Operations",
-    "Trucking",
-    "Insurance - Reinsurance",
-    "Insurance - Diversified",
-    "Drug Manufacturers - General",
-    "Oil & Gas Drilling",
-    "Banks - Diversified",
-    "REIT - Residential",
-    "Oil & Gas Refining & Marketing",
-    "Shell Companies",
-    "Financial Conglomerates",
-    "Paper & Paper Products",
-    "Insurance - Life",
-    "REIT - Office",
-    "Railroads",
-    "Oil & Gas Integrated",
-    "Healthcare Plans",
-    "REIT - Healthcare Facilities",
-    "Discount Stores",
-]
-
-IndustryGroup = Literal[
-    "publishing",
-    "internet-retail",
-    "scientific-technical-instruments",
-    "engineering-construction",
-    "diagnostics-research",
-    "software-infrastructure",
-    "thermal-coal",
-    "software-application",
-    "auto-manufacturers",
-    "farm-products",
-    "medical-devices",
-    "education-training-services",
-    "auto-parts",
-    "specialty-chemicals",
-    "marine-shipping",
-    "biotechnology",
-    "real-estate-services",
-    "gold",
-    "entertainment",
-    "specialty-retail",
-    "utilities-independent-power-producers",
-    "steel",
-    "mortgage-finance",
-    "communication-equipment",
-    "drug-manufacturers-specialty-generic",
-    "electronic-gaming-multimedia",
-    "banks-regional",
-    "oil-gas-e-p",
-    "travel-services",
-    "real-estate-diversified",
-    "telecom-services",
-    "uranium",
-    "consulting-services",
-    "waste-management",
-    "agricultural-inputs",
-    "utilities-diversified",
-    "auto-truck-dealerships",
-    "confectioners",
-    "other-industrial-metals-mining",
-    "beverages-wineries-distilleries",
-    "oil-gas-midstream",
-    "recreational-vehicles",
-    "electrical-equipment-parts",
-    "household-personal-products",
-    "packaging-containers",
-    "reit-specialty",
-    "home-improvement-retail",
-    "electronic-components",
-    "asset-management",
-    "consumer-electronics",
-    "conglomerates",
-    "health-information-services",
-    "medical-instruments-supplies",
-    "building-products-equipment",
-    "information-technology-services",
-    "specialty-industrial-machinery",
-    "food-distribution",
-    "packaged-foods",
-    "rental-leasing-services",
-    "medical-distribution",
-    "grocery-stores",
-    "advertising-agencies",
-    "beverages-non-alcoholic",
-    "apparel-manufacturing",
-    "oil-gas-equipment-services",
-    "coking-coal",
-    "industrial-distribution",
-    "restaurants",
-    "beverages-brewers",
-    "chemicals",
-    "real-estate-development",
-    "credit-services",
-    "tobacco",
-    "metal-fabrication",
-    "building-materials",
-    "residential-construction",
-    "specialty-business-services",
-    "reit-hotel-motel",
-    "internet-content-information",
-    "lodging",
-    "furnishings-fixtures-appliances",
-    "airlines",
-    "computer-hardware",
-    "integrated-freight-logistics",
-    "solar",
-    "capital-markets",
-    "leisure",
-    "airports-air-services",
-    "aluminum",
-    "insurance-brokers",
-    "semiconductors",
-    "reit-retail",
-    "luxury-goods",
-    "lumber-wood-production",
-    "reit-mortgage",
-    "semiconductor-equipment-materials",
-    "aerospace-defense",
-    "security-protection-services",
-    "utilities-renewable",
-    "utilities-regulated-gas",
-    "apparel-retail",
-    "pollution-treatment-controls",
-    "broadcasting",
-    "resorts-casinos",
-    "other-precious-metals-mining",
-    "financial-data-stock-exchanges",
-    "footwear-accessories",
-    "medical-care-facilities",
-    "electronics-computer-distribution",
-    "gambling",
-    "tools-accessories",
-    "insurance-property-casualty",
-    "utilities-regulated-water",
-    "insurance-specialty",
-    "personal-services",
-    "pharmaceutical-retailers",
-    "farm-heavy-construction-machinery",
-    "utilities-regulated-electric",
-    "department-stores",
-    "staffing-employment-services",
-    "textile-manufacturing",
-    "silver",
-    "reit-industrial",
-    "reit-diversified",
-    "copper",
-    "business-equipment-supplies",
-    "infrastructure-operations",
-    "trucking",
-    "insurance-reinsurance",
-    "insurance-diversified",
-    "drug-manufacturers-general",
-    "oil-gas-drilling",
-    "banks-diversified",
-    "reit-residential",
-    "oil-gas-refining-marketing",
-    "shell-companies",
-    "financial-conglomerates",
-    "paper-paper-products",
-    "insurance-life",
-    "reit-office",
-    "railroads",
-    "oil-gas-integrated",
-    "healthcare-plans",
-    "reit-healthcare-facilities",
-    "discount-stores",
-]
-
-Sector = Literal[
-    "Communication Services",
-    "Consumer Cyclical",
-    "Technology",
-    "Industrials",
-    "Healthcare",
-    "Energy",
-    "Consumer Defensive",
-    "Basic Materials",
-    "Real Estate",
-    "Utilities",
-    "Financial Services",
-    "Conglomerates",
-]
-
-Country = Literal[
-    "Australia",
-    "China",
-    "Japan",
-    "United kingdom",
-    "United states",
-    "Poland",
-    "Switzerland",
-    "Canada",
-    "Greece",
-    "Spain",
-    "Germany",
-    "Indonesia",
-    "Belgium",
-    "France",
-    "Netherlands",
-    "British virgin islands",
-    "Italy",
-    "Hungary",
-    "Austria",
-    "Finland",
-    "Sweden",
-    "Bermuda",
-    "Taiwan",
-    "Israel",
-    "Ukraine",
-    "Singapore",
-    "Jersey",
-    "Ireland",
-    "Luxembourg",
-    "Cyprus",
-    "Cayman islands",
-    "Norway",
-    "Denmark",
-    "Hong kong",
-    "New zealand",
-    "Kazakhstan",
-    "Nigeria",
-    "Argentina",
-    "Brazil",
-    "Czech republic",
-    "Mauritius",
-    "South africa",
-    "India",
-    "Mexico",
-    "Mongolia",
-    "Slovenia",
-    "Thailand",
-    "Malaysia",
-    "Costa rica",
-    "Isle of man",
-    "Egypt",
-    "Turkey",
-    "United arab emirates",
-    "Colombia",
-    "Gibraltar",
-    "Malta",
-    "Liechtenstein",
-    "Guernsey",
-    "Peru",
-    "Estonia",
-    "French guiana",
-    "Portugal",
-    "Uruguay",
-    "Chile",
-    "Martinique",
-    "Monaco",
-    "Panama",
-    "Papua new guinea",
-    "South korea",
-    "Macau",
-    "Gabon",
-    "Romania",
-    "Senegal",
-    "Morocco",
-    "Jordan",
-    "Lithuania",
-    "Dominican republic",
-    "Reunion",
-    "Zambia",
-    "Cambodia",
-    "Myanmar",
-    "Bahamas",
-    "Philippines",
-    "Bangladesh",
-    "Latvia",
-    "Vietnam",
-    "Iceland",
-    "Azerbaijan",
-    "Georgia",
-    "Liberia",
-    "Kenya",
-]
 SIZE_RANGE = 2
 
 

bullish/analysis/functions.py

@@ -1,4 +1,3 @@
-import datetime
 import logging
 from datetime import date
 from typing import Optional, Callable, cast
@@ -16,10 +15,17 @@ except Exception:
     logger.warning("Talib is not installed, skipping analysis")
 
 
+def cross_simple(
+    series_a: pd.Series, series_b: pd.Series, above: bool = True
+) -> pd.Series:
+    crossing = ta.cross(series_a=series_a, series_b=series_b, above=above)
+    return crossing  # type: ignore
+
+
 def cross(
     series_a: pd.Series, series_b: pd.Series, above: bool = True
 ) -> Optional[date]:
-    crossing = ta.cross(series_a=series_a, series_b=series_b, above=above)
+    crossing = cross_simple(series_a=series_a, series_b=series_b, above=above)
     if not crossing[crossing == 1].index.empty:
         return crossing[crossing == 1].last_valid_index().date()  # type: ignore
     return None
@@ -32,10 +38,10 @@ def cross_value(series: pd.Series, number: int, above: bool = True) -> Optional[
 def cross_value_series(
     series_a: pd.Series, number: int, above: bool = True
 ) -> pd.Series:
-    crossing = ta.cross(
-        series_a=series_a, series_b=pd.Series(number, index=series_a.index), above=above
+    crossing = cross_simple(
+        series_a, pd.Series(number, index=series_a.index), above=above
     )
-    return crossing  # type: ignore
+    return crossing
 
 
 def compute_adx(data: pd.DataFrame) -> pd.DataFrame:
@@ -282,24 +288,6 @@ def compute_price(data: pd.DataFrame) -> pd.DataFrame:
     return results
 
 
-def compute_percentile_return_after_rsi_crossover(
-    data: pd.DataFrame, rsi_threshold: int = 45, period: int = 90
-) -> float:
-    data_ = cross_value_series(data.RSI, rsi_threshold)
-    values = []
-    for crossing_date in data_[data_ == 1].index:
-        data_crossed = data[
-            (data.index >= crossing_date)
-            & (data.index <= crossing_date + datetime.timedelta(days=period))
-        ]
-        v = (
-            data_crossed.CLOSE.pct_change(periods=len(data_crossed.CLOSE) - 1).iloc[-1]
-            * 100
-        )
-        values.append(v)
-    return float(np.percentile(values, 30))
-
-
 def find_last_true_run_start(series: pd.Series) -> Optional[date]:
     if not series.iloc[-1]:
         return None
@@ -311,12 +299,14 @@ def find_last_true_run_start(series: pd.Series) -> Optional[date]:
     return series.index[last_true_run_start].date()  # type: ignore
 
 
-def momentum(data: pd.DataFrame) -> Optional[date]:
+def sma_50_above_sma_200(data: pd.DataFrame) -> Optional[date]:
+    date_1 = find_last_true_run_start(data.SMA_50 > data.SMA_200)
+    return date_1
+
+
+def price_above_sma50(data: pd.DataFrame) -> Optional[date]:
     date_1 = find_last_true_run_start(data.SMA_50 < data.CLOSE)
-    date_2 = find_last_true_run_start(data.SMA_200 < data.SMA_50)
-    if date_1 is None or date_2 is None:
-        return None
-    return max(date_1, date_2)
+    return date_1
 
 
 class IndicatorFunction(BaseModel):