⬆️ 1.6.10

- Checkpoint image now supports custom asset types (SPOT/FUTURE/OPTION)

empiric-package/empiric/core/types.py

@@ -55,6 +55,8 @@
 }
 RPC_CLIENT = FullNodeClient(node_url=RPC_URLS[NETWORK])
 
+AssetType = Literal["SPOT", "FUTURE", "OPTION"]
+
 
 # aggregation mode enum
 @unique

empiric-package/empiric/publisher/assets.py

@@ -1,7 +1,9 @@
 from typing import Dict, List, Tuple, Union
 
 from empiric.core.utils import key_for_asset
+from empiric.core.types import AssetType
 from typing_extensions import TypedDict
+from empiric.publisher.types import UnsupportedAssetError
 
 
 class EmpiricSpotAsset(TypedDict):
@@ -88,6 +90,15 @@ class EmpiricOnchainAsset(TypedDict):
     for asset in EMPIRIC_ALL_ASSETS
 }
 
+# TODO: Add support for option asset type
+def get_asset_spec_for_pair_id_by_type(pair_id: str, asset_type: AssetType) -> EmpiricAsset:
+    if asset_type == "SPOT":
+        return get_spot_asset_spec_for_pair_id(pair_id)
+    elif asset_type == "FUTURE":
+        return get_future_asset_spec_for_pair_id(pair_id)
+    else:
+        raise UnsupportedAssetError("Only SPOT & FUTURE are supported for now.")
+
 
 def get_spot_asset_spec_for_pair_id(pair_id: str) -> EmpiricSpotAsset:
     if pair_id not in _EMPIRIC_ASSET_BY_KEY:

empiric-package/empiric/publisher/types.py

@@ -29,3 +29,12 @@ def __init__(self, message: str):
 
     def serialize(self):
         return self.message
+
+class UnsupportedAssetError:
+    message: str
+
+    def __init__(self, message: str):
+        self.message = message
+
+    def serialize(self):
+        return self.message

empiric-package/pyproject.toml

@@ -7,7 +7,7 @@ profile = "black"
 
 [tool.poetry]
 name = "empiric-network"
-version = "1.6.9"
+version = "1.6.10"
 authors = ["Pragma <[email protected]>"]
 description = "Core package for rollup-native Pragma Oracle"
 readme = "README.md"

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pragma"
-version = "1.6.9"
+version = "1.6.10"
 description = "Pragma, the leading Oracle on zkRollups."
 authors = ["0xevolve <[email protected]>"]
 license = "MIT"

stagecoach/jobs/publishers/checkpoint/Dockerfile

@@ -4,7 +4,7 @@ RUN yum install -y gcc python-devel gmp-devel git
 
 RUN python -m pip install --upgrade pip
 
-RUN pip install empiric-network==1.6.8 typeguard==2.13.3
+RUN pip install empiric-network==1.6.10 typeguard==2.13.3
 
 COPY app.py ${LAMBDA_TASK_ROOT}
 

stagecoach/jobs/publishers/checkpoint/app.py

@@ -5,18 +5,19 @@
 import boto3
 from empiric.core.logger import get_stream_logger
 from empiric.core.utils import currency_pair_to_pair_id
-from empiric.publisher.assets import get_spot_asset_spec_for_pair_id
+from empiric.publisher.assets import get_asset_spec_for_pair_id_by_type
 from empiric.publisher.client import EmpiricPublisherClient
 
 logger = get_stream_logger()
 
 SECRET_NAME = os.environ["SECRET_NAME"]
 NETWORK = os.environ["NETWORK"]
 ASSETS = os.environ["ASSETS"]
+ASSET_TYPE = os.environ.get("ASSET_TYPE", "SPOT")
 
 
 def handler(event, context):
-    assets = [get_spot_asset_spec_for_pair_id(asset) for asset in ASSETS.split(",")]
+    assets = [get_asset_spec_for_pair_id_by_type(asset, ASSET_TYPE) for asset in ASSETS.split(",")]
     invocation = asyncio.run(_handler(assets))
     return {
         "result": invocation,
@@ -41,7 +42,7 @@ async def _handler(assets):
 
     account_address = int(os.environ.get("ACCOUNT_ADDRESS"))
     pairs = [
-        currency_pair_to_pair_id(*p["pair"]) for p in assets if p["type"] == "SPOT"
+        currency_pair_to_pair_id(*p["pair"]) for p in assets if p["type"] == ASSET_TYPE
     ]
 
     publisher_client = EmpiricPublisherClient(