假定你的代码中都没有直接从locals()/globals()中获取b的引用,那这还是比较容易的,下面给你举个例子吧。
比如这段代码:from a import b
b
你可以用python的compiler模块去解析它:import compiler
compiler.parse("from a import b b")
得到的输出为这段代码的AST(abstract syntax tree)Module(None, Stmt([From("a", [("b", None)], 0), Discard(Name("b"))]))
有了AST,你就可以很容易地去处理它了,为了更简单一点,可以这样:import parser
st = parser.suite("from a import b b")
print parser.st2list(st)
输出是:[257, [267, [268, [269, [281, [283, [1, "from"], [288, [1, "a"]], [1, "import"], [286, [284, [1, "b"]]]]]], [4, ""]]], [267, [268, [269, [270, [327, [304, [305, [306, [307, [308, [310, [311, [312, [313, [314, [315, [316, [317, [318, [1, "b"]]]]]]]]]]]]]]]]]], [4, ""]], [0, ""]]
这就是一棵用list构成的树了,你要做的事情就是先序遍历,遍历的过程中用一个dict保存所有被import的name,遇到引用name的时候,记录它是否出现过,最后输出dict中没有出现过的name即可。
另外,如果有这样的情况:from a import b
b = 123 #后续对b的引用与import的b无关
你也可以特殊处理一下赋值操作,检查被赋值的name是否在dict中且没有被引用过,如果是,输出它,从dict里删除,然后继续。
EDIT
补充一下,有以下这些import语句需要考虑:
//from Python/Include/graminit.h#define import_stmt 281
#define import_name 282
#define import_from 283
#define import_as_name 284
#define import_as_names 286