Merge pull request #11 from PhantomResearch/support-data-system

Add cached last price function for Polygon API

Author: alineberry

phantom_core/polygon.py

@@ -12,6 +12,7 @@
 from .constants import DATA_TIME_ZONE, PERIOD_CNAME
 from .ohlcv import clean_ohlcv
 from .market_dataframe import MarketDataFrame
+from .cache import ttl_cached
 
 
 def _response_empty(resp: Any) -> bool:
@@ -275,4 +276,9 @@ def get_last_price(ticker: str) -> float:
     assert isinstance(last_trade, LastTrade)
     assert last_trade.price is not None
     last_price = float(last_trade.price)
-    return last_price
+    return last_price
+
+
+@ttl_cached(20)
+def get_last_price_cached(ticker: str) -> float:
+    return get_last_price(ticker)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "phantom-core"
-version = "0.0.10"
+version = "0.0.11"
 description = "Phantom Core"
 authors = ["Adam Lineberry <ablineberry@gmail.com>"]
 license = "MIT"