什么是Python主方法？

在Python中，主方法指的是当Python脚本直接运行时执行的代码块。它通常使用以下结构：

if __name__ == '__main__':
    # 主程序代码
    main()

这个特殊的条件语句检查内置变量__name__是否等于'__main__'，如果是，则执行其中的代码。

为什么需要主方法？

主方法基础示例

下面是一个简单的Python脚本，展示了主方法的基本用法：

# 文件名: main_demo.py

def greet(name):
    """打招呼的函数"""
    return f"你好, {name}!"

def main():
    """主函数，程序入口"""
    user = input("请输入你的名字: ")
    message = greet(user)
    print(message)

if __name__ == '__main__':
    main()

解释：当直接运行此脚本时，__name__会被设置为'__main__'，从而调用main()函数。 如果将此文件作为模块导入，__name__将是模块名，不会执行main()函数。

命令行参数处理

主方法通常需要处理命令行参数，Python提供了argparse模块来实现：

import argparse

def process_data(input_file, output_file):
    """处理数据的函数"""
    print(f"处理 {input_file} 并保存到 {output_file}")

def main():
    # 创建参数解析器
    parser = argparse.ArgumentParser(description='数据处理脚本')
    
    # 添加参数
    parser.add_argument('--input', required=True, help='输入文件路径')
    parser.add_argument('--output', required=True, help='输出文件路径')
    
    # 解析参数
    args = parser.parse_args()
    
    # 使用参数
    process_data(args.input, args.output)

if __name__ == '__main__':
    main()

主方法最佳实践

完整示例：文件处理工具

下面是一个完整的文件处理工具示例，展示了主方法的高级用法：

"""
file_processor.py - 文件处理工具

该脚本提供了文件复制和行数统计功能。
"""

import argparse
import shutil
import sys

def copy_file(source, destination):
    """复制文件"""
    try:
        shutil.copy(source, destination)
        print(f"成功复制文件: {source} -> {destination}")
    except Exception as e:
        print(f"复制文件出错: {e}")
        sys.exit(1)

def count_lines(filename):
    """统计文件行数"""
    try:
        with open(filename, 'r', encoding='utf-8') as file:
            lines = file.readlines()
            return len(lines)
    except Exception as e:
        print(f"读取文件出错: {e}")
        sys.exit(1)

def main():
    """主函数，处理命令行参数并执行操作"""
    parser = argparse.ArgumentParser(
        description="文件处理工具",
        epilog="示例: python file_processor.py --copy source.txt dest.txt --lines source.txt"
    )
    
    # 添加互斥命令组
    group = parser.add_mutually_exclusive_group(required=True)
    group.add_argument('--copy', nargs=2, metavar=('SOURCE', 'DEST'), 
                      help="复制源文件到目标位置")
    group.add_argument('--lines', metavar='FILE', 
                      help="统计文件行数")
    
    args = parser.parse_args()
    
    if args.copy:
        source, dest = args.copy
        copy_file(source, dest)
    elif args.lines:
        line_count = count_lines(args.lines)
        print(f"文件 {args.lines} 共有 {line_count} 行")

if __name__ == '__main__':
    main()

常见问题解答