summaryrefslogtreecommitdiff
path: root/services/strategy-engine/strategies
diff options
context:
space:
mode:
Diffstat (limited to 'services/strategy-engine/strategies')
-rw-r--r--services/strategy-engine/strategies/__init__.py0
-rw-r--r--services/strategy-engine/strategies/base.py17
-rw-r--r--services/strategy-engine/strategies/grid_strategy.py77
-rw-r--r--services/strategy-engine/strategies/rsi_strategy.py77
4 files changed, 171 insertions, 0 deletions
diff --git a/services/strategy-engine/strategies/__init__.py b/services/strategy-engine/strategies/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/services/strategy-engine/strategies/__init__.py
diff --git a/services/strategy-engine/strategies/base.py b/services/strategy-engine/strategies/base.py
new file mode 100644
index 0000000..06101d0
--- /dev/null
+++ b/services/strategy-engine/strategies/base.py
@@ -0,0 +1,17 @@
+from abc import ABC, abstractmethod
+from shared.models import Candle, Signal
+
+
+class BaseStrategy(ABC):
+ name: str = "base"
+
+ @abstractmethod
+ def on_candle(self, candle: Candle) -> Signal | None:
+ pass
+
+ @abstractmethod
+ def configure(self, params: dict) -> None:
+ pass
+
+ def reset(self) -> None:
+ pass
diff --git a/services/strategy-engine/strategies/grid_strategy.py b/services/strategy-engine/strategies/grid_strategy.py
new file mode 100644
index 0000000..f669f09
--- /dev/null
+++ b/services/strategy-engine/strategies/grid_strategy.py
@@ -0,0 +1,77 @@
+from decimal import Decimal
+from typing import Optional
+
+import numpy as np
+
+from shared.models import Candle, Signal, OrderSide
+from strategies.base import BaseStrategy
+
+
+class GridStrategy(BaseStrategy):
+ name: str = "grid"
+
+ def __init__(self) -> None:
+ self._lower_price: float = 0.0
+ self._upper_price: float = 0.0
+ self._grid_count: int = 5
+ self._quantity: Decimal = Decimal("0.01")
+ self._grid_levels: list[float] = []
+ self._last_zone: Optional[int] = None
+
+ def configure(self, params: dict) -> None:
+ self._lower_price = float(params["lower_price"])
+ self._upper_price = float(params["upper_price"])
+ self._grid_count = int(params.get("grid_count", 5))
+ self._quantity = Decimal(str(params.get("quantity", "0.01")))
+ self._grid_levels = list(
+ np.linspace(self._lower_price, self._upper_price, self._grid_count + 1)
+ )
+ self._last_zone = None
+
+ def reset(self) -> None:
+ self._last_zone = None
+
+ def _get_zone(self, price: float) -> int:
+ """Return the grid zone index for a given price.
+
+ Zone 0 is below the lowest level, zone grid_count is above the highest level.
+ Zones 1..grid_count-1 are between levels.
+ """
+ for i, level in enumerate(self._grid_levels):
+ if price < level:
+ return i
+ return len(self._grid_levels)
+
+ def on_candle(self, candle: Candle) -> Signal | None:
+ price = float(candle.close)
+ current_zone = self._get_zone(price)
+
+ if self._last_zone is None:
+ self._last_zone = current_zone
+ return None
+
+ prev_zone = self._last_zone
+ self._last_zone = current_zone
+
+ if current_zone < prev_zone:
+ # Price moved to a lower zone → BUY
+ return Signal(
+ strategy=self.name,
+ symbol=candle.symbol,
+ side=OrderSide.BUY,
+ price=candle.close,
+ quantity=self._quantity,
+ reason=f"Grid: price crossed down from zone {prev_zone} to {current_zone}",
+ )
+ elif current_zone > prev_zone:
+ # Price moved to a higher zone → SELL
+ return Signal(
+ strategy=self.name,
+ symbol=candle.symbol,
+ side=OrderSide.SELL,
+ price=candle.close,
+ quantity=self._quantity,
+ reason=f"Grid: price crossed up from zone {prev_zone} to {current_zone}",
+ )
+
+ return None
diff --git a/services/strategy-engine/strategies/rsi_strategy.py b/services/strategy-engine/strategies/rsi_strategy.py
new file mode 100644
index 0000000..aebbafc
--- /dev/null
+++ b/services/strategy-engine/strategies/rsi_strategy.py
@@ -0,0 +1,77 @@
+from collections import deque
+from decimal import Decimal
+
+import pandas as pd
+
+from shared.models import Candle, Signal, OrderSide
+from strategies.base import BaseStrategy
+
+
+def _compute_rsi(series: pd.Series, period: int) -> float | None:
+ """Compute RSI using Wilder's smoothing (EMA-based)."""
+ if len(series) < period + 1:
+ return None
+ delta = series.diff()
+ gain = delta.clip(lower=0)
+ loss = -delta.clip(upper=0)
+ avg_gain = gain.ewm(com=period - 1, min_periods=period).mean()
+ avg_loss = loss.ewm(com=period - 1, min_periods=period).mean()
+ rs = avg_gain / avg_loss.replace(0, float("nan"))
+ rsi = 100 - (100 / (1 + rs))
+ value = rsi.iloc[-1]
+ if pd.isna(value):
+ return None
+ return float(value)
+
+
+class RsiStrategy(BaseStrategy):
+ name: str = "rsi"
+
+ def __init__(self) -> None:
+ self._closes: deque[float] = deque(maxlen=200)
+ self._period: int = 14
+ self._oversold: float = 30.0
+ self._overbought: float = 70.0
+ self._quantity: Decimal = Decimal("0.01")
+
+ def configure(self, params: dict) -> None:
+ self._period = int(params.get("period", 14))
+ self._oversold = float(params.get("oversold", 30))
+ self._overbought = float(params.get("overbought", 70))
+ self._quantity = Decimal(str(params.get("quantity", "0.01")))
+
+ def reset(self) -> None:
+ self._closes.clear()
+
+ def on_candle(self, candle: Candle) -> Signal | None:
+ self._closes.append(float(candle.close))
+
+ if len(self._closes) < self._period + 1:
+ return None
+
+ series = pd.Series(list(self._closes))
+ rsi_value = _compute_rsi(series, self._period)
+
+ if rsi_value is None:
+ return None
+
+ if rsi_value < self._oversold:
+ return Signal(
+ strategy=self.name,
+ symbol=candle.symbol,
+ side=OrderSide.BUY,
+ price=candle.close,
+ quantity=self._quantity,
+ reason=f"RSI {rsi_value:.2f} below oversold threshold {self._oversold}",
+ )
+ elif rsi_value > self._overbought:
+ return Signal(
+ strategy=self.name,
+ symbol=candle.symbol,
+ side=OrderSide.SELL,
+ price=candle.close,
+ quantity=self._quantity,
+ reason=f"RSI {rsi_value:.2f} above overbought threshold {self._overbought}",
+ )
+
+ return None