以模块方式运行Python脚本时sys.argv不符合预期的问题

当使用python -m <模块名>运行脚本时，sys.argv不会保留-m <模块名>的形式，而是直接指向脚本的绝对路径，这会导致重载脚本时无法维持模块运行的方式。以下是具体的解决方案：

问题本质 #

Python以模块方式运行时，会将模块对应的脚本文件路径填入sys.argv[0]，而非保留-m和模块名的参数结构，导致os.execv重载时无法按模块方式重启脚本。

解决方案：自动构造模块运行的正确argv #

通过推断当前脚本的模块名，构造符合预期的argv列表，保证重载时以模块方式启动。

实现代码 #

在你的重载逻辑中替换原有sys.argv的使用，改用以下代码生成的new_argv：

import sys
import os

def get_module_name():
    """从主模块文件路径推断模块名"""
    main_module = sys.modules['__main__']
    if not hasattr(main_module, '__file__'):
        return None  # 处理交互式环境等特殊情况
    
    main_file = os.path.abspath(main_module.__file__)
    # 找到sys.path中最长的匹配路径（模块所在的根目录）
    longest_match = ''
    for path in sys.path:
        abs_path = os.path.abspath(path)
        if main_file.startswith(abs_path) and len(abs_path) > len(longest_match):
            longest_match = abs_path
    
    if not longest_match:
        return None
    
    # 计算相对路径并转换为模块名（替换路径分隔符为.，去掉.py后缀）
    rel_path = os.path.relpath(main_file, longest_match)
    module_name = os.path.splitext(rel_path)[0].replace(os.sep, '.')
    return module_name

# 判断是否是以模块方式运行
is_module_run = False
main_module = sys.modules['__main__']
# 模块运行时__package__会被设置，或sys.argv[0]是脚本文件且所在目录在sys.path中
if hasattr(main_module, '__package__') and main_module.__package__ is not None:
    is_module_run = True
elif len(sys.argv) > 0 and os.path.isfile(sys.argv[0]):
    script_dir = os.path.dirname(os.path.abspath(sys.argv[0]))
    is_module_run = script_dir in [os.path.abspath(p) for p in sys.path]

# 构造正确的argv
module_name = get_module_name()
if is_module_run and module_name:
    # 模块方式：executable + '-m' + 模块名 + 原有额外参数
    new_argv = [sys.executable, '-m', module_name] + sys.argv[1:]
else:
    # 常规运行：保持原有结构
    new_argv = [sys.executable] + sys.argv

# 重载脚本时使用new_argv
os.execv(sys.executable, new_argv)

方案说明 #

1. 模块名推断：通过主模块的文件路径，结合sys.path找到模块所在的根目录，将相对路径转换为符合Python模块命名规则的名称。
2. 模块运行判断：通过检查__package__属性或脚本目录是否在sys.path中，判断当前是否是以模块方式启动。
3. argv构造：如果是模块运行，生成[python.exe, '-m', 模块名] + 额外参数的列表，保证重载时维持模块运行方式。

测试验证 #

使用python -m test运行你的脚本时，构造出的new_argv会是：

['C:\\Program Files\\Python\\Python311\\python.exe', '-m', 'test']

完全符合预期，重载时会以模块方式重启脚本。

内容的提问来源于stack exchange，提问作者Sushant