9

私はPythonにかなり慣れていないので、コマンドライン引数を使用するときに簡単なスクリプトを構築する方法にこだわっています.

スクリプトの目的は、画像の並べ替えと操作に関連する私の仕事のいくつかの日常的なタスクを自動化することです。

引数を指定して、関連する関数を呼び出すようにできますが、引数が指定されていない場合のデフォルトのアクションも設定したいと考えています。

これが私の現在の構造です。

parser = argparse.ArgumentParser()
parser.add_argument("-l", "--list", help="Create CSV of images", action="store_true")
parser.add_argument("-d", "--dimensions", help="Copy images with incorrect dimensions to new directory", action="store_true")
parser.add_argument("-i", "--interactive", help="Run script in interactive mode", action="store_true")
args = parser.parse_args()

if args.list:
    func1()
if args.interactive:
    func2()
if args.dimensions:
    func3()

しかし、引数を指定しないと、何も呼び出されません。

Namespace(dimensions=False, interactive=False, list=False)

私が欲しいのは、引数が指定されていない場合のデフォルトの動作です

if args.list:
        func1()
    if args.interactive:
        func2()
    if args.dimensions:
        func3()
    if no args supplied:
        func1()
        func2()
        func3()

これはかなり簡単に達成できるように思えますが、引数をループしてすべてが false かどうかをテストせずに、すべての引数が false であることを検出する方法のロジックに迷っています。

アップデート

複数の引数は一緒に有効です。そのため、elif ルートをたどりませんでした。

更新 2

@unutbuからの回答を考慮して更新されたコードは次のとおりです

すべてがifステートメントでラップされているため、理想的ではないようですが、短期的にはより良い解決策を見つけることができませんでした. @unutbu からの回答を喜んで受け入れます。提供されたその他の改善点をいただければ幸いです。

lists = analyseImages()
    if lists:
        statusTable(lists)

        createCsvPartial = partial(createCsv, lists['file_list'])
        controlInputParital = partial(controlInput, lists)
        resizeImagePartial = partial(resizeImage, lists['resized'])
        optimiseImagePartial = partial(optimiseImage, lists['size_issues'])
        dimensionIssuesPartial = partial(dimensionIssues, lists['dim_issues'])

        parser = argparse.ArgumentParser()
        parser.add_argument(
        "-l", "--list", 
        dest='funcs', action="append_const", const=createCsvPartial,
        help="Create CSV of images",)
        parser.add_argument(
        "-c", "--convert", 
        dest='funcs', action="append_const", const=resizeImagePartial,
        help="Convert images from 1500 x 2000px to 900 x 1200px ",)
        parser.add_argument(
        "-o", "--optimise", 
        dest='funcs', action="append_const", const=optimiseImagePartial,    
        help="Optimise filesize for 900 x 1200px images",)
        parser.add_argument(
        "-d", "--dimensions", 
        dest='funcs', action="append_const", const=dimensionIssuesPartial,
        help="Copy images with incorrect dimensions to new directory",)
        parser.add_argument(
        "-i", "--interactive", 
        dest='funcs', action="append_const", const=controlInputParital,
        help="Run script in interactive mode",)
        args = parser.parse_args()

        if not args.funcs:
            args.funcs = [createCsvPartial, resizeImagePartial, optimiseImagePartial, dimensionIssuesPartial]

        for func in args.funcs:
            func()

    else:
        print 'No jpegs found'
4

3 に答える 3

8

オプションが設定されていない場合はappend_const、属性 に funcs を適用し、args.funcs1 つの if ステートメントを使用してデフォルトの動作を提供できます。

if not args.funcs:
    args.funcs = [func1, func2, func3]

import argparse

def func1(): pass
def func2(): pass
def func3(): pass

parser = argparse.ArgumentParser()
parser.add_argument(
    "-l", "--list",
    dest='funcs', action="append_const", const=func1,
    help="Create CSV of images", )
parser.add_argument(
    "-i", "--interactive",
    dest='funcs', action="append_const", const=func2,
    help="Run script in interactive mode",)
parser.add_argument(
    "-d", "--dimensions",
    dest='funcs', action='append_const', const=func3,
    help="Copy images with incorrect dimensions to new directory")
args = parser.parse_args()
if not args.funcs:
    args.funcs = [func1, func2, func3]

for func in args.funcs:
    print(func.func_name)
    func()

% test.py
func1
func2
func3

% test.py -d
func3

% test.py -d -i
func3
func2

元のコードとは異なり、これにより、ユーザーは関数が呼び出される順序を制御できることに注意してください。

% test.py -i -d
func2
func3

それは望ましいかもしれないし、そうでないかもしれません。

Update 2への対応:

あなたのコードは問題なく動作します。ただし、これを整理する別の方法を次に示します。

  • メインプログラムを句内にネストする代わりにif、次を使用できます

    if not lists:
    sys.exit('No jpegs found')
# put main program here, unnested

    sys.exitは、終了コード 1 に出力して終了しますNo jpegs found。stderr

  • 最初は を使用することを提案しましたfunctools.partialが、別の (おそらくもっと単純な) 方法が思い浮かびます。

    for func in args.funcs:
    func()

    私たちは言うことができます

    for func, args in args.funcs:
    func(args)

    関数だけではなく、タプル(func, args)を格納するだけです。args.func

例えば：

import argparse
import sys

def parse_args(lists):
    funcs = {
        'createCsv': (createCsv, lists['file_list']),
        'resizeImage': (resizeImage, lists['resized']),
        'optimiseImage': (optimiseImage, lists['size_issues']),
        'dimensionIssues': (dimensionIssues, lists['dim_issues']),
        'controlInput': (controlInput, lists)
    }
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-l", "--list",
        dest='funcs', action="append_const", const=funcs['createCsv'],
        help="Create CSV of images",)
    parser.add_argument(
        "-c", "--convert",
        dest='funcs', action="append_const", const=funcs['resizeImage'],
        help="Convert images from 1500 x 2000px to 900 x 1200px ",)
    parser.add_argument(
        "-o", "--optimise",
        dest='funcs', action="append_const", const=funcs['optimiseImage'],
        help="Optimise filesize for 900 x 1200px images",)
    parser.add_argument(
        "-d", "--dimensions",
        dest='funcs', action="append_const", const=funcs['dimensionIssues'],
        help="Copy images with incorrect dimensions to new directory",)
    parser.add_argument(
        "-i", "--interactive",
        dest='funcs', action="append_const", const=funcs['controlInput'],
        help="Run script in interactive mode",)
    args = parser.parse_args()
    if not args.funcs:
        args.funcs = [funcs[task] for task in
                      ('createCsv', 'resizeImage', 'optimiseImage', 'dimensionIssues')]
    return args

if __name__ == '__main__':
    lists = analyseImages()

    if not lists:
        sys.exit('No jpegs found')

    args = parse_args(lists)   
    statusTable(lists)    
    for func, args in args.funcs:
        func(args)
于 2013-01-16T13:19:39.203 に答える
4

それはあなたが望むものですか？

if args.list:
    func1()
if args.interactive:
    func2()
if args.dimensions:
    func3()
if not any(vars(args).values()):
    func1()
    func2()
    func3()

（バージョンについては@JFSebastianに感謝しますany）

于 2013-01-16T12:46:12.243 に答える