Use choices kwarg of add_argument()