Fix flake8 warnings

Author: MDUYN

investing_algorithm_framework/app/context.py

@@ -1068,7 +1068,8 @@ def add_stop_loss(
         Example of trailing stop loss:
             * You buy BTC at $40,000.
             * You set a TSL of 5%, setting the sell price at $38,000.
-            * BTC price increases to $42,000 → New TSL level at $39,900 (42,000 - 5%).
+            * BTC price increases to $42,000 → New TSL level
+                at $39,900 (42,000 - 5%).
             * BTC price drops to $39,900 → SL level reached, trade closes.
 
         Args:

investing_algorithm_framework/app/strategy.py

@@ -215,14 +215,6 @@ def strategy_identifier(self):
 
         return self.worker_id
 
-    @property
-    def context(self):
-        return self._context
-
-    @context.setter
-    def context(self, context):
-        self._context = context
-
     def add_trace(
         self,
         symbol: str,

investing_algorithm_framework/create_app.py

@@ -12,7 +12,7 @@ def create_app(
     config: dict = None,
     state_handler=None,
     web: bool = False,
-    name = None
+    name=None
 ) -> App:
     """
     Factory method to create an app instance.

investing_algorithm_framework/domain/models/trade/trade.py

@@ -6,7 +6,8 @@
     TradeStatus
 from investing_algorithm_framework.domain.models.trade.trade_stop_loss import \
     TradeStopLoss
-from investing_algorithm_framework.domain.models.trade.trade_take_profit import TradeTakeProfit
+from investing_algorithm_framework.domain.models.trade\
+    .trade_take_profit import TradeTakeProfit
 
 
 class Trade(BaseModel):
@@ -272,7 +273,6 @@ def is_trailing_take_profit_triggered(self):
 
         return self.last_reported_price >= take_profit_price
 
-
     def to_dict(self, datetime_format=None):
 
         if datetime_format is not None:

investing_algorithm_framework/domain/models/trade/trade_stop_loss.py

@@ -14,9 +14,10 @@ class TradeStopLoss(BaseModel):
             trailing or fixed
         percentage: float - the stop loss percentage
         sell_percentage: float - the percentage of the trade to sell when the
-            take profit is hit. Default is 100% of the trade. If the take profit percentage is lower than 100% a check must be made that
-            the combined sell percentage of all take profits is less or
-            equal than 100%.
+            take profit is hit. Default is 100% of the trade. If the
+            take profit percentage is lower than 100% a check must
+            be made that the combined sell percentage of all
+            take profits is less or equal than 100%.
         sell_amount: float - the amount to sell when the stop loss triggers
         sold_amount: float - the amount that has been sold
         high_water_mark: float - the highest price of the trade
@@ -30,7 +31,8 @@ class TradeStopLoss(BaseModel):
         But if the price keeps falling to $95, the stop loss triggers,
             and you exit with a $5 loss.
 
-    if trade_risk_type is trailing, the stop loss price is calculated as follows:
+    if trade_risk_type is trailing, the stop loss price is
+    calculated as follows:
         You buy a stock at $100.
         You set a 5% trailing stop loss, meaning you will sell if
             the price drops 5% from its peak at $96
@@ -70,10 +72,12 @@ def __init__(
 
     def update_with_last_reported_price(self, current_price: float):
         """
-        Function to update the take profit price based on the last reported price.
-        The take profit price is only updated when the trade risk type is trailing.
-        The take profit price is updated based on the current price and the percentage
-        of the take profit.
+        Function to update the take profit price based on the last
+        reported price.
+        The take profit price is only updated when the trade risk
+        type is trailing.
+        The take profit price is updated based on the current price
+        and the percentage of the take profit.
 
         Args:
             current_price: float - the last reported price of the trade

investing_algorithm_framework/domain/models/trade/trade_take_profit.py

@@ -14,11 +14,13 @@ class TradeTakeProfit(BaseModel):
             trailing or fixed
         percentage: float - the take profit percentage
         sell_percentage: float - the percentage of the trade to sell when the
-            take profit is hit. Default is 100% of the trade. If the take profit percentage is lower than 100% a check must be made that
-            the combined sell percentage of all take profits is less or
-            equal than 100%.
+            take profit is hit. Default is 100% of the trade.
+            If the take profit percentage is lower than 100% a check
+            must be made that the combined sell percentage of
+            all take profits is less or equal than 100%.
 
-     if trade_risk_type is fixed, the take profit price is calculated as follows:
+    if trade_risk_type is fixed, the take profit price is
+    calculated as follows:
         You buy a stock at $100.
         You set a 5% take profit, meaning you will sell if the price
             rises to $105.
@@ -27,18 +29,21 @@ class TradeTakeProfit(BaseModel):
         But if the price keeps falling below $105, the take profit is not
             triggered.
 
-    if trade_risk_type is trailing, the take profit price is calculated as follows:
+    if trade_risk_type is trailing, the take profit price is
+    calculated as follows:
         You buy a stock at $100.
         You set a 5% trailing take profit, the moment the price rises
             5% the initial take profit mark will be set. This means you
             will set the take_profit_price initially at none and
             only if the price hits $105, you will set the
             take_profit_price to $105.
         if the price drops below $105, the take profit is triggered.
-        If the price rises to $120, the take profit adjusts to $114 (5% below $120).
-        If the price falls to $114, the position is closed, securing a $14 profit.
-        But if the price keeps rising to $150, the take profit moves up to $142.50.
-
+        If the price rises to $120, the take profit adjusts to
+        $114 (5% below $120).
+        If the price falls to $114, the position is closed,
+        securing a $14 profit.
+        But if the price keeps rising to $150, the take profit
+        moves up to $142.50.
     """
 
     def __init__(
@@ -67,10 +72,12 @@ def __init__(
 
     def update_with_last_reported_price(self, current_price: float):
         """
-        Function to update the take profit price based on the last reported price.
-        The take profit price is only updated when the trade risk type is trailing.
-        The take profit price is updated based on the current price and the percentage
-        of the take profit.
+        Function to update the take profit price based on
+            the last reported price.
+        The take profit price is only updated when the
+            trade risk type is trailing.
+        The take profit price is updated based on the
+            current price and the percentage of the take profit.
 
         Args:
             current_price: float - the last reported price of the trade
@@ -110,7 +117,7 @@ def update_with_last_reported_price(self, current_price: float):
 
         return
 
-    def has_triggered(self, current_price: float=None) -> bool:
+    def has_triggered(self, current_price: float = None) -> bool:
 
         if TradeRiskType.FIXED.equals(self.trade_risk_type):
             # Check if the current price is less than the high water mark

investing_algorithm_framework/infrastructure/models/market_data_sources/ccxt.py

@@ -165,7 +165,9 @@ def _precompute_sliding_windows(self):
     def load_data(self):
         file_path = self._create_file_path()
         self.data = polars.read_csv(
-            file_path, schema_overrides={"Datetime": polars.Datetime}, low_memory=True
+            file_path,
+            schema_overrides={"Datetime": polars.Datetime},
+            low_memory=True
         )  # Faster parsing
         first_row = self.data.head(1)
         last_row = self.data.tail(1)

investing_algorithm_framework/infrastructure/models/market_data_sources/csv.py

@@ -66,7 +66,9 @@ def csv_file_path(self):
 
     def _load_data(self, file_path):
         return polars.read_csv(
-            file_path, schema_overrides={"Datetime": polars.Datetime}, low_memory=True
+            file_path,
+            schema_overrides={"Datetime": polars.Datetime},
+            low_memory=True
         ).with_columns(
             polars.col("Datetime").cast(
                 polars.Datetime(time_unit="ms", time_zone="UTC")

investing_algorithm_framework/infrastructure/models/order/order_metadata.py

@@ -21,15 +21,14 @@ class SQLOrderMetadata(SQLBaseModel, SQLAlchemyModelExtension):
     amount = Column(Float)
     amount_pending = Column(Float)
 
-
     def __init__(
         self,
         order_id,
         amount,
         amount_pending,
-        trade_id = None,
-        stop_loss_id = None,
-        take_profit_id = None,
+        trade_id=None,
+        stop_loss_id=None,
+        take_profit_id=None,
     ):
         self.order_id = order_id
         self.trade_id = trade_id

investing_algorithm_framework/infrastructure/models/trades/trade.py

@@ -1,4 +1,4 @@
-from sqlalchemy import Column, Integer, String, DateTime, Float, Boolean
+from sqlalchemy import Column, Integer, String, DateTime, Float
 from sqlalchemy.orm import relationship
 
 from investing_algorithm_framework.domain import Trade, TradeStatus