bbstrader 0.2.98__py3-none-any.whl → 0.2.991__py3-none-any.whl

bbstrader/metatrader/trade.py

@@ -150,6 +150,7 @@ Orders = Literal[
     "sell_stop_limits",
 ]
 
+EXPERT_ID = 98181105
 
 class Trade(RiskManagement):
     """
@@ -209,7 +210,7 @@ class Trade(RiskManagement):
         self,
         symbol: str = "EURUSD",
         expert_name: str = "bbstrader",
-        expert_id: int = 9818,
+        expert_id: int = EXPERT_ID,
         version: str = "2.0",
         target: float = 5.0,
         start_time: str = "0:00",
@@ -376,21 +377,18 @@ class Trade(RiskManagement):
 
     def summary(self):
         """Show a brief description about the trading program"""
+        start = datetime.strptime(self.start, "%H:%M").time()
+        finish = datetime.strptime(self.finishing, "%H:%M").time()
+        end = datetime.strptime(self.end, "%H:%M").time()
         summary_data = [
             ["Expert Advisor Name", f"@{self.expert_name}"],
             ["Expert Advisor Version", f"@{self.version}"],
             ["Expert | Strategy ID", self.expert_id],
             ["Trading Symbol", self.symbol],
             ["Trading Time Frame", self.tf],
-            ["Start Trading Time", f"{self.start_time_hour}:{self.start_time_minutes}"],
-            [
-                "Finishing Trading Time",
-                f"{self.finishing_time_hour}:{self.finishing_time_minutes}",
-            ],
-            [
-                "Closing Position After",
-                f"{self.ending_time_hour}:{self.ending_time_minutes}",
-            ],
+            ["Start Trading Time", f"{start}"],
+            ["Finishing Trading Time", f"{finish}"],
+            ["Closing Position After", f"{end}"],
         ]
         # Custom table format
         summary_table = tabulate(
@@ -520,7 +518,7 @@ class Trade(RiskManagement):
         tp: Optional[float] = None,
     ):
         """
-        Open a Buy positin
+        Open a Buy position
 
         Args:
             action (str): `BMKT` for Market orders or `BLMT`,
@@ -533,6 +531,9 @@ class Trade(RiskManagement):
             mm (bool): Weither to put stop loss and tp or not
             trail (bool): Weither to trail the stop loss or not
             comment (str): The comment for the opening position
+            volume (float): The volume (lot) to trade
+            sl (float): The stop loss price
+            tp (float): The take profit price
         """
         Id = id if id is not None else self.expert_id
         point = self.get_symbol_info(self.symbol).point
@@ -582,7 +583,7 @@ class Trade(RiskManagement):
         return False
 
     def _order_type(self):
-        type = {
+        return {
             "BMKT": (Mt5.ORDER_TYPE_BUY, "BUY"),
             "SMKT": (Mt5.ORDER_TYPE_BUY, "SELL"),
             "BLMT": (Mt5.ORDER_TYPE_BUY_LIMIT, "BUY_LIMIT"),
@@ -592,7 +593,6 @@ class Trade(RiskManagement):
             "BSTPLMT": (Mt5.ORDER_TYPE_BUY_STOP_LIMIT, "BUY_STOP_LIMIT"),
             "SSTPLMT": (Mt5.ORDER_TYPE_SELL_STOP_LIMIT, "SELL_STOP_LIMIT"),
         }
-        return type
 
     def open_sell_position(
         self,
@@ -609,7 +609,7 @@ class Trade(RiskManagement):
         tp: Optional[float] = None,
     ):
         """
-        Open a sell positin
+        Open a sell position
 
         Args:
             action (str): `SMKT` for Market orders
@@ -622,10 +622,9 @@ class Trade(RiskManagement):
             mm (bool): Weither to put stop loss and tp or not
             trail (bool): Weither to trail the stop loss or not
             comment (str): The comment for the closing position
