onesecondtrader 0.33.0__tar.gz → 0.34.0__tar.gz

{onesecondtrader-0.33.0 → onesecondtrader-0.34.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onesecondtrader
-Version: 0.33.0
+Version: 0.34.0
 Summary: The Trading Infrastructure Toolkit for Python. Research, simulate, and deploy algorithmic trading strategies — all in one place.
 License-File: LICENSE
 Author: Nils P. Kujath

{onesecondtrader-0.33.0 → onesecondtrader-0.34.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "onesecondtrader"
-version = "0.33.0"
+version = "0.34.0"
 description = "The Trading Infrastructure Toolkit for Python. Research, simulate, and deploy algorithmic trading strategies — all in one place."
 authors = [
     {name = "Nils P. Kujath",email = "63961429+NilsKujath@users.noreply.github.com"}

{onesecondtrader-0.33.0 → onesecondtrader-0.34.0}/src/onesecondtrader/__init__.py

@@ -4,6 +4,7 @@ __all__ = [
     "BarReceived",
     "BrokerBase",
     "Close",
+    "Datafeed",
     "FillRecord",
     "High",
     "Indicator",
@@ -15,14 +16,16 @@ __all__ = [
     "OrderSide",
     "OrderSubmission",
     "OrderType",
-    "SimpleMovingAverage",
     "SimulatedBroker",
+    "SimulatedDatafeed",
+    "SimpleMovingAverage",
     "SMACrossover",
     "StrategyBase",
     "Volume",
 ]
 
 from onesecondtrader.brokers import BrokerBase, SimulatedBroker
+from onesecondtrader.datafeeds import Datafeed, SimulatedDatafeed
 from onesecondtrader.events import (
     BarProcessed,
     BarReceived,

onesecondtrader-0.34.0/src/onesecondtrader/datafeeds/__init__.py

@@ -0,0 +1,7 @@
+__all__ = [
+    "Datafeed",
+    "SimulatedDatafeed",
+]
+
+from .base import Datafeed
+from .simulated import SimulatedDatafeed

onesecondtrader-0.34.0/src/onesecondtrader/datafeeds/base.py

@@ -0,0 +1,19 @@
+import abc
+
+from onesecondtrader import events, messaging, models
+
+
+class Datafeed(abc.ABC):
+    def __init__(self, event_bus: messaging.EventBus) -> None:
+        self._event_bus = event_bus
+
+    def _publish(self, event: events.EventBase) -> None:
+        self._event_bus.publish(event)
+
+    @abc.abstractmethod
+    def stream(self, symbols: list[str], bar_period: models.BarPeriod) -> None:
+        pass
+
+    @abc.abstractmethod
+    def shutdown(self) -> None:
+        pass

onesecondtrader-0.34.0/src/onesecondtrader/datafeeds/simulated.py

@@ -0,0 +1,99 @@
+import pathlib
+import threading
+
+import pandas as pd
+
+from onesecondtrader import events, messaging, models
+from .base import Datafeed
+
+_RTYPE_MAP = {
+    models.BarPeriod.SECOND: 32,
+    models.BarPeriod.MINUTE: 33,
+    models.BarPeriod.HOUR: 34,
+    models.BarPeriod.DAY: 35,
+}
+
+
+class SimulatedDatafeed(Datafeed):
+    csv_path: str = ""
+
+    def __init__(self, event_bus: messaging.EventBus) -> None:
+        super().__init__(event_bus)
+        self._thread: threading.Thread | None = None
+        self._stop_event = threading.Event()
+
+    def stream(self, symbols: list[str], bar_period: models.BarPeriod) -> None:
+        csv_path = pathlib.Path(self.csv_path)
+        if self._thread and self._thread.is_alive():
+            raise RuntimeError("Already streaming")
+        if not csv_path.exists():
+            raise FileNotFoundError(f"CSV file not found: {csv_path}")
+        if not symbols:
+            raise ValueError("symbols list cannot be empty")
+
+        self._stop_event.clear()
+        self._thread = threading.Thread(
+            target=self._stream,
+            args=(symbols, bar_period),
+            name=self.__class__.__name__,
+            daemon=False,
+        )
+        self._thread.start()
+
+    def wait(self) -> None:
+        if self._thread:
+            self._thread.join()
+
+    def shutdown(self) -> None:
+        self._stop_event.set()
+        if self._thread and self._thread.is_alive():
+            self._thread.join()
+
+    def _stream(self, symbols: list[str], bar_period: models.BarPeriod) -> None:
+        symbols_set = set(symbols)
+        rtype = _RTYPE_MAP[bar_period]
+
+        for chunk in pd.read_csv(
+            self.csv_path,
+            usecols=[
+                "ts_event",
+                "rtype",
+                "open",
+                "high",
+                "low",
+                "close",
+                "volume",
+                "symbol",
+            ],
+            dtype={
+                "ts_event": int,
+                "rtype": int,
+                "open": int,
+                "high": int,
+                "low": int,
+                "close": int,
+                "volume": int,
+                "symbol": str,
+            },
+            chunksize=10_000,
+        ):
+            for row in chunk.itertuples():
+                if self._stop_event.is_set():
+                    return
+
+                if row.symbol not in symbols_set or row.rtype != rtype:
+                    continue
+
+                self._publish(
+                    events.BarReceived(
+                        ts_event=pd.Timestamp(row.ts_event, unit="ns", tz="UTC"),
+                        symbol=row.symbol,
+                        bar_period=bar_period,
+                        open=row.open / 1e9,
+                        high=row.high / 1e9,
+                        low=row.low / 1e9,
+                        close=row.close / 1e9,
+                        volume=row.volume,
+                    )
+                )
+                self._event_bus.wait_until_system_idle()