Argparse: Argumen wajib 'y' jika 'x' ada

Saya memiliki persyaratan sebagai berikut:

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

untuk argumen prox, saya menggunakan action = 'store_true' untuk memeriksa apakah ada atau tidak. Saya tidak membutuhkan argumen apapun. Tapi, jika --prox di set, saya membutuhkan rport dan lport juga. Apakah ada cara mudah untuk melakukan ini dengan argparse tanpa menulis pengkodean bersyarat khusus.

Kode Lainnya:

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

Tidak, tidak ada opsi di argparse untuk membuat kumpulan opsi yang saling inklusif .

Cara termudah untuk mengatasinya adalah:

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

Anda sedang berbicara tentang memiliki argumen yang diwajibkan secara bersyarat. Seperti yang dikatakan @borntyping, Anda dapat memeriksa kesalahan dan melakukan parser.error(), atau Anda dapat menerapkan persyaratan yang terkait dengan --proxsaat Anda menambahkan argumen baru.

Solusi sederhana untuk contoh Anda adalah:

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)

Cara ini requiredmenerima salah satu Trueatau Falsetergantung pada apakah pengguna digunakan --prox. Ini juga menjamin -lportdan -rportmemiliki perilaku independen antara satu sama lain.

Bagaimana kalau menggunakan parser.parse_known_args()metode dan kemudian menambahkan --lportdan --rportargs sebagai diperlukan args jika --proxada.

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

Juga perlu diingat bahwa Anda dapat menyediakan namespace yang optsdihasilkan setelah penguraian pertama sambil mengurai argumen yang tersisa untuk kedua kalinya. Dengan begitu, pada akhirnya, setelah semua penguraian selesai, Anda akan memiliki satu namespace dengan semua opsi.

Kekurangan:

• Jika --proxtidak ada, dua opsi dependen lainnya bahkan tidak ada di namespace. Meskipun berdasarkan kasus penggunaan Anda, jika --proxtidak ada, apa yang terjadi pada opsi lain tidak relevan.
• Perlu mengubah pesan penggunaan karena parser tidak mengetahui struktur lengkapnya
• --lportdan --rporttidak muncul di pesan bantuan

Apakah Anda menggunakan lportsaat proxtidak diatur. Jika tidak, mengapa tidak membuat lportdan rportmemperdebatkan prox? misalnya

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

Pengguna Anda tidak perlu mengetik. Hal ini hanya sebagai mudah untuk tes if args.prox is not None:sebagai if args.prox:.

Jawaban yang diterima bekerja dengan baik untuk saya! Karena semua kode rusak tanpa tes, inilah cara saya menguji jawaban yang diterima. parser.error()tidak menimbulkan argparse.ArgumentErrorkesalahan melainkan keluar dari proses. Anda harus mengujinya SystemExit.

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

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

terinspirasi dari: Menggunakan unittest untuk menguji argparse - kesalahan keluar