strideutils 0.3.7__tar.gz → 0.3.8__tar.gz

{strideutils-0.3.7 → strideutils-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: strideutils
-Version: 0.3.7
+Version: 0.3.8
 Summary: 
 Author: Your Name
 Author-email: you@example.com

{strideutils-0.3.7 → strideutils-0.3.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "strideutils"
-version = "0.3.7"
+version = "0.3.8"
 description = ""
 authors = ["Your Name <you@example.com>"]
 readme = "README.md"

{strideutils-0.3.7 → strideutils-0.3.8}/strideutils/coingecko.py

@@ -12,6 +12,7 @@ from strideutils.stride_config import config
 
 COINGECKO_ENDPOINT = "https://pro-api.coingecko.com/"
 COINGECKO_PRICE_QUERY = "api/v3/simple/price?ids={ticker_id}&vs_currencies=usd"
+COINGECKO_TVL_QUERY = "api/v3/coins/{ticker_id}"
 
 
 def get_coingecko_name(chain_config: stride_config.ChainConfig) -> str:
@@ -31,7 +32,6 @@ def get_token_price(
 ):
     """
     Reads token price from coingecko
-    If a redis_client is provided, then store the value.
     """
     # TODO: Consider using the coingecko ID for stTokens instead of the redemption rate
     # Get redemption rate for calculating st token prices
@@ -53,6 +53,23 @@ def get_token_price(
     return price
 
 
+def get_tvl(
+    chain_config: stride_config.ChainConfig,
+    api_token: str = config.COINGECKO_API_TOKEN,
+    cache_response: bool = False,
+) -> float:
+    """
+    Fetch TVL from coingecko in USD
+    """
+    coingecko_name = get_coingecko_name(chain_config)
+    endpoint = COINGECKO_ENDPOINT + COINGECKO_TVL_QUERY.format(ticker_id=coingecko_name)
+    headers = {'x-cg-pro-api-key': api_token}
+
+    # The data structure is huge https://docs.coingecko.com/reference/coins-id
+    response = stride_requests.request(endpoint, headers=headers, cache_response=cache_response)
+    return float(response['market_data']['market_cap']['usd'])
+
+
 def get_token_price_history(token: str, num_days=90, use_key=False, memo={}) -> pd.Series:
     """
     returns a Pandas Series with token price data going back num_days
@@ -62,17 +79,18 @@ def get_token_price_history(token: str, num_days=90, use_key=False, memo={}) ->
     price_key = (token, num_days)
     if price_key not in memo:
         tstr = get_coingecko_name(config.get_chain(ticker=token))
-        rurl = f"https://api.coingecko.com/api/v3/coins/{tstr}/market_chart?vs_currency=usd&days={num_days}"
-        rel_headers = {"x-cg-pro-api-key": config.COINGECKO_API_TOKEN}
-        if not use_key:
-            rel_headers = None
+        rurl = (
+            f"https://api.coingecko.com/api/v3/coins/{tstr}/market_chart?vs_currency=usd&days={num_days}"  # noqa:E231
+        )
+        rel_headers = {"x-cg-pro-api-key": config.COINGECKO_API_TOKEN} if use_key else None
+
         r = requests.get(
             rurl,
             headers=rel_headers,
         )
         assert r.status_code == 200
         out_df = pd.DataFrame(r.json()["prices"])
-        out_df.columns = ["time", "price"]
+        out_df.columns = pd.Index(["time", "price"])
         out_df["time"] = pd.to_datetime(out_df["time"], unit="ms")
         out_df["price"] = out_df["price"].astype(float)
         out_df.set_index("time", inplace=True)
@@ -89,17 +107,18 @@ def get_token_volume_history(token: str, num_days=90, use_key=False, memo={}) ->
     volume_key = (token, num_days)
     if volume_key not in memo:
         tstr = get_coingecko_name(config.get_chain(ticker=token))
-        rurl = f"https://api.coingecko.com/api/v3/coins/{tstr}/market_chart?vs_currency=usd&days={num_days}"
-        rel_headers = {"x-cg-pro-api-key": config.COINGECKO_API_TOKEN}
-        if not use_key:
-            rel_headers = None
+        rurl = (
+            f"https://api.coingecko.com/api/v3/coins/{tstr}/market_chart?vs_currency=usd&days={num_days}"  # noqa:E231
+        )
+        rel_headers = {"x-cg-pro-api-key": config.COINGECKO_API_TOKEN} if use_key else None
+
         r = requests.get(
             rurl,
             headers=rel_headers,
         )
         assert r.status_code == 200
         out_df = pd.DataFrame(r.json()["total_volumes"])
-        out_df.columns = ["time", "volume"]
+        out_df.columns = pd.Index(["time", "volume"])
         out_df["time"] = pd.to_datetime(out_df["time"], unit="ms")
         out_df["volume"] = out_df["volume"].astype(float)
         out_df.set_index("time", inplace=True)

{strideutils-0.3.7 → strideutils-0.3.8}/strideutils/stride_config.py

@@ -361,7 +361,7 @@ def _parse_raw_configs(raw_config: dict[str, Any], raw_secrets: dict[str, Option
     # Load non-nested configs, then overwrite with the chain configs
     config_dict = raw_secrets
     config_dict.update(raw_config)
-    config_dict.update(chain_configs)
+    config_dict.update(chain_configs)  # type: ignore
     return Config(**config_dict)  # type: ignore