@@ -3554,7 +3554,11 @@ class EdaCmd(cmd.Cmd):
35543554 entity_cache [entity_id ] = True
35553555 entity_cnt += 1
35563556 if entity_cnt <= 1000 :
3557- json_data = sdk_wrapper .call_sdk ("get_entity_by_entity_id" , get_flag_list , entity_id )
3557+ try :
3558+ json_data = sdk_wrapper .call_sdk ("get_entity_by_entity_id" , get_flag_list , entity_id )
3559+ except SzError as err :
3560+ print_message (err , "error" )
3561+ continue
35583562 entity_name = json_data ["RESOLVED_ENTITY" ]["ENTITY_NAME" ]
35593563 tbl1 .rows .append (
35603564 [
@@ -3904,8 +3908,15 @@ class EdaCmd(cmd.Cmd):
39043908 print_message ("Invalid parameter: expected a numeric entity ID" , "warning" )
39053909 return
39063910
3907- sdk_wrapper .begin_call_cache (inspect .stack ()[0 ].function )
3908- view_report (sdk_wrapper .get_relationship_tree (root_entity_id , ** kwargs ))
3911+ try :
3912+ sdk_wrapper .begin_call_cache (inspect .stack ()[0 ].function )
3913+ view_report (sdk_wrapper .get_relationship_tree (root_entity_id , ** kwargs ))
3914+ except SzError as err :
3915+ print_message (err , "error" )
3916+ return
3917+ except Exception as ex :
3918+ print_exception (ex )
3919+ return
39093920
39103921 def help_why (self ):
39113922 print (
@@ -4010,7 +4021,13 @@ class EdaCmd(cmd.Cmd):
40104021 return
40114022
40124023 sdk_wrapper .begin_call_cache (inspect .stack ()[0 ].function )
4013- sdk_wrapper .how_viewer (int (arg ), ** kwargs )
4024+ try :
4025+ sdk_wrapper .how_viewer (int (arg ), ** kwargs )
4026+ except SzError as err :
4027+ print_message (err , "error" )
4028+ except Exception as ex :
4029+ print_exception (ex )
4030+ return
40144031
40154032 def help_score (self ):
40164033 print (
@@ -4142,8 +4159,6 @@ class EdaCmd(cmd.Cmd):
41424159 if not file_name :
41434160 file_name = f"{ entity_list [0 ]} .jsonl" if len (entity_list ) == 1 else "records.jsonl"
41444161
4145- print (entity_list )
4146-
41474162 json_lines = sdk_wrapper .export_records (entity_list , degree = degree )
41484163 if json_lines :
41494164 with open (file_name , "a" if additive else "w" , encoding = "utf-8" ) as f :
0 commit comments