onesecondtrader 0.9.0__py3-none-any.whl → 0.10.1__py3-none-any.whl

onesecondtrader/core/models.py

@@ -131,3 +131,29 @@ class OrderRejectionReason(enum.Enum):
 
     UNKNOWN = enum.auto()
     NEGATIVE_QUANTITY = enum.auto()
+
+
+class TimeFrame(enum.Enum):
+    """
+    Enum for timeframes.
+
+    **Attributes:**
+
+    | Enum | Value | Description |
+    |------|-------|-------------|
+    | `SECOND` | `enum.auto()` | 1 second |
+    | `MINUTE` | `enum.auto()` | 1 minute |
+    | `HOUR` | `enum.auto()` | 1 hour |
+    | `DAY` | `enum.auto()` | 1 day |
+    | `WEEK` | `enum.auto()` | 1 week |
+    | `MONTH` | `enum.auto()` | 1 month |
+    | `YEAR` | `enum.auto()` | 1 year
+    """
+
+    SECOND = enum.auto()
+    MINUTE = enum.auto()
+    HOUR = enum.auto()
+    DAY = enum.auto()
+    WEEK = enum.auto()
+    MONTH = enum.auto()
+    YEAR = enum.auto()

onesecondtrader/datafeeds/base_datafeed.py

@@ -0,0 +1,255 @@
+"""
+This module provides the base class for all datafeeds.
+"""
+
+import abc
+import threading
+from onesecondtrader.messaging import eventbus
+from onesecondtrader.core import models
+from onesecondtrader.monitoring import console
+
+
+class BaseDatafeed(abc.ABC):
+    """
+    Base class for all datafeeds.
+    """
+
+    def __init__(self, event_bus: eventbus.EventBus) -> None:
+        """
+        Initialize the datafeed with the provided event bus.
+
+        Args:
+            event_bus: The event bus to publish events to.
+
+        Attributes:
+            self.event_bus: The event bus to publish events to.
+            self._lock: Lock for thread safety.
+            self._is_connected: Flag indicating if the datafeed is connected.
+            self._watched_symbols: Set of symbols currently being watched.
+        """
+        self.event_bus = event_bus
+        self._lock = threading.Lock()
+        self._is_connected = False
+        self._watched_symbols: set[tuple[str, models.TimeFrame]] = set()
+
+    def connect(self) -> bool:
+        """
+        Connect to the datafeed.
+
+        Returns:
+            bool: True if connection successful, False otherwise.
+        """
+        with self._lock:
+            if self._is_connected:
+                console.logger.warning(f"{self.__class__.__name__} already connected")
+                return True
+
+            console.logger.info(f"Connecting to {self.__class__.__name__}...")
+            try:
+                success = self._connect()
+                if success:
+                    self._is_connected = True
+                    console.logger.info(
+                        f"Successfully connected to {self.__class__.__name__}"
+                    )
+                    return True
+                else:
+                    console.logger.error(
+                        f"Failed to connect to {self.__class__.__name__}"
+                    )
+                    return False
+            except Exception as e:
+                console.logger.error(
+                    f"Connection failed for {self.__class__.__name__}: {e}"
+                )
+                return False
+
+    @abc.abstractmethod
+    def _connect(self) -> bool:
+        """
+        Implement connection logic for the specific datafeed.
+
+        Returns:
+            bool: True if connection successful, False otherwise.
+        """
+        pass
+
+    def disconnect(self) -> bool:
+        """
+        Disconnect from the datafeed.
+
+        Returns:
+            bool: True if disconnection successful, False otherwise.
+        """
+        with self._lock:
+            if not self._is_connected:
+                console.logger.warning(
+                    f"{self.__class__.__name__} already disconnected"
+                )
+                return True
+
+            console.logger.info(f"Disconnecting from {self.__class__.__name__}...")
+            try:
+                success = self._disconnect()
+                if success:
+                    self._is_connected = False
+                    self._watched_symbols.clear()
+                    console.logger.info(
+                        f"Successfully disconnected from {self.__class__.__name__}"
+                    )
+                    return True
+                else:
+                    console.logger.error(
+                        f"Failed to disconnect from {self.__class__.__name__}"
+                    )
+                    return False
+            except Exception as e:
+                console.logger.error(
+                    f"Disconnection failed for {self.__class__.__name__}: {e}"
+                )
+                self._is_connected = False
+                self._watched_symbols.clear()
+                return False
+
+    @abc.abstractmethod
+    def _disconnect(self) -> bool:
+        """
+        Implement disconnection logic for the specific datafeed.
+
+        Returns:
+            bool: True if disconnection successful, False otherwise.
+        """
+        pass
+
+    def watch(self, symbols: list[tuple[str, models.TimeFrame]]) -> bool:
+        """
+        Start watching market data for the specified symbols and timeframes.
+
+        Args:
+            symbols: List of (symbol, timeframe) tuples to start watching.
+
+        Returns:
+            bool: True if watching started successfully, False otherwise.
+        """
+        if not symbols:
+            console.logger.warning("No symbols provided for watching")
+            return True
+
+        with self._lock:
+            if not self._is_connected:
+                console.logger.error("Cannot start watching: datafeed not connected")
+                return False
+
+            new_symbols = set(symbols) - self._watched_symbols
+            if not new_symbols:
+                console.logger.info("All requested symbols are already being watched")
+                return True
+
+            try:
+                success = self._watch(list(new_symbols))
+                if success:
+                    self._watched_symbols.update(new_symbols)
+                    console.logger.info(
+                        f"Successfully started watching {len(new_symbols)} symbols"
+                    )
+                    return True
+                else:
+                    console.logger.error("Failed to start watching symbols")
+                    return False
+            except Exception as e:
+                console.logger.error(f"Exception while starting watching: {e}")
+                return False
+
+    @abc.abstractmethod
+    def _watch(self, symbols: list[tuple[str, models.TimeFrame]]) -> bool:
+        """
+        Implement watching startup logic for the specific datafeed.
+
+        Args:
+            symbols: List of (symbol, timeframe) tuples to start watching.
+                    These are guaranteed to be new symbols not already being watched.
+
+        Returns:
+            bool: True if watching started successfully, False otherwise.
+        """
+        pass
+
+    def unwatch(self, symbols: list[tuple[str, models.TimeFrame]]) -> bool:
+        """
+        Stop watching market data for the specified symbols and timeframes.
+
+        Args:
+            symbols: List of (symbol, timeframe) tuples to stop watching.
+
+        Returns:
+            bool: True if unwatching stopped successfully, False otherwise.
+        """
+        if not symbols:
+            console.logger.warning("No symbols provided for unwatching")
+            return True
+
+        with self._lock:
+            if not self._is_connected:
+                console.logger.warning(
+                    "Datafeed not connected, but removing symbols from tracking"
+                )
+                self._watched_symbols.difference_update(symbols)
+                return True
+
+            symbols_to_stop = set(symbols) & self._watched_symbols
+            if not symbols_to_stop:
+                console.logger.info(
+                    "None of the requested symbols are currently being watched"
+                )
+                return True
+
+            console.logger.info(f"Unwatching {len(symbols_to_stop)} symbols")
+            try:
+                success = self._unwatch(list(symbols_to_stop))
+                if success:
+                    self._watched_symbols.difference_update(symbols_to_stop)
+                    console.logger.info(
+                        f"Successfully unwatched {len(symbols_to_stop)} symbols"
+                    )
+                    return True
+                else:
+                    console.logger.error("Failed to unwatch symbols")
+                    return False
+            except Exception as e:
+                console.logger.error(f"Exception while unwatching: {e}")
+                self._watched_symbols.difference_update(symbols_to_stop)
+                return False
+
+    @abc.abstractmethod
+    def _unwatch(self, symbols: list[tuple[str, models.TimeFrame]]) -> bool:
+        """
+        Implement unwatching logic for the specific datafeed.
+
+        Args:
+            symbols: List of (symbol, timeframe) tuples to stop watching.
+                    These are guaranteed to be symbols currently being watched.
+
+        Returns:
+            bool: True if unwatching stopped successfully, False otherwise.
+        """
+        pass
+
+    def is_connected(self) -> bool:
+        """
+        Check if the datafeed is currently connected.
+
+        Returns:
+            bool: True if connected, False otherwise.
+        """
+        with self._lock:
+            return self._is_connected
+
+    def get_watched_symbols(self) -> set[tuple[str, models.TimeFrame]]:
+        """
+        Get the set of currently watched symbols and timeframes.
+
+        Returns:
+            set: Set of (symbol, timeframe) tuples currently being watched.
+        """
+        with self._lock:
+            return self._watched_symbols.copy()

