python - 根据另一个参数验证参数值

Question

我想根据另一个参数的值来验证一个参数。

SampleScript.py

    def parse_args():
        parser = argparse.ArgumentParser()
        parser.add_argument('-a', '--arg1', help="Year-Month or Year")
        parser.add_argument('-b', '--arg2', help="Needs Month from argument arg1", action="store_true")
        args = parser.parse_args()
        return args

     def main():
        args = parse_args()

     if __name__ == "__main__":
        main()

的值arg1可以是格式YYYY或YYYY-MM. 但是，arg2设置时，我想确保arg1传递的值是格式“YYYY-MM”而不是“YYYY”。

即当我运行时python SampleScript.py -a 2011，程序应该可以正常工作。当我运行时python SampleScript.py -a 2011 -b，程序应该抛出一个错误，因为-a不是YYYY-MM格式

是否可以使用 进行此类验证argparse？