From 7d2653746aca05de032971b4792c682aba74123a Mon Sep 17 00:00:00 2001
From: Tim Olson <>
Date: Tue, 5 Dec 2023 23:16:40 -0400
Subject: [PATCH] vault abi from IVault

---
 src/dexorder/contract/__init__.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/dexorder/contract/__init__.py b/src/dexorder/contract/__init__.py
index 0924b33..d7f9935 100644
--- a/src/dexorder/contract/__init__.py
+++ b/src/dexorder/contract/__init__.py
@@ -6,8 +6,14 @@ from .contract_proxy import ContractProxy
 
 
 def get_contract_data(name):
-    with open(f'../contract/out/{name}.sol/{name}.json', 'rt') as file:
-        return json.load(file)
+    try:
+        # try the interface file first
+        with open(f'../contract/out/I{name}.sol/I{name}.json', 'rt') as file:
+            return json.load(file)
+    except FileNotFoundError:
+        # if no interface exists, use the plain name
+        with open(f'../contract/out/{name}.sol/{name}.json', 'rt') as file:
+            return json.load(file)
 
 
 def get_contract_event(contract_name:str, event_name:str):