Merge pull request #262 from openmoneymarket/baln-price

Add method to get BALN price from dex
ommfinance · Dec 21, 2021 · 5cb5fc4 · 5cb5fc4
2 parents 4694ada + 67f6b16
commit 5cb5fc4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/score/priceOracle/interfaces.py b/score/priceOracle/interfaces.py
@@ -24,6 +24,10 @@ def getPoolStats(self, _id: int) -> dict:
     def getPriceByName(self, _name: str) -> int:
         pass
 
+    @interface
+    def getBalnPrice(self) -> int:
+        pass
+
 
 class TokenInterface(InterfaceScore):
     @interface

diff --git a/score/priceOracle/priceOracle.py b/score/priceOracle/priceOracle.py
@@ -56,6 +56,9 @@ def getOMMPool(self) -> str:
     def _get_price(self, _base: str, _quote) -> int:
         if _base in STABLE_TOKENS:
             return 1 * 10 ** 18
+        elif _base == "BALN":
+            dex = self.create_interface_score(self.getAddress(DEX), DataSourceInterface)
+            return dex.getBalnPrice()
         else:
             oracle = self.create_interface_score(self.getAddress(BAND_ORACLE), OracleInterface)
             price = oracle.get_reference_data(_base, _quote)