Cho phép các giá trị cụ thể cho đối số Argparse

Question

Có thể yêu cầu đối số argparse là một trong số ít giá trị đặt trước không?

Cách tiếp cận hiện tại của tôi là kiểm tra đối số theo cách thủ công và nếu đó không phải là một trong các giá trị được phép gọi print_help() và thoát.

Đây là việc thực hiện hiện tại:

... 
parser.add_argument('--val', dest='val', action='store', 
         help='Special testing value') 

args = parser.parse_args(sys.argv[1:]) 
if args.val not in ['a','b','c']: 
    parser.print_help() 
    sys.exit(1) 

Nó không phải là điều này đặc biệt khó khăn, nhưng đúng hơn là nó dường như là lộn xộn.

Answer 1

Một cuộc tranh cãi argparse có thể được giới hạn ở giá trị cụ thể với các tham số choices:

... 
parser.add_argument('--val', dest='val', action='store', 
         choices=['a','b','c'], 
         help='Special testing value') 

args = parser.parse_args(sys.argv[1:]) 

Xem docs để biết thêm chi tiết.