Argparse: Erforderliches Argument 'y', wenn 'x' vorhanden ist

Ich habe folgende Anforderungen:

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

Für das Argument prox verwende ich action = 'store_true', um zu überprüfen, ob es vorhanden ist oder nicht. Ich benötige keines der Argumente. Aber wenn --prox gesetzt ist, benötige ich auch rport und lport. Gibt es eine einfache Möglichkeit, dies mit argparse zu tun, ohne eine benutzerdefinierte bedingte Codierung zu schreiben?

Mehr 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.')

Nein, es gibt in argparse keine Option, um sich gegenseitig einschließende Optionssätze zu erstellen .

Der einfachste Weg, damit umzugehen, wäre:

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

Sie sprechen von bedingt erforderlichen Argumenten. Wie @borntyping sagte, können Sie nach dem Fehler suchen und dies tun parser.error(), oder Sie können einfach eine Anforderung anwenden, die sich auf --proxdas Hinzufügen eines neuen Arguments bezieht .

Eine einfache Lösung für Ihr Beispiel könnte sein:

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)

Dieser Weg requiredempfängt entweder Trueoder Falseabhängig davon, ob der Benutzer verwendet wird --prox. Dies garantiert auch, dass -lportund -rportein unabhängiges Verhalten untereinander haben.

Wie wäre es, wenn Sie die parser.parse_known_args()Methode verwenden und dann die Argumente --lportund --rportargs nach Bedarf hinzufügen, falls --proxvorhanden.

# 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)

Beachten Sie auch, dass Sie den optsnach dem ersten Parsen generierten Namespace angeben können, während Sie die verbleibenden Argumente beim zweiten Parsen analysieren. Auf diese Weise haben Sie am Ende nach Abschluss der Analyse einen einzigen Namespace mit allen Optionen.

Nachteile:

• Wenn --proxnicht vorhanden, sind die beiden anderen abhängigen Optionen nicht einmal im Namespace vorhanden. Obwohl basierend auf Ihrem Anwendungsfall, --proxist das, was mit den anderen Optionen passiert, irrelevant , wenn es nicht vorhanden ist.
• Die Verwendungsnachricht muss geändert werden, da der Parser die vollständige Struktur nicht kennt
• --lportund --rportnicht in der Hilfemeldung angezeigt werden

Verwenden Sie, lportwenn proxnicht eingestellt ist. Wenn nicht, warum nicht machen lportund rportargumentieren prox? z.B

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

Das erspart Ihren Benutzern das Tippen. Es ist genauso einfach zu testen if args.prox is not None:wie if args.prox:.

Die akzeptierte Antwort hat bei mir sehr gut funktioniert! Da der gesamte Code ohne Tests fehlerhaft ist, habe ich hier die akzeptierte Antwort getestet. parser.error()löst keinen argparse.ArgumentErrorFehler aus, sondern beendet den Prozess. Sie müssen testen SystemExit.

mit 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"])

mit 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"])

inspiriert von: Verwenden von unittest zum Testen von Argparse - Beenden von Fehlern