kuhl-haus-mdp 0.1.7__tar.gz → 0.1.8__tar.gz

{kuhl_haus_mdp-0.1.7 → kuhl_haus_mdp-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kuhl-haus-mdp
-Version: 0.1.7
+Version: 0.1.8
 Summary: Market data processing pipeline for stock market scanner
 Author-Email: Tom Pounders <git@oldschool.engineer>
 License: The MIT License (MIT)

{kuhl_haus_mdp-0.1.7 → kuhl_haus_mdp-0.1.8}/pyproject.toml

@@ -30,7 +30,7 @@ dependencies = [
     "uvicorn[standard]",
     "websockets",
 ]
-version = "0.1.7"
+version = "0.1.8"
 
 [project.license]
 file = "LICENSE.txt"

{kuhl_haus_mdp-0.1.7 → kuhl_haus_mdp-0.1.8}/src/kuhl_haus/mdp/analyzers/top_stocks.py

@@ -126,6 +126,7 @@ class TopStocksAnalyzer(Analyzer):
             prev_day_close = cached_data["prev_day_close"]
             prev_day_volume = cached_data["prev_day_volume"]
             prev_day_vwap = cached_data["prev_day_vwap"]
+            free_float = cached_data["free_float"]
         else:
             # Get snapshot for previous day's data
             retry_count = 0
@@ -140,12 +141,10 @@ class TopStocksAnalyzer(Analyzer):
                     prev_day_volume = snapshot.prev_day.volume
                     prev_day_vwap = snapshot.prev_day.vwap
                     break
-                except BadResponse as e:
-                    self.logger.error(f"Error getting snapshot for {event.symbol}: {repr(e)}", exc_info=e, stack_info=True)
+                except Exception:
                     retry_count += 1
             if retry_count == max_tries and prev_day_close == 0:
                 self.logger.error(f"Failed to get snapshot for {event.symbol} after {max_tries} tries.")
-                return
 
             # Get average volume
             retry_count = 0
@@ -155,12 +154,23 @@ class TopStocksAnalyzer(Analyzer):
                 try:
                     avg_volume = await self.cache.get_avg_volume(event.symbol)
                     break
-                except (BadResponse, ZeroDivisionError) as e:
-                    self.logger.error(f"Error getting average volume for {event.symbol}: {repr(e)}", exc_info=e, stack_info=True)
+                except Exception:
                     retry_count += 1
             if retry_count == max_tries and avg_volume == 0:
                 self.logger.error(f"Failed to get average volume for {event.symbol} after {max_tries} tries.")
-                return
+
+            # Get free float - this uses an experimental API
+            retry_count = 0
+            max_tries = 2
+            free_float = 0
+            while retry_count < max_tries:
+                try:
+                    free_float = await self.cache.get_free_float(event.symbol)
+                    break
+                except Exception:
+                    retry_count += 1
+            if retry_count == max_tries and free_float == 0:
+                self.logger.error(f"Failed to get free float for {event.symbol} after {max_tries} tries.")
 
         # Calculate relative volume
         if avg_volume == 0:
@@ -196,6 +206,7 @@ class TopStocksAnalyzer(Analyzer):
         self.cache_item.symbol_data_cache[event.symbol] = {
             "symbol": event.symbol,
             "volume": event.volume,
+            "free_float": free_float,
             "accumulated_volume": event.accumulated_volume,
             "relative_volume": relative_volume,
             "official_open_price": event.official_open_price,

{kuhl_haus_mdp-0.1.7 → kuhl_haus_mdp-0.1.8}/src/kuhl_haus/mdp/components/market_data_cache.py

@@ -57,77 +57,6 @@ class MarketDataCache:
                 ticker=ticker
             )
             self.logger.info(f"Snapshot result: {snapshot}")
-            # data = {
-            #     "day": {
-            #         "open": snapshot.day.open,
-            #         "high": snapshot.day.high,
-            #         "low": snapshot.day.low,
-            #         "close": snapshot.day.close,
-            #         "volume": snapshot.day.volume,
-            #         "vwap": snapshot.day.vwap,
-            #         "timestamp": snapshot.day.timestamp,
-            #         "transactions": snapshot.day.transactions,
-            #         "otc": snapshot.day.otc,
-            #     },
-            #     "last_quote": {
-            #         "ticker": snapshot.last_quote.ticker,
-            #         "trf_timestamp": snapshot.last_quote.trf_timestamp,
-            #         "sequence_number": snapshot.last_quote.sequence_number,
-            #         "sip_timestamp": snapshot.last_quote.sip_timestamp,
-            #         "participant_timestamp": snapshot.last_quote.participant_timestamp,
-            #         "ask_price": snapshot.last_quote.ask_price,
-            #         "ask_size": snapshot.last_quote.ask_size,
-            #         "ask_exchange": snapshot.last_quote.ask_exchange,
-            #         "conditions": snapshot.last_quote.conditions,
-            #         "indicators": snapshot.last_quote.indicators,
-            #         "bid_price": snapshot.last_quote.bid_price,
-            #         "bid_size": snapshot.last_quote.bid_size,
-            #         "bid_exchange": snapshot.last_quote.bid_exchange,
-            #         "tape": snapshot.last_quote.tape,
-            #     },
-            #     "last_trade": {
-            #         "ticker": snapshot.last_trade.ticker,
-            #         "trf_timestamp": snapshot.last_trade.trf_timestamp,
-            #         "sequence_number": snapshot.last_trade.sequence_number,
-            #         "sip_timestamp": snapshot.last_trade.sip_timestamp,
-            #         "participant_timestamp": snapshot.last_trade.participant_timestamp,
-            #         "conditions": snapshot.last_trade.conditions,
-            #         "correction": snapshot.last_trade.correction,
-            #         "id": snapshot.last_trade.id,
-            #         "price": snapshot.last_trade.price,
-            #         "trf_id": snapshot.last_trade.trf_id,
-            #         "size": snapshot.last_trade.size,
-            #         "exchange": snapshot.last_trade.exchange,
-            #         "tape": snapshot.last_trade.tape,
-            #     },
-            #     "min": {
-            #         "accumulated_volume": snapshot.min.accumulated_volume,
-            #         "open": snapshot.min.open,
-            #         "high": snapshot.min.high,
-            #         "low": snapshot.min.low,
-            #         "close": snapshot.min.close,
-            #         "volume": snapshot.min.volume,
-            #         "vwap": snapshot.min.vwap,
-            #         "otc": snapshot.min.otc,
-            #         "timestamp": snapshot.min.timestamp,
-            #         "transactions": snapshot.min.transactions,
-            #     },
-            #     "prev_day": {
-            #         "open": snapshot.prev_day.open,
-            #         "high": snapshot.prev_day.high,
-            #         "low": snapshot.prev_day.low,
-            #         "close": snapshot.prev_day.close,
-            #         "volume": snapshot.prev_day.volume,
-            #         "vwap": snapshot.prev_day.vwap,
-            #         "timestamp": snapshot.prev_day.timestamp,
-            #         "transactions": snapshot.prev_day.transactions,
-            #         "otc": snapshot.prev_day.otc,
-            #     },
-            #     "ticker": snapshot.ticker,
-            #     "todaysChange": snapshot.todays_change,
-            #     "todaysChangePerc": snapshot.todays_change_percent,
-            #     "updated": snapshot.updated,
-            # }
             data = ticker_snapshot_to_dict(snapshot)
             await self.cache_data(
                 data=data,

{kuhl_haus_mdp-0.1.7 → kuhl_haus_mdp-0.1.8}/src/kuhl_haus/mdp/helpers/utils.py

@@ -1,5 +1,6 @@
 import logging
 import os
+from typing import Dict, Any
 
 from massive.rest.models import TickerSnapshot
 
@@ -38,7 +39,7 @@ def get_massive_api_key():
     return api_key
 
 
-def ticker_snapshot_to_dict(snapshot: TickerSnapshot) -> dict:
+def ticker_snapshot_to_dict(snapshot: TickerSnapshot) -> Dict[str, Any]:
     """
     Convert a TickerSnapshot instance into a JSON-serializable dictionary.
 

{kuhl_haus_mdp-0.1.7 → kuhl_haus_mdp-0.1.8}/src/kuhl_haus/mdp/models/market_data_cache_keys.py

@@ -15,10 +15,10 @@ class MarketDataCacheKeys(Enum):
     UNKNOWN = 'unknown'
 
     # MARKET DATA CACHE
-    DAILY_AGGREGATES = 'aggregate:daily'
-    TICKER_SNAPSHOTS = 'snapshots'
-    TICKER_AVG_VOLUME = 'avg_volume'
-    TICKER_FREE_FLOAT = 'free_float'
+    DAILY_AGGREGATES = 'mdc:aggregate:daily'
+    TICKER_SNAPSHOTS = 'mdc:snapshots'
+    TICKER_AVG_VOLUME = 'mdc:avg_volume'
+    TICKER_FREE_FLOAT = 'mdc:free_float'
 
     # MARKET DATA PROCESSOR CACHE
     TOP_TRADES_SCANNER = f'cache:{MarketDataScannerNames.TOP_TRADES.value}'

{kuhl_haus_mdp-0.1.7 → kuhl_haus_mdp-0.1.8}/src/kuhl_haus/mdp/models/top_stocks_cache_item.py

@@ -45,6 +45,7 @@ class TopStocksCacheItem:
                 ret.append({
                     "symbol": ticker,
                     "volume": self.symbol_data_cache[ticker]["volume"],
+                    "free_float": self.symbol_data_cache[ticker]["free_float"],
                     "accumulated_volume": self.symbol_data_cache[ticker]["accumulated_volume"],
                     "relative_volume": self.symbol_data_cache[ticker]["relative_volume"],
                     "official_open_price": self.symbol_data_cache[ticker]["official_open_price"],
@@ -81,6 +82,7 @@ class TopStocksCacheItem:
                 ret.append({
                     "symbol": ticker,
                     "volume": self.symbol_data_cache[ticker]["volume"],
+                    "free_float": self.symbol_data_cache[ticker]["free_float"],
                     "accumulated_volume": self.symbol_data_cache[ticker]["accumulated_volume"],
                     "relative_volume": self.symbol_data_cache[ticker]["relative_volume"],
                     "official_open_price": self.symbol_data_cache[ticker]["official_open_price"],
@@ -117,6 +119,7 @@ class TopStocksCacheItem:
                 ret.append({
                     "symbol": ticker,
                     "volume": self.symbol_data_cache[ticker]["volume"],
+                    "free_float": self.symbol_data_cache[ticker]["free_float"],
                     "accumulated_volume": self.symbol_data_cache[ticker]["accumulated_volume"],
                     "relative_volume": self.symbol_data_cache[ticker]["relative_volume"],
                     "official_open_price": self.symbol_data_cache[ticker]["official_open_price"],

{kuhl_haus_mdp-0.1.7 → kuhl_haus_mdp-0.1.8}/tests/components/test_market_data_cache.py

@@ -5,6 +5,8 @@ import pytest
 from kuhl_haus.mdp.components.market_data_cache import MarketDataCache
 from massive.rest.models import TickerSnapshot
 
+from kuhl_haus.mdp.models.market_data_cache_keys import MarketDataCacheKeys
+
 
 @pytest.fixture
 def mock_massive_api_key():
@@ -93,7 +95,7 @@ async def test_get_ticker_snapshot_with_cache_hit_expect_ticker_snapshot_returne
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-    mock_cache_key = "snapshots:TEST"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_SNAPSHOTS.value}:TEST"
     mock_cached_value = mock_data_dict
     mock_redis_client.get.return_value = json.dumps(mock_cached_value)
     mock_snapshot.return_value = TickerSnapshot(**mock_cached_value)
@@ -115,7 +117,7 @@ async def test_get_ticker_snapshot_without_cache_hit_expect_ticker_snapshot_retu
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-    mock_cache_key = "snapshots:TEST"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_SNAPSHOTS.value}:TEST"
     mock_snapshot_instance = MagicMock(spec=TickerSnapshot)
     mock_snapshot_instance.ticker = "TEST"
     mock_snapshot_instance.todays_change = 5.0
@@ -145,7 +147,7 @@ async def test_get_ticker_snapshot_with_invalid_cache_data_expect_exception(mock
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-    mock_cache_key = "snapshots:TEST"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_SNAPSHOTS.value}:TEST"
     mock_redis_client.get.return_value = json.dumps({"invalid": "data"})
     mock_from_dict.side_effect = ValueError("Invalid cache data")
 
@@ -163,7 +165,7 @@ async def test_get_ticker_snapshot_with_invalid_cache_data_expect_exception():
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-    mock_cache_key = "snapshots:TEST"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_SNAPSHOTS.value}:TEST"
     mock_redis_client.get.return_value = json.dumps({"invalid": "data"})
 
     # Act & Assert
@@ -180,7 +182,7 @@ async def test_get_avg_volume_with_cache_hit_expect_cached_value_returned():
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-    mock_cache_key = "avg_volume:TEST"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_AVG_VOLUME.value}:TEST"
     mock_cached_value = 1500000
     mock_redis_client.get.return_value = json.dumps(mock_cached_value)
 
@@ -199,7 +201,7 @@ async def test_get_avg_volume_without_cache_hit_expect_avg_volume_returned():
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-    mock_cache_key = "avg_volume:TEST"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_AVG_VOLUME.value}:TEST"
     mock_avg_volume = 2500000
 
     # Create mock FinancialRatio object
@@ -225,7 +227,7 @@ async def test_get_avg_volume_without_cache_hit_expect_avg_volume_returned():
 #     mock_redis_client = AsyncMock()
 #     mock_rest_client = MagicMock()
 #     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-#     mock_cache_key = "avg_volume:TEST"
+#     mock_cache_key = f"{MarketDataCacheKeys.TICKER_AVG_VOLUME.value}:TEST"
 #
 #     mock_redis_client.get.return_value = None
 #     mock_rest_client.list_financials_ratios.return_value = iter([])
@@ -245,7 +247,7 @@ async def test_get_avg_volume_without_cache_hit_expect_avg_volume_returned():
 #     mock_redis_client = AsyncMock()
 #     mock_rest_client = MagicMock()
 #     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-#     mock_cache_key = "avg_volume:TEST"
+#     mock_cache_key = f"{MarketDataCacheKeys.TICKER_AVG_VOLUME.value}:TEST"
 #
 #     # Create multiple mock FinancialRatio objects
 #     mock_financial_ratio_1 = MagicMock()
@@ -271,7 +273,7 @@ async def test_get_avg_volume_caches_with_correct_ttl():
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-    mock_cache_key = "avg_volume:TEST"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_AVG_VOLUME.value}:TEST"
     mock_avg_volume = 3500000
 
     # Create mock FinancialRatio object
@@ -300,7 +302,7 @@ async def test_get_avg_volume_caches_with_correct_ttl():
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-    mock_cache_key = "avg_volume:TEST"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_AVG_VOLUME.value}:TEST"
     mock_avg_volume = 3500000
 
     # Create mock FinancialRatio object
@@ -329,12 +331,12 @@ async def test_get_free_float_with_cache_hit_expect_cached_value_returned():
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-    mock_cache_key = "free_float:TSLA"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_FREE_FLOAT.value}:TEST"
     mock_cached_value = 2643494955
     mock_redis_client.get.return_value = json.dumps(mock_cached_value)
 
     # Act
-    result = await sut.get_free_float("TSLA")
+    result = await sut.get_free_float("TEST")
 
     # Assert
     mock_redis_client.get.assert_awaited_once_with(mock_cache_key)
@@ -347,7 +349,7 @@ async def test_get_free_float_without_cache_hit_expect_free_float_returned():
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_api_key")
-    mock_cache_key = "free_float:TSLA"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_FREE_FLOAT.value}:TEST"
     mock_free_float = 2643494955
 
     # Mock API response
@@ -358,7 +360,7 @@ async def test_get_free_float_without_cache_hit_expect_free_float_returned():
                 "effective_date": "2025-11-14",
                 "free_float": mock_free_float,
                 "free_float_percent": 79.5,
-                "ticker": "TSLA"
+                "ticker": "TEST"
             }
         ],
         "status": "OK"
@@ -382,14 +384,14 @@ async def test_get_free_float_without_cache_hit_expect_free_float_returned():
     sut.http_session = mock_session
 
     # Act
-    result = await sut.get_free_float("TSLA")
+    result = await sut.get_free_float("TEST")
 
     # Assert
     mock_redis_client.get.assert_awaited_once_with(mock_cache_key)
     mock_session.get.assert_called_once()
     call_args = mock_session.get.call_args
     assert call_args[0][0] == "https://api.massive.com/stocks/vX/float"
-    assert call_args[1]["params"]["ticker"] == "TSLA"
+    assert call_args[1]["params"]["ticker"] == "TEST"
     assert call_args[1]["params"]["apiKey"] == "test_api_key"
     mock_response.json.assert_awaited_once()
     mock_redis_client.setex.assert_awaited_once()
@@ -402,7 +404,7 @@ async def test_get_free_float_caches_with_correct_ttl():
     mock_redis_client = AsyncMock()
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
-    mock_cache_key = "free_float:TSLA"
+    mock_cache_key = f"{MarketDataCacheKeys.TICKER_FREE_FLOAT.value}:TEST"
     mock_free_float = 2643494955
 
     # Mock API response
@@ -413,7 +415,7 @@ async def test_get_free_float_caches_with_correct_ttl():
                 "effective_date": "2025-11-14",
                 "free_float": mock_free_float,
                 "free_float_percent": 79.5,
-                "ticker": "TSLA"
+                "ticker": "TEST"
             }
         ],
         "status": "OK"
@@ -436,7 +438,7 @@ async def test_get_free_float_caches_with_correct_ttl():
     sut.http_session = mock_session
 
     # Act
-    result = await sut.get_free_float("TSLA")
+    result = await sut.get_free_float("TEST")
 
     # Assert
     mock_redis_client.get.assert_awaited_once_with(mock_cache_key)
@@ -482,8 +484,8 @@ async def test_get_free_float_with_empty_results_expect_exception():
     sut.http_session = mock_session
 
     # Act & Assert
-    with pytest.raises(Exception, match="No free float data returned for TSLA"):
-        await sut.get_free_float("TSLA")
+    with pytest.raises(Exception, match="No free float data returned for TEST"):
+        await sut.get_free_float("TEST")
 
     mock_redis_client.setex.assert_not_awaited()
 
@@ -523,8 +525,8 @@ async def test_get_free_float_with_invalid_status_expect_exception():
     sut.http_session = mock_session
 
     # Act & Assert
-    with pytest.raises(Exception, match="Invalid response from Massive API for TSLA"):
-        await sut.get_free_float("TSLA")
+    with pytest.raises(Exception, match="Invalid response from Massive API for TEST"):
+        await sut.get_free_float("TEST")
 
     mock_redis_client.setex.assert_not_awaited()
 
@@ -559,7 +561,7 @@ async def test_get_free_float_with_client_error_expect_exception():
 
     # Act & Assert
     with pytest.raises(aiohttp.ClientError, match="Connection timeout"):
-        await sut.get_free_float("TSLA")
+        await sut.get_free_float("TEST")
 
     mock_redis_client.setex.assert_not_awaited()
 
@@ -592,7 +594,7 @@ async def test_get_free_float_with_http_error_expect_exception():
 
     # Act & Assert
     with pytest.raises(Exception, match="HTTP 500 Error"):
-        await sut.get_free_float("TSLA")
+        await sut.get_free_float("TEST")
 
     mock_redis_client.setex.assert_not_awaited()
 
@@ -674,8 +676,8 @@ async def test_publish_data_expect_publish_called():
     mock_rest_client = MagicMock()
     sut = MarketDataCache(rest_client=mock_rest_client, redis_client=mock_redis_client, massive_api_key="test_key")
 
-    test_data = {"symbol": "TSLA", "price": 250.50, "volume": 1000000}
-    test_publish_key = "market:updates:TSLA"
+    test_data = {"symbol": "TEST", "price": 250.50, "volume": 1000000}
+    test_publish_key = "market:updates:TEST"
 
     # Act
     await sut.publish_data(data=test_data, publish_key=test_publish_key)

{kuhl_haus_mdp-0.1.7 → kuhl_haus_mdp-0.1.8}/tests/models/test_top_stocks_cache_item.py

@@ -35,17 +35,17 @@ class TestTopStocksCacheItem(unittest.TestCase):
         """Test the top_volume method with a limit."""
         self.cache_item.top_volume_map = {"AAPL": 1200, "GOOG": 600, "AMZN": 500}
         self.cache_item.symbol_data_cache = {
-            "AAPL": {"volume": 1000, "accumulated_volume": 1200, "relative_volume": 1.2, "official_open_price": 150,
+            "AAPL": {"volume": 1000, "free_float": 14831485766, "accumulated_volume": 1200, "relative_volume": 1.2, "official_open_price": 150,
                      "vwap": 155, "open": 145, "close": 152, "high": 160, "low": 142, "aggregate_vwap": 156,
                      "average_size": 50, "avg_volume": 1000, "prev_day_close": 148, "prev_day_volume": 900,
                      "prev_day_vwap": 154, "change": 4, "pct_change": 2.7, "change_since_open": 7,
                      "pct_change_since_open": 4.8, "start_timestamp": 100000, "end_timestamp": 110000},
-            "AMZN": {"volume": 300, "accumulated_volume": 500, "relative_volume": 1.2, "official_open_price": 3300,
+            "AMZN": {"volume": 300, "free_float": 9698671061,  "accumulated_volume": 500, "relative_volume": 1.2, "official_open_price": 3300,
                      "vwap": 3500, "open": 3250, "close": 3520, "high": 3600, "low": 3200, "aggregate_vwap": 3450,
                      "average_size": 85, "avg_volume": 4000, "prev_day_close": 3200, "prev_day_volume": 3900,
                      "prev_day_vwap": 3400, "change": 200, "pct_change": 10.0, "change_since_open": 270,
                      "pct_change_since_open": 8.3, "start_timestamp": 300000, "end_timestamp": 310000},
-            "GOOG": {"volume": 500, "accumulated_volume": 600, "relative_volume": 1.0, "official_open_price": 2500,
+            "GOOG": {"volume": 500, "free_float": 5029591400,  "accumulated_volume": 600, "relative_volume": 1.0, "official_open_price": 2500,
                      "vwap": 2550, "open": 2450, "close": 2520, "high": 2600, "low": 2400, "aggregate_vwap": 2560,
                      "average_size": 65, "avg_volume": 2000, "prev_day_close": 2510, "prev_day_volume": 1900,
                      "prev_day_vwap": 2530, "change": 10, "pct_change": 0.4, "change_since_open": 70,
@@ -60,17 +60,17 @@ class TestTopStocksCacheItem(unittest.TestCase):
         """Test the top_gappers method with a limit."""
         self.cache_item.top_gappers_map = {"AAPL": 5.0, "GOOG": -3.0, "AMZN": 10.0}
         self.cache_item.symbol_data_cache = {
-            "AAPL": {"volume": 1000, "accumulated_volume": 1200, "relative_volume": 1.5, "official_open_price": 150,
+            "AAPL": {"volume": 1000, "free_float": 14831485766, "accumulated_volume": 1200, "relative_volume": 1.5, "official_open_price": 150,
                      "vwap": 155, "open": 145, "close": 152, "high": 160, "low": 142, "aggregate_vwap": 156,
                      "average_size": 50, "avg_volume": 1000, "prev_day_close": 148, "prev_day_volume": 900,
                      "prev_day_vwap": 154, "change": 4, "pct_change": 5.0, "change_since_open": 7,
                      "pct_change_since_open": 2.5, "start_timestamp": 100000, "end_timestamp": 110000},
-            "AMZN": {"volume": 300, "accumulated_volume": 500, "relative_volume": 1.2, "official_open_price": 3300,
+            "AMZN": {"volume": 300, "free_float": 9698671061, "accumulated_volume": 500, "relative_volume": 1.2, "official_open_price": 3300,
                      "vwap": 3500, "open": 3250, "close": 3520, "high": 3600, "low": 3200, "aggregate_vwap": 3450,
                      "average_size": 85, "avg_volume": 4000, "prev_day_close": 3200, "prev_day_volume": 3900,
                      "prev_day_vwap": 3400, "change": 200, "pct_change": 10.0, "change_since_open": 270,
                      "pct_change_since_open": 8.3, "start_timestamp": 300000, "end_timestamp": 310000},
-            "GOOG": {"volume": 500, "accumulated_volume": 600, "relative_volume": 1.0, "official_open_price": 2500,
+            "GOOG": {"volume": 500, "free_float": 5029591400,  "accumulated_volume": 600, "relative_volume": 1.0, "official_open_price": 2500,
                      "vwap": 2550, "open": 2450, "close": 2520, "high": 2600, "low": 2400, "aggregate_vwap": 2560,
                      "average_size": 65, "avg_volume": 2000, "prev_day_close": 2510, "prev_day_volume": 1900,
                      "prev_day_vwap": 2530, "change": 10, "pct_change": 0.4, "change_since_open": 70,
@@ -84,17 +84,17 @@ class TestTopStocksCacheItem(unittest.TestCase):
         """Test the top_gainers method with a limit."""
         self.cache_item.top_gainers_map = {"AAPL": 2.5, "GOOG": -0.5, "AMZN": 8.3}
         self.cache_item.symbol_data_cache = {
-            "AAPL": {"volume": 500, "accumulated_volume": 800, "relative_volume": 1.6, "official_open_price": 140,
+            "AAPL": {"volume": 500, "free_float": 14831485766, "accumulated_volume": 800, "relative_volume": 1.6, "official_open_price": 140,
                      "vwap": 145, "open": 130, "close": 150, "high": 155, "low": 128, "aggregate_vwap": 150,
                      "average_size": 45, "avg_volume": 900, "prev_day_close": 142, "prev_day_volume": 850,
                      "prev_day_vwap": 145, "change": 8, "pct_change": 2.5, "change_since_open": 20,
                      "pct_change_since_open": 15.4, "start_timestamp": 150000, "end_timestamp": 160000},
-            "AMZN": {"volume": 800, "accumulated_volume": 1200, "relative_volume": 1.4, "official_open_price": 3200,
+            "AMZN": {"volume": 800, "free_float": 9698671061, "accumulated_volume": 1200, "relative_volume": 1.4, "official_open_price": 3200,
                      "vwap": 3300, "open": 3150, "close": 3400, "high": 3450, "low": 3100, "aggregate_vwap": 3350,
                      "average_size": 70, "avg_volume": 3900, "prev_day_close": 3250, "prev_day_volume": 3800,
                      "prev_day_vwap": 3300, "change": 150, "pct_change": 4.6, "change_since_open": 250,
                      "pct_change_since_open": 8.3, "start_timestamp": 170000, "end_timestamp": 180000},
-            "GOOG": {"volume": 500, "accumulated_volume": 600, "relative_volume": 1.0, "official_open_price": 2500,
+            "GOOG": {"volume": 500, "free_float": 5029591400, "accumulated_volume": 600, "relative_volume": 1.0, "official_open_price": 2500,
                      "vwap": 2550, "open": 2450, "close": 2520, "high": 2600, "low": 2400, "aggregate_vwap": 2560,
                      "average_size": 65, "avg_volume": 2000, "prev_day_close": 2510, "prev_day_volume": 1900,
                      "prev_day_vwap": 2530, "change": 10, "pct_change": 0.4, "change_since_open": 70,