python给__name__ == '__main__'传参
时间: 2024-01-10 22:21:25 浏览: 85
在Python中,`__name__`是一个特殊的变量,用于指示当前模块的名称。当一个模块被直接执行时,`__name__`的值为`'__main__'`;当一个模块被导入时,`__name__`的值为模块的名称。
要给`__name__ == '__main__'`传参,可以使用命令行参数或环境变量来实现。下面是两种方法的示例:
1. 使用命令行参数:
```python
import sys
def main(arg1, arg2):
# 在这里编写你的代码
print("arg1:", arg1)
print("arg2:", arg2)
if __name__ == '__main__':
arg1 = sys.argv[1] # 获取第一个命令行参数
arg2 = sys.argv[2] # 获取第二个命令行参数
main(arg1, arg2)
```
在命令行中执行脚本时,可以传递参数,例如:
```
python script.py value1 value2
```
这样,`arg1`的值将是`value1`,`arg2`的值将是`value2`。
2. 使用环境变量:
```python
import os
def main():
arg1 = os.environ.get('ARG1') # 获取环境变量ARG1的值
arg2 = os.environ.get('ARG2') # 获取环境变量ARG2的值
# 在这里编写你的代码
print("arg1:", arg1)
print("arg2:", arg2)
if __name__ == '__main__':
main()
```
在命令行中设置环境变量,例如:
```
export ARG1=value1
export ARG2=value2
python script.py
```
这样,`arg1`的值将是`value1`,`arg2`的值将是`value2`。
阅读全文