如何在不导入的情况下获取python模块中定义的所有变量？

Question

例如，如果我具有以下结构：

base.py

foo = 1
bar = 2

扩展名

from base import *

baz = 3
qux = 4

我希望仅获取在extended.py中定义的变量。

---

我试过使用dir和'inspect'，

import inspect
import extended
vars = dir(extended)
members = inspect.getmembers(extended)

但这使vars = ['bar', 'baz', 'foo', 'qux', ...]和vars = ['bar', 'baz', 'foo', 'qux', ...] members=[('bar', 2), ('baz', 3), ('foo', 1), ('qux', 4), ...]

鉴于定义了extended.py的这种结构，在Python中是否有实际执行此操作的方法？

Answer 1

我决定为个人兴趣而设置这个bash，它可能会满足您的需求，我还没有研究更复杂的任务的工作方式，例如切片任务等：

import ast

with open("extended.py") as f:
    code = f.read()

tree = ast.parse(code)

assignments = [
    node
    for node in ast.walk(tree)
    if isinstance(node, ast.Assign)
]

target_variables = set()
for a in assignments:
    print("{}: {}".format(a, ast.dump(a)))
    for target in a.targets:
        target_variables.add(target.id)

print("Found {} assignments to these variable names:".format(len(assignments)))
print(target_variables)

Answer 2

我发现了一种比较简单的方法，虽然可能并不完美，但是可以正常工作：

def find_names_of_module(module):
    code = compile(open(module.__file__).read(), module.__name__, "exec")
    return code.co_names

可悲的是，这确实还会返回from module import *或import module以及功能。 如果不需要它们，可以对其进行过滤以使其不具有模块和功能。

如果只需要顶级变量名，这是汤姆·道尔顿（Tom Dalton）答案的一项调整，仅包含顶级变量：

def find_names_of_module(module):

    with open(module.__file__) as f:
        code = f.read()

    tree = ast.parse(code)

    assignments = (node for node in tree.body if isinstance(node, ast.Assign))
    return {target.id for a in assignments for target in a.targets}

Answer 3

如果您可以更改结构，则可以使用以下解决方案：

base.py

foo = 1
bar = 2

extension.py

baz = 3
qux = 4

扩展名

from base import *
from extension import *

否则，我不确定此代码是否确实有效，并且必须有更好的方法，但是您可以使用模块成员之间的区别：

import inspect
import base
import extended
base_members = inspect.getmembers(base)
ext_members = inspect.getmembers(extended)
ext_only_members = [member for member in ext_members if member not in base_members]

由于列表内容不可散列，因此不能使用集合来获取差异。