Python - 命令行参数提取为字符串 ImportError

Question

我正在尝试执行以下操作:

在命令行中:

python __main__.py 127.0.0.1

ma​​in.py 中的:

address = sys.argv[1]

然后在我的 config.py 文件中，我导入这样的地址:

from __main__.py import address

...

EXAMPLE_URL = f"http://{address}/login"

我在一个简单的测试场景中使用 URL，它是从配置导入的，我收到以下错误:

ImportError: cannot import name 'address' from '__main__' (__main__.py)

这是我的目录结构:

QA System/
├── config/
│   ├── config.py
│   ├── __init__.py
├── .... some other unneccessary stuff
└── tests/
    ├── test_scenarios
       ├── test_scenario_01.py
       ├── test_scenario_02.py
       ├── __init__.py
    |── test_suite.py
    |── __init__.py
|
|-- __main__.py   < --- I launch tests from here
|-- __init__.py

似乎错误是在导入过程中的配置文件中，但我不明白错误在哪里。提前致谢!

主要.py文件:

import argparse
import sys

from tests.test_suite import runner

if __name__ == "__main__":
    address = str(sys.argv[1])
    runner()   # This runs the tests, and the tests also use config.py for
                 various settings, I am worried something happens with the
                 imports there.

Answer 1

你有一个 circular import

当您在 Python 中导入模块时，例如，import __main__ 就像您的示例中那样，将为模块的命名空间创建一个 module 对象，该对象最初是空的.然后，随着模块主体中的代码被执行——分配变量、定义函数和类等，命名空间将按顺序填充。例如。采取以下脚本:

$ cat a.py
print(locals())
an_int = 1

print("after an_int = 1")
print(locals())

def a_func(): pass
print("after def a_func(): pass")
print(locals())

然后运行它:

$ python a.py
{'__builtins__': <module '__builtin__' (built-in)>, '__name__': '__main__', '__file__': 'a.py', '__doc__': None, '__package__': None}
after an_int = 1
{'an_int': 1, '__builtins__': <module '__builtin__' (built-in)>, '__file__': 'a.py', '__package__': None, '__name__': '__main__', '__doc__': None}
after def a_func(): pass
{'an_int': 1, 'a_func': <function a_func at 0x6ffffeed758>, '__builtins__': <module '__builtin__' (built-in)>, '__file__': 'a.py', '__package__': None, '__name__': '__main__', '__doc__': None}

您可以看到命名空间被逐行填充。

现在假设我们将其修改为:

$ cat a.py
print(locals())
an_int = 1

print("after an_int = 1")
print(locals())

import b
print("after import b")

def a_func(): pass
print("after def a_func(): pass")
print(locals())

并添加b.py:

$ cat b.py
import sys
print('a is in progress of being imported:', sys.modules['a'])
print("is a_func defined in a? `'a_func' in sys.modules['a'].__dict__`:",
      'a_func' in sys.modules['a'].__dict__)
from a import a_func

然后像这样运行它:

python -c 'import a'

你会得到一些以 Traceback 结尾的输出:

...
after an_int = 1
{'an_int': 1, '__builtins__': <module '__builtin__' (built-in)>, '__file__': 'a.py', '__package__': None, '__name__': '__main__', '__doc__': None}
a is in progress of being imported: <module 'a' from '/path/to/a.py'>
is a_func defined in a? `'a_func' in sys.modules['a'].__dict__`: False
Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "a.py", line 7, in <module>
    import b
  File "b.py", line 1, in <module>
    from a import a_func
ImportError: cannot import name 'a_func'

但是，如果您将 import b 移动到 a_func 的定义之后:

$ cat a.py
print(locals())
an_int = 1

print("after an_int = 1")
print(locals())

def a_func(): pass
print("after def a_func(): pass")
print(locals())

import b
print("after import b")

再次运行 python -c 'import a' 你会看到它起作用了，输出以“after import b”结束。

红利问题:为什么我要运行 python -c 'import a' 而不仅仅是 python a.py？如果您尝试后者，以前的版本实际上有效 并且似乎导入了 a.py 两次。这是因为当您运行 python somemodule.py 时，它最初不是作为 somemodule 导入的，而是作为 __main__ 导入的。因此，从导入系统的角度来看，a 模块在运行 from a import a_func 时尚未导入。一个非常令人困惑的警告。

---

所以在你的情况下，如果你有类似 __main__.py 的东西:

import config
address = 1

在 config.py 中:

from __main__ import address

当你运行 python __main__.py 时，当它运行 import config 时，address 还没有分配，所以代码在 config 中尝试从 __main__ 导入 address 导致 ImportError。

在你的情况下，它有点复杂，因为你没有直接在 __main__ 中导入 config 从它的样子，但间接地，这仍然是正在发生的事情。

在这种情况下，您不应该通过 import 语句在模块之间传递变量。事实上，__main__ 应该只是你代码的命令行前端，你的代码的其余部分应该能够独立于它工作(例如，一个好的设计可以让你运行from tests.test_runner import runner 并从交互式 Python 提示符调用 runner()，原则上，即使您实际上从未那样使用它)。

因此，让 runner(...) 为它需要的任何选项获取参数。然后 __main__.py 只会从命令行参数中获取这些参数。例如:

def runner(address=None):
    # Or maybe just runner(address) if you don't want to make the
    # address argument optional

然后

if __name__ == '__main__':
    address = sys.argv[1]  # Use argparse instead if you can
    runner(address=address)