如何用__all__来实现这种导入？

Question

tree ana
ana
├── api
│   ├── __init__.py
│   └── tool.py
└── __init__.py


cat ana/__init__.py
from .api import *

cat ana/api/__init__.py
from .tool import *

cat ana/api/tool.py
def get_data():
    print("ok")

我导入包

from ana import *
get_data()
ok

现在我想通过__all__来实现，

cat ana/__init__.py
__all__ = ['api']

cat ana/api/__init__.py
__all__ = ['tool','get_data']

测试

from ana import *
get_data()
NameError: name 'get_data' is not defined

请问，可否通过__all__的设置,将函数的名字导入到名字空间，进行调用？

Answer 1

from ana import * 导入的是 ana/__init__.py 中定义的 __all__ 里的名字。你需要把 get_data 加进这个 __all__ 里。另外，你让 get_data 在这个 ana/__init__.py 里存在，所以需要 from .api import * 或者 from .api import get_data 或者 from .api.tool import get_data。