如何为argparse互斥组设置默认值？Django测试命令优化

你可以通过argparse的store_const动作配合统一的目标参数（dest），再设置全局默认值的方式，让参数解析器自动处理默认逻辑，不用在SiteDiscoverRunner里手动判断。

具体步骤如下：

1. 修改自定义测试命令的add_arguments方法，把三个互斥选项改成使用store_const，并且指定同一个dest参数，每个选项对应不同的常量值：

from django.core.management.commands.test import Command as TestCommand

class CustomTestCommand(TestCommand):
    def add_arguments(self, parser):
        super().add_arguments(parser)
        
        # 创建互斥参数组
        lang_group = parser.add_mutually_exclusive_group()
        
        lang_group.add_argument(
            '--test-all-languages',
            action='store_const',
            const='all',
            dest='test_language_mode',
            help='运行所有语言版本的测试用例'
        )
        lang_group.add_argument(
            '--test-default-languages',
            action='store_const',
            const='default',
            dest='test_language_mode',
            help='仅运行默认语言版本的测试用例'
        )
        lang_group.add_argument(
            '--test-only-english',
            action='store_const',
            const='english',
            dest='test_language_mode',
            help='仅运行英文版本的测试用例'
        )
        
        # 设置默认模式为default，对应--test-default-languages的行为
        parser.set_defaults(test_language_mode='default')

2. 在你的SiteDiscoverRunner里，直接使用解析后的test_language_mode参数即可，不用再做默认值判断：

from django.test.runner import DiscoverRunner

class SiteDiscoverRunner(DiscoverRunner):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # 原来手动设置test_default_languages的代码可以彻底删掉了
        
    def run_tests(self, test_labels, extra_tests=None, **kwargs):
        # 根据test_language_mode执行对应逻辑
        if self.test_language_mode == 'all':
            # 处理所有语言测试的逻辑
            pass
        elif self.test_language_mode == 'english':
            # 处理仅英文测试的逻辑
            pass
        else: # default模式
            # 处理默认语言测试的逻辑
            pass
        return super().run_tests(test_labels, extra_tests, **kwargs)

为什么这么做？ #

• 用store_const替代store_true，可以把三个互斥选项的结果统一到一个参数里，避免多个布尔值的判断混乱。
• 通过parser.set_defaults()直接给test_language_mode设置默认值为default，当用户没有指定任何语言参数时，这个参数会自动被设为default，完全由参数解析器处理默认逻辑，不需要在Runner里手动判断。

这样就能彻底移除Runner中手动设置默认值的代码，把逻辑统一到参数解析阶段。

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