{onesecondtrader-0.9.0.dist-info → onesecondtrader-0.10.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: onesecondtrader
-Version: 0.9.0
+Version: 0.10.1
 Summary: The Trading Infrastructure Toolkit for Python. Research, simulate, and deploy algorithmic trading strategies — all in one place.
 Author: Nils P. Kujath
 Author-email: 63961429+NilsKujath@users.noreply.github.com

{onesecondtrader-0.9.0.dist-info → onesecondtrader-0.10.1.dist-info}/RECORD

@@ -1,7 +1,9 @@
 onesecondtrader/__init__.py,sha256=TNqlT20sH46-J7F6giBxwWYG1-wFZZt7toDbZeQK6KQ,210
 onesecondtrader/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-onesecondtrader/core/models.py,sha256=Pw3og5b2mzcoPpMfv3ZdKPvUvhLvh_eT68WJSlIpRyo,3325
+onesecondtrader/core/models.py,sha256=gvspEZp7pISZgpHeaKNQfvNJgwZY7DnS_kWpoGwXC-c,3940
 onesecondtrader/core/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+onesecondtrader/datafeeds/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+onesecondtrader/datafeeds/base_datafeed.py,sha256=cHt8bdA9zON6CJLjz7blgNcs4xu4-iWxM15s3Gu3FO0,8774
 onesecondtrader/messaging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 onesecondtrader/messaging/eventbus.py,sha256=R2K85INeYVwJ1tMOybC3WpRraK0ZKVe8WehCbAzzznU,19359
 onesecondtrader/messaging/events.py,sha256=eaWXQQIUnRNOR-9n5-6lyLbZ6bUtzjD4GI567U_vh4g,23625
@@ -9,7 +11,7 @@ onesecondtrader/monitoring/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 onesecondtrader/monitoring/console.py,sha256=1mrojXkyL4ro7ebkvDMGNQiCL-93WEylRuwnfmEKzVs,299
 onesecondtrader/monitoring/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 onesecondtrader/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-onesecondtrader-0.9.0.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-onesecondtrader-0.9.0.dist-info/METADATA,sha256=zc6MwJQZZr8ULUHsipjJ7zAkHnkf03CxD6wpDYvlYF8,9518
-onesecondtrader-0.9.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-onesecondtrader-0.9.0.dist-info/RECORD,,
+onesecondtrader-0.10.1.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+onesecondtrader-0.10.1.dist-info/METADATA,sha256=tGtxjkcRkaC39mi4glsawyMVYHmuBDzpsZQwR4-k3nY,9519
+onesecondtrader-0.10.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+onesecondtrader-0.10.1.dist-info/RECORD,,