Skip to content

Commit

Permalink
enhance error msg when run deal library
Browse files Browse the repository at this point in the history
  • Loading branch information
yellowbean committed Jul 17, 2023
1 parent 46148b3 commit aa038e9
Showing 1 changed file with 17 additions and 14 deletions.
31 changes: 17 additions & 14 deletions absbox/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,11 +218,6 @@ def runLibrary(self,_id,**p):
#d = {'dealid':_id, 'assump':dealAssump, 'pricing':pricingAssump} | p
runReq = self.build_req(_id, dealAssump, pricingAssump)
result = self._send_req(runReq, deal_library_url, headers={"Authorization":f"Bearer {self.token}"})
try:
result = json.loads(result)
except Exception as e:
console.print(f"❌[bold red]message from API server:{result}")
console.print(f"❌[bold red]{e}")
def lookupReader(x):
match x:
case "china.SPV":
Expand All @@ -231,15 +226,23 @@ def lookupReader(x):
return Generic
case _:
raise RuntimeError(f"Failed to match reader:{x}")

console.print(f"✅[bold green],run success")
classReader = lookupReader(p['reader'])
if read and isinstance(result,list):
return classReader.read(result)
elif read and isinstance(result, dict):
return {k:classReader.read(v) for k,v in result.items()}
else:
return result
try:
result = json.loads(result)

classReader = lookupReader(p['reader'])
#if "error" in result:
# console.print(f"❌[bold red]-> Error: {result['error']}")
# return None
console.print(f"✅[bold green],run success")
if read and isinstance(result,list):
return classReader.read(result)
elif read and isinstance(result, dict):
return {k:classReader.read(v) for k,v in result.items()}
else:
return result
except Exception as e:
console.print(f"❌[bold red]message from API server:{result}")
console.print(f"❌[bold red]{e}")

def _send_req(self,_req,_url,timeout=10,headers={})->dict:
with console.status("") as status:
Expand Down

0 comments on commit aa038e9

Please sign in to comment.