refactoring

Author: gitoleg

plugins/bap/utils/ida.py

@@ -1,14 +1,16 @@
 """Utilities that interact with IDA."""
 import idaapi
 import idc
-from idaapi import *
-from idc import *
 import idautils
 
 from ._service import Service
 from ._comment_handler import CommentHandlers
 from ._ctyperewriter import Rewriter
 
+try:
+    from idc import get_segm_name
+except ImportError:
+    from idaapi import get_segm_name
 
 service = Service()
 comment = CommentHandlers()

tests/mockidaapi.py

@@ -14,3 +14,4 @@ def get_cmt(ea, off): NotImplemented
 def set_cmt(ea, off): NotImplemented
 def askyn_c(dflt, title): NotImplemented
 def get_input_file_path() : NotImplemented
+def get_segm_name(ea): NotImplemented