added check if market is open at this moment in utils (#288)

* added check if market is open at this moment

* added test

* works for half days now.

Author: Quenos

tastytrade/utils.py

@@ -40,6 +40,31 @@ def today_in_new_york() -> date:
     return now_in_new_york().date()
 
 
+def is_market_open_now() -> bool:
+    """
+    Check if the market is currently open.
+
+    Returns True if:
+    - The market is open on the current date (not a holiday/weekend)
+    - The current Eastern Time hour is between [ 9:30 AM, 4:00 PM )
+    - or on half days between [ 9:30 AM, 1:00 PM )
+
+    Returns:
+        bool: True if market is open now, False otherwise
+    """
+    today = today_in_new_york()
+    sched = NYSE.schedule(start_date=today, end_date=today)
+    if sched.empty:
+        # Closed full day (weekend or holiday)
+        return False
+
+    # Use iloc[0] since schedule has only one row for a single day
+    market_open = sched.iloc[0]["market_open"]
+    market_close = sched.iloc[0]["market_close"]
+
+    return market_open <= now_in_new_york() < market_close
+
+
 def is_market_open_on(day: date | None = None) -> bool:
     """
     Returns whether the market was/is/will be open at ANY point

tests/test_utils.py

@@ -1,4 +1,6 @@
-from datetime import date
+from datetime import date, datetime
+from unittest.mock import patch
+from zoneinfo import ZoneInfo
 
 from tastytrade.utils import (
     get_future_fx_monthly,
@@ -9,9 +11,12 @@
     get_future_treasury_monthly,
     get_tasty_monthly,
     get_third_friday,
+    is_market_open_now,
     today_in_new_york,
 )
 
+TZ = ZoneInfo("US/Eastern")
+
 
 def test_get_third_friday():
     assert get_third_friday(date(2024, 3, 2)) == date(2024, 3, 15)
@@ -158,3 +163,68 @@ def test_get_future_index_monthly():
     ]
     for exp in exps:
         assert get_future_index_monthly(exp) == exp
+
+
+def test_is_market_open_now():
+    # Use a known Tuesday (March 12, 2024) for weekday tests
+    tuesday = datetime(2024, 3, 12, 10, 0, 0, tzinfo=TZ)
+
+    # Test market open during trading hours (10:00 AM on a weekday)
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = tuesday
+        assert is_market_open_now() is True
+
+    # Test market closed before opening (8:00 AM on a weekday)
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 3, 12, 8, 0, 0, tzinfo=TZ)
+        assert is_market_open_now() is False
+
+    # Test market closed after closing (5:00 PM on a weekday)
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 3, 12, 17, 0, 0, tzinfo=TZ)
+        assert is_market_open_now() is False
+
+    # Test market closed on weekend (Saturday, March 16, 2024)
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 3, 16, 12, 0, 0, tzinfo=TZ)
+        assert is_market_open_now() is False
+
+    # Test edge case: exactly 9:30 AM (market opens)
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 3, 12, 9, 30, 0, tzinfo=TZ)
+        assert is_market_open_now() is True
+
+    # Test edge case: just before 9:30 AM (9:29 AM)
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 3, 12, 9, 29, 0, tzinfo=TZ)
+        assert is_market_open_now() is False
+
+    # Test edge case: exactly 4:00 PM (market closes, should be False)
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 3, 12, 16, 0, 0, tzinfo=TZ)
+        assert is_market_open_now() is False
+
+    # Test edge case: just before 4:00 PM (3:59 PM)
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 3, 12, 15, 59, 0, tzinfo=TZ)
+        assert is_market_open_now() is True
+
+    # Test edge case: Black Friday just before market open
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 11, 29, 9, 29, 0, tzinfo=TZ)
+        assert is_market_open_now() is False
+
+    # Test edge case: Black Friday just after market open
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 11, 29, 9, 30, 0, tzinfo=TZ)
+        assert is_market_open_now() is True
+
+    # Test edge case: Black Friday just before market close half day
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 11, 29, 12, 59, 0, tzinfo=TZ)
+        assert is_market_open_now() is True
+
+    # Test edge case: Black Friday just after market close half day
+    with patch("tastytrade.utils.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2024, 11, 29, 13, 0, 0, tzinfo=TZ)
+        assert is_market_open_now() is False