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

Question

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

在 CMD 中:

python __main__.py 127.0.0.1

ma​​in.py 中的:

地址 = 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为模块的 namespace 创建对象，该 namespace 最初为空。然后，当执行模块主体中的代码时，分配变量、定义函数和类等，命名空间将按顺序填充。例如。采用以下脚本:

$ 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 中的代码尝试导入 address来自__main__结果是 ImportError .

就您而言，情况有点复杂，因为您没有导入 config直接在__main__从表面上看，但间接地，这仍然是正在发生的事情。

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

所以改为 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)