как наследовать параметры верхнего уровня в позиционных аргументах в argparse - PullRequest
Новая
       44

как наследовать параметры верхнего уровня в позиционных аргументах в argparse

1 голос
/ 23 апреля 2020

У меня есть некоторый код, который я хотел бы реализовать, например, git обработка опций. 

import argparse

def get_options():
    parser = argparse.ArgumentParser(description='database maintance')
    parser.add_argument('--database', default = 'series' )

    subparsers = parser.add_subparsers(dest='operation', help='commands')
    list_parser = subparsers.add_parser('list', help='list')
    list_parser.add_argument('--tables', nargs='*', default=['all'], help="tables to list")

    repair_parser = subparsers.add_parser('repair', help='repair data')
    repair_parser.add_argument('date', help='Repair missing dates')

    delete_parser = subparsers.add_parser('delete', help='delete data')
    delete_parser.add_argument('table', nargs=1, default=argparse.SUPPRESS, help="table to delete date in")
    delete_parser.add_argument('--tables', nargs='*', default=[], help="additional tables to apply delete")

    opts = parser.parse_args()
    print(opts)
    return opts

if __name__ == '__main__':
    opts = get_options()

Проблема с этим кодом заключается в том, что хотя вы можете назвать его таким образом. 

$ ./maint.py list --tables t1 t2
Namespace(database='series', operation='list', tables=['t1', 't2'])

или таким образом .. 

$ ./maint.py --database main list --tables t1 t2
Namespace(database='main', operation='list', tables=['t1', 't2'])

Вы не можете назвать это так ... 

$ ./maint.py list --tables t1 t2 --database main
usage: maint.py [-h] [--database DATABASE] {list,repair,delete} ...
maint.py: error: unrecognized arguments: --database main

Для того, чтобы это работало, мне нужно было бы добавить все аргументы с верхнего уровня парсер для sub_parsers. 

list_parser.add_argument('--database', default = 'series' ) 
repair_parser.add_argument('--database', default = 'series' )
delete_parser.add_argument('--database', default = 'series' )

Есть ли лучший способ сделать это так, чтобы опция верхнего уровня могла быть вызвана всеми sub_parsers?

1 Ответ

1 голос
/ 23 апреля 2020

Вы можете определить опцию --database в отдельном парсере, который передается как основному парсеру, так и всем подпарсерам через опцию parents. 

db = argparse.ArgumentParser(add_help=False)
db.add_argument("--database", default='series')

parser = argparse.ArgumentParser(description='database maintance', <b>parents=[db]</b>)
subparsers = parser.add_subparsers(dest='operation', help='commands')
list_parser = subparsers.add_parser('list', <b>parents=[db],</b> help='list')
repair_parser = subparsers.add_parser('repair', <b>parents=[db],</b> help='repair data')
delete_parser = subparsers.add_parser('delete', <b>parents=[db],</b> help='delete data')

--database затем можно использовать либо до, либо после подкоманды.

...