在Python中，__name__ 是一个特殊的内置变量，用于判断当前模块是被导入还是直接运行。理解这个概念对于编写可重用和模块化的代码非常重要。

__name__ 变量的作用

1. 当模块被导入时

当一个Python文件（模块）被导入时，Python解释器会：

  • 执行该模块中的所有代码
  • 将该模块的 __name__ 设置为模块的名称（不包含.py扩展名）
# module.py
print(f"在module.py中，__name__ = {__name__}")

def hello():
    print("Hello from module!")
# main.py
import module

print(f"在main.py中，__name__ = {__name__}")
module.hello()

输出：

在module.py中，__name__ = module
在main.py中，__name__ = __main__
Hello from module!

2. 当模块直接运行时

当一个Python文件被直接运行时：

  • Python解释器会执行该文件中的所有代码
  • 将该文件的 __name__ 设置为 "__main__"
# script.py
print(f"这个脚本被直接运行，__name__ = {__name__}")

if __name__ == "__main__":
    print("这是主程序入口")

直接运行 python script.py 输出：

这个脚本被直接运行，__name__ = __main__
这是主程序入口

if __name__ == "__main__": 的常见用法

1. 测试代码

# math_utils.py
def add(a, b):
    return a + b

def multiply(a, b):
    return a * b

# 测试代码
if __name__ == "__main__":
    print("测试数学函数:")
    print(f"add(2, 3) = {add(2, 3)}")
    print(f"multiply(2, 3) = {multiply(2, 3)}")

2. 主程序入口

# main_app.py
import sys

def process_data(data):
    # 数据处理逻辑
    return data.upper()

def main():
    if len(sys.argv) > 1:
        data = sys.argv[1]
        result = process_data(data)
        print(f"处理结果: {result}")
    else:
        print("请提供输入数据")

if __name__ == "__main__":
    main()

3. 配置设置

# config.py
DEBUG = False

def setup():
    global DEBUG
    DEBUG = True
    print("调试模式已启用")

if __name__ == "__main__":
    setup()
    print(f"DEBUG = {DEBUG}")

实际应用示例

示例1：可重用的模块

# calculator.py
def calculate(expression):
    try:
        return eval（expression)
    except:
        return "无效的表达式"

if __name__ == "__main__":
    # 直接运行时显示交互式计算器
    while True:
        expr = input("输入表达式 (或 'quit' 退出): ")
        if expr.lower() == 'quit':
            break
        result = calculate(expr)
        print(f"结果: {result}")

示例2：多文件项目结构

# utils/helpers.py
def helper_function():
    return "这是辅助函数"

if __name__ == "__main__":
    print("测试helpers模块")
    print(helper_function())
# main.py
from utils import helpers

def main():
    print("主程序运行中...")
    result = helpers.helper_function()
    print(result)

if __name__ == "__main__":
    main()

重要注意事项

  1. 避免全局代码副作用：在模块中避免在全局作用域中执行具有副作用的代码，应该将它们放在 if __name__ == "__main__": 块中
  2. 可测试性：这种模式使得模块既可以作为独立程序运行，也可以被其他模块导入而不执行测试代码
  3. 代码组织：有助于保持代码的清晰和组织性

总结

  • __name__ 是一个内置变量，表示当前模块的名称
  • 当模块被直接运行时，__name__ 的值为 "__main__"
  • 当模块被导入时，__name__ 的值为模块的名称
  • if __name__ == "__main__": 惯用法用于区分模块是被导入还是直接运行
  • 这是Python中编写可重用代码和模块化程序的重要模式

掌握这个概念对于编写专业的Python代码至关重要，它确保了代码的可重用性和模块化设计。