Argparse: argument obligatoire 'y' si 'x' est présent

J'ai une exigence comme suit:

./xyifier --prox --lport lport --rport rport

pour l'argument prox, j'utilise action = 'store_true' pour vérifier s'il est présent ou non. Je n'ai besoin d'aucun argument. Mais, si --prox est défini, j'ai également besoin de rport et de lport. Existe-t-il un moyen simple de faire cela avec argparse sans écrire de codage conditionnel personnalisé.

Plus de code:

non_int.add_argument('--prox', action='store_true', help='Flag to turn on proxy')
non_int.add_argument('--lport', type=int, help='Listen Port.')
non_int.add_argument('--rport', type=int, help='Proxy port.')

Non, il n'y a aucune option dans argparse pour créer des ensembles d'options mutuellement inclusifs .

La façon la plus simple de gérer cela serait:

if args.prox and (args.lport is None or args.rport is None):
    parser.error("--prox requires --lport and --rport.")

Vous parlez d'avoir des arguments conditionnellement requis. Comme @borntyping l'a dit, vous pouvez vérifier l'erreur et le faire parser.error(), ou vous pouvez simplement appliquer une exigence liée au --proxmoment où vous ajoutez un nouvel argument.

Une solution simple pour votre exemple pourrait être:

non_int.add_argument('--prox', action='store_true', help='Flag to turn on proxy')
non_int.add_argument('--lport', required='--prox' in sys.argv, type=int)
non_int.add_argument('--rport', required='--prox' in sys.argv, type=int)

De cette façon requiredreçoit soit Trueou Falseselon que l'utilisateur est utilisé --prox. Cela garantit également cela -lportet -rportavoir un comportement indépendant entre eux.

Que diriez-vous d'utiliser la parser.parse_known_args()méthode et d'ajouter ensuite les arguments --lportet les --rportarguments requis si les arguments --proxsont présents.

# just add --prox arg now
non_int = argparse.ArgumentParser(description="stackoverflow question", 
                                  usage="%(prog)s [-h] [--prox --lport port --rport port]")
non_int.add_argument('--prox', action='store_true', 
                     help='Flag to turn on proxy, requires additional args lport and rport')
opts, rem_args = non_int.parse_known_args()
if opts.prox:
    non_int.add_argument('--lport', required=True, type=int, help='Listen Port.')
    non_int.add_argument('--rport', required=True, type=int, help='Proxy port.')
    # use options and namespace from first parsing
    non_int.parse_args(rem_args, namespace = opts)

Gardez également à l'esprit que vous pouvez fournir l'espace de noms optsgénéré après la première analyse tout en analysant les arguments restants la deuxième fois. De cette façon, à la fin, une fois l'analyse terminée, vous aurez un seul espace de noms avec toutes les options.

Désavantages:

• Si --proxn'est pas présent, les deux autres options dépendantes ne sont même pas présentes dans l'espace de noms. Bien que basé sur votre cas d'utilisation, si --proxn'est pas présent, ce qui arrive aux autres options n'est pas pertinent.
• Besoin de modifier le message d'utilisation car l'analyseur ne connaît pas la structure complète
• --lportet --rportn'apparaissent pas dans le message d'aide

Utilisez-vous lportquand proxn'est pas défini. Sinon, pourquoi ne pas faire lportet rportarguments prox? par exemple

parser.add_argument('--prox', nargs=2, type=int, help='Prox: listen and proxy ports')

Cela évite à vos utilisateurs de taper. C'est aussi simple à tester if args.prox is not None:que if args.prox:.

La réponse acceptée a très bien fonctionné pour moi! Puisque tout le code est cassé sans tests, voici comment j'ai testé la réponse acceptée. parser.error()ne génère pas d' argparse.ArgumentErrorerreur, il quitte le processus. Vous devez tester SystemExit.

avec pytest

import pytest
from . import parse_arguments  # code that rasises parse.error()


def test_args_parsed_raises_error():
    with pytest.raises(SystemExit):
        parse_arguments(["argument that raises error"])

avec unittests

from unittest import TestCase
from . import parse_arguments  # code that rasises parse.error()

class TestArgs(TestCase):

    def test_args_parsed_raises_error():
        with self.assertRaises(SystemExit) as cm:
            parse_arguments(["argument that raises error"])

inspiré de: Utilisation d'unittest pour tester argparse - erreurs de sortie