add getSettType differentiator
@@ -1227,6 +1227,15 @@ class BadgerSystem:
def getStrategyWant(self, id):
return interface.IERC20(self.sett_system.strategies[id].want())
+ def getSettType(self, id):
+ """
+ Look at the artifact type of the sett and determine it's version. Currently hardcoded.
+ """
+ if id == "yearn.wbtc":
+ return "v2"
+ else:
+ return "v1"
+
def getStrategyArtifact(self, id):
return self.strategy_artifacts[id].artifact