-                        symbol (str): The symbol to trade
             volume (float): The volume (lot) to trade
-            sl (float): The stop loss in points
-            tp (float): The take profit in points
+            sl (float): The stop loss price
+            tp (float): The take profit price
         """
         Id = id if id is not None else self.expert_id
         point = self.get_symbol_info(self.symbol).point
@@ -691,11 +690,11 @@ class Trade(RiskManagement):
             LOGGER.info(f"Not Trading time, SYMBOL={self.symbol}")
             return False
         elif not self.is_risk_ok():
-            LOGGER.error(f"Account Risk not allowed, SYMBOL={self.symbol}")
+            LOGGER.warning(f"Account Risk not allowed, SYMBOL={self.symbol}")
             self._check(comment)
             return False
         elif self.is_max_trades_reached():
-            LOGGER.error(f"Maximum trades reached for Today, SYMBOL={self.symbol}")
+            LOGGER.warning(f"Maximum trades reached for Today, SYMBOL={self.symbol}")
             return False
         elif self.profit_target():
             self._check(f"Profit target Reached !!! SYMBOL={self.symbol}")
@@ -835,7 +834,8 @@ class Trade(RiskManagement):
             action (str): (`'BMKT'`, `'SMKT'`) for Market orders
                 or (`'BLMT', 'SLMT', 'BSTP', 'SSTP', 'BSTPLMT', 'SSTPLMT'`) for pending orders
             price (float): The price at which to open an order
-            stoplimit (float): A price a pending Limit order is set at when the price reaches the 'price' value (this condition is mandatory).
+            stoplimit (float): A price a pending Limit order is set at 
+                when the price reaches the 'price' value (this condition is mandatory).
                 The pending order is not passed to the trading system until that moment
             id (int): The strategy id or expert Id
             mm (bool): Weither to put stop loss and tp or not
@@ -843,43 +843,32 @@ class Trade(RiskManagement):
             comment (str): The comment for the closing position
             symbol (str): The symbol to trade
             volume (float): The volume (lot) to trade
-            sl (float): The stop loss in points
-            tp (float): The take profit in points
+            sl (float): The stop loss price
+            tp (float): The take profit price
         """
         BUYS = ["BMKT", "BLMT", "BSTP", "BSTPLMT"]
         SELLS = ["SMKT", "SLMT", "SSTP", "SSTPLMT"]
         if action in BUYS:
-            return self.open_buy_position(
-                action=action,
-                price=price,
-                stoplimit=stoplimit,
-                id=id,
-                mm=mm,
-                trail=trail,
-                comment=comment,
-                symbol=symbol,
-                volume=volume,
-                sl=sl,
-                tp=tp,
-            )
+            open_position = self.open_buy_position
         elif action in SELLS:
-            return self.open_sell_position(
-                action=action,
-                price=price,
-                stoplimit=stoplimit,
-                id=id,
-                mm=mm,
-                trail=trail,
-                comment=comment,
-                symbol=symbol,
-                volume=volume,
-                sl=sl,
-                tp=tp,
-            )
+            open_position = self.open_sell_position
         else:
             raise ValueError(
                 f"Invalid action type '{action}', must be {', '.join(BUYS + SELLS)}"
             )
+        return open_position(
+            action=action,
+            price=price,
+            stoplimit=stoplimit,
+            id=id,
+            mm=mm,
+            trail=trail,
+            comment=comment,
+            symbol=symbol,
+            volume=volume,
+            sl=sl,
+            tp=tp,
+        )
 
     @property
     def orders(self):
@@ -1134,9 +1123,9 @@ class Trade(RiskManagement):
         be = self.get_break_even()
         if trail_after_points is not None:
             if isinstance(trail_after_points, int):
-                assert trail_after_points > be, (
-                    "trail_after_points must be greater than break even or set to None"
-                )
+                assert (
+                    trail_after_points > be
+                ), "trail_after_points must be greater than break even or set to None"
             trail_after_points = self._get_trail_after_points(trail_after_points)
         if positions is not None:
             for position in positions:
@@ -1225,7 +1214,8 @@ class Trade(RiskManagement):
         Sets the break-even level for a given trading position.
 
         Args:
-            position (TradePosition): The trading position for which the break-even is to be set. This is the value return by `mt5.positions_get()`.
+            position (TradePosition): The trading position for which the break-even is to be set. 
+                This is the value return by `mt5.positions_get()`.
             be (int): The break-even level in points.
             level (float): The break-even level in price, if set to None , it will be calated automaticaly.
             price (float): The break-even price, if set to None , it will be calated automaticaly.
@@ -1456,7 +1446,8 @@ class Trade(RiskManagement):
         Args:
             ticket (int): Order ticket to modify (e.g TradeOrder.ticket)
             price (float): The price at which to modify the order
-            stoplimit (float): A price a pending Limit order is set at when the price reaches the 'price' value (this condition is mandatory).
+            stoplimit (float): A price a pending Limit order is set at 
+                when the price reaches the 'price' value (this condition is mandatory).
                 The pending order is not passed to the trading system until that moment
             sl (float): The stop loss in points
             tp (float): The take profit in points
@@ -1606,7 +1597,8 @@ class Trade(RiskManagement):
     ):
         """
         Args:
-            order_type (str): Type of orders to close ('all', 'buy_stops', 'sell_stops', 'buy_limits', 'sell_limits', 'buy_stop_limits', 'sell_stop_limits')
+            order_type (str): Type of orders to close 
+                ('all', 'buy_stops', 'sell_stops', 'buy_limits', 'sell_limits', 'buy_stop_limits', 'sell_stop_limits')
             id (int): The unique ID of the Expert or Strategy
             comment (str): Comment for the closing position
         """
@@ -1915,7 +1907,7 @@ def create_trade_instance(
                 if ids is not None and isinstance(ids, (int, float))
                 else params["expert_id"]
                 if "expert_id" in params
-                else None
+                else EXPERT_ID
             )
             params["pchange_sl"] = (
                 pchange_sl[symbol]

bbstrader/models/factors.py

@@ -16,6 +16,101 @@ __all__ = [
     "search_coint_candidate_pairs",
 ]
 
+def _download_and_process_data(source, tickers, start, end, tf, path, **kwargs):
+    """Download and process data for a list of tickers from the specified source."""
+    data_list = []
+    for ticker in tickers:
+        try:
+            if source == "yf":
+                data = yf.download(
+                    ticker,
+                    start=start,
+                    end=end,
+                    progress=False,
+                    multi_level_index=False,
+                )
+                data = data.drop(columns=["Adj Close"], axis=1)
+            elif source == "mt5":
+                start, end = pd.Timestamp(start), pd.Timestamp(end)
+                data = download_historical_data(
+                    symbol=ticker,
+                    timeframe=tf,
+                    date_from=start,
+                    date_to=end,
+                    **{"path": path},
+                )
+                data = data.drop(columns=["adj_close"], axis=1)
+            elif source in ["fmp", "eodhd"]:
+                handler_class = (
+                    FMPDataHandler if source == "fmp" else EODHDataHandler
+                )
+                handler = handler_class(events=None, symbol_list=[ticker], **kwargs)
+                data = handler.data[ticker]
+            else:
+                raise ValueError(f"Invalid source: {source}")
+
+            data = data.reset_index()
+            data = data.rename(columns=str.lower)
+            data["ticker"] = ticker
+            data_list.append(data)
+
+        except Exception as e:
+            print(f"No Data found for {ticker}: {e}")
+            continue
+
+    return pd.concat(data_list)
+
+def _handle_date_range(start, end, window):
+    """Handle start and end date generation."""
+    if start is None or end is None:
+        end = pd.Timestamp(datetime.now()).strftime("%Y-%m-%d")
+        start = (
+            pd.Timestamp(datetime.now())
+            - pd.DateOffset(years=window)
+            + pd.DateOffset(days=1)
+        ).strftime("%Y-%m-%d")
+    return start, end
+
+def _period_search(start, end, securities, candidates, window, npairs):
+    if window < 3 or (pd.Timestamp(end) - pd.Timestamp(start)).days / 365 < 3:
+        raise ValueError(
+            "The date range must be at least two (2) years for period search."
+        )
+    top_pairs = []
+    p_start = pd.Timestamp(end) - pd.DateOffset(years=1)
+    periods = pd.date_range(start=p_start, end=pd.Timestamp(end), freq="BQE")
+    npairs = max(round(npairs / 2), 1)
+    for period in periods:
+        s_start = period - pd.DateOffset(years=2) + pd.DateOffset(days=1)
+        print(f"Searching for pairs in period: {s_start} - {period}")
+        pairs = find_cointegrated_pairs(
+            securities,
+            candidates,
+            n=npairs,
+            start=str(s_start),
+            stop=str(period),
+            coint=True,
+        )
+        pairs["period"] = period
+        top_pairs.append(pairs)
+    top_pairs = pd.concat(top_pairs)
+    if len(top_pairs.columns) <= 1:
+        raise ValueError(
+            "No pairs found in the specified period."
+            "Please adjust the date range or increase the number of pairs."
+        )
+    return top_pairs.head(npairs * 2)
+
+def _process_asset_data(securities, candidates, universe, rolling_window):
+    """Process and select assets from the data."""
+    securities = select_assets(
+        securities, n=universe, rolling_window=rolling_window
+    )
+    candidates = select_assets(
+        candidates, n=universe, rolling_window=rolling_window
+    )
+    return securities, candidates
+
 
 def search_coint_candidate_pairs(
     securities: pd.DataFrame | List[str] = None,
@@ -145,101 +240,6 @@ def search_coint_candidate_pairs(
 
     """
 
-    def _download_and_process_data(source, tickers, start, end, tf, path, **kwargs):
-        """Download and process data for a list of tickers from the specified source."""
-        data_list = []
-        for ticker in tickers:
-            try:
-                if source == "yf":
-                    data = yf.download(
-                        ticker,
-                        start=start,
-                        end=end,
-                        progress=False,
-                        multi_level_index=False,
-                    )
-                    data = data.drop(columns=["Adj Close"], axis=1)
-                elif source == "mt5":
-                    start, end = pd.Timestamp(start), pd.Timestamp(end)
-                    data = download_historical_data(
-                        symbol=ticker,
-                        timeframe=tf,
-                        date_from=start,
-                        date_to=end,
-                        **{"path": path},
-                    )
-                    data = data.drop(columns=["adj_close"], axis=1)
-                elif source in ["fmp", "eodhd"]:
-                    handler_class = (
-                        FMPDataHandler if source == "fmp" else EODHDataHandler
-                    )
-                    handler = handler_class(events=None, symbol_list=[ticker], **kwargs)
-                    data = handler.data[ticker]
-                else:
-                    raise ValueError(f"Invalid source: {source}")
-
-                data = data.reset_index()
-                data = data.rename(columns=str.lower)
-                data["ticker"] = ticker
-                data_list.append(data)
-
-            except Exception as e:
-                print(f"No Data found for {ticker}: {e}")
-                continue
-
-        return pd.concat(data_list)
-
-    def _handle_date_range(start, end, window):
-        """Handle start and end date generation."""
-        if start is None or end is None:
-            end = pd.Timestamp(datetime.now()).strftime("%Y-%m-%d")
-            start = (
-                pd.Timestamp(datetime.now())
-                - pd.DateOffset(years=window)
-                + pd.DateOffset(days=1)
-            ).strftime("%Y-%m-%d")
-        return start, end
-
-    def _period_search(start, end, securities, candidates, npairs=npairs):
-        if window < 3 or (pd.Timestamp(end) - pd.Timestamp(start)).days / 365 < 3:
-            raise ValueError(
-                "The date range must be at least two (2) years for period search."
-            )
-        top_pairs = []
-        p_start = pd.Timestamp(end) - pd.DateOffset(years=1)
-        periods = pd.date_range(start=p_start, end=pd.Timestamp(end), freq="BQE")
-        npairs = max(round(npairs / 2), 1)
-        for period in periods:
-            s_start = period - pd.DateOffset(years=2) + pd.DateOffset(days=1)
-            print(f"Searching for pairs in period: {s_start} - {period}")
-            pairs = find_cointegrated_pairs(
-                securities,
-                candidates,
-                n=npairs,
-                start=str(s_start),
-                stop=str(period),
-                coint=True,
-            )
-            pairs["period"] = period
-            top_pairs.append(pairs)
-        top_pairs = pd.concat(top_pairs)
-        if len(top_pairs.columns) <= 1:
-            raise ValueError(
-                "No pairs found in the specified period."
-                "Please adjust the date range or increase the number of pairs."
-            )
-        return top_pairs.head(npairs * 2)
-
-    def _process_asset_data(securities, candidates, universe, rolling_window):
-        """Process and select assets from the data."""
-        securities = select_assets(
-            securities, n=universe, rolling_window=rolling_window
-        )
-        candidates = select_assets(
-            candidates, n=universe, rolling_window=rolling_window
-        )
-        return securities, candidates
-
     if (
         securities is not None
         and candidates is not None
@@ -255,7 +255,7 @@ def search_coint_candidate_pairs(
         if period_search:
             start = securities.index.get_level_values("date").min()
             end = securities.index.get_level_values("date").max()
-            top_pairs = _period_search(start, end, securities, candidates)
+            top_pairs = _period_search(start, end, securities, candidates, window, npairs)
         else:
             top_pairs = find_cointegrated_pairs(
                 securities, candidates, n=npairs, coint=True
@@ -291,7 +291,7 @@ def search_coint_candidate_pairs(
         )
         if period_search:
             top_pairs = _period_search(
-                start, end, securities_data, candidates_data
+                start, end, securities_data, candidates_data, window, npairs
             ).head(npairs)
         else:
             top_pairs = find_cointegrated_pairs(

bbstrader/models/nlp.py

@@ -506,6 +506,7 @@ class SentimentAnalyzer(object):
             reddit_posts = news.get_reddit_posts(
                 ticker, n_posts=top_news, **{k: kwargs.get(k) for k in rd_params}
             )
+            coindesk_news = news.get_coindesk_news(query=ticker, list_of_str=True)
             fmp_source_news = []
             fmp_news = news.get_fmp_news(kwargs.get("fmp_api"))
             for source in ["articles"]:  # , "releases", asset_type]:
@@ -518,7 +519,7 @@ class SentimentAnalyzer(object):
                     source_news = []
             if any([len(s) > 0 for s in [yahoo_news, google_news]]):
                 sources += 1
-            for source in [reddit_posts, fmp_source_news]:
+            for source in [reddit_posts, fmp_source_news, coindesk_news]:
                 if len(source) > 0:
                     sources += 1
             # Compute sentiment
@@ -531,11 +532,17 @@ class SentimentAnalyzer(object):
             fmp_sentiment = self.analyze_sentiment(
                 fmp_source_news, lexicon=lexicon, textblob=True
             )
+            coindesk_sentiment = self.analyze_sentiment(
+                coindesk_news, lexicon=lexicon, textblob=True
+            )
 
             # Weighted average sentiment score
             if sources != 0:
                 overall_sentiment = (
-                    news_sentiment + reddit_sentiment + fmp_sentiment
+                    news_sentiment
+                    + reddit_sentiment
+                    + fmp_sentiment
+                    + coindesk_sentiment
                 ) / sources
             else:
                 overall_sentiment = 0.0