サブ関数からモジュールをmain（）にインポートするにはどうしたらいいですか？

Question

Pythonスクリプトでは、サブ関数からmain（）へのモジュールのインポートを制御したいと思います。これは可能ですか？どうやって？

なぜ：コマンドライン引数（argparseを使用）を処理し、ユーザー入力に基づいてモジュールをインポートするサブ関数があります。具体的には、matplotlib.pylabをインポートする前にmatplotlibのバックエンドを指定させる必要があります。しかし、私はその質問がより一般的な使用を持っていると思う。あなたは、グローバルを使用していないにもかかわらず、

def main(): 

    args = handleCommandLine(); 

    fig, ax = plt.subplots(1) # ERROR: plt not defined 

    # Snip ... 


def handleCommandLine(): 
    p = argparse.ArgumentParser() 
    p.add_argument('--backend', '-b', default=None, help='Specify plotting backend') 
    args = p.parse_args() 

    if args.backend != None: 
     matplotlib.use(args.backend) #Must occur before importing pyplot 

    import matplotlib.pyplot as plt #Must occur after setting backend, if desired 
    return args 

Answer 1

あなたはモジュールの上部にimport matplotlib.pyplot as pltを実行しているかのようにそれが振る舞うしたい場合：ここで

は、コードスニペットです

def handleCommandLine(): 
    p = argparse.ArgumentParser() 
    p.add_argument('--backend', '-b', default=None, help='Specify plotting backend') 
    args = p.parse_args() 

    if args.backend != None: 
     matplotlib.use(args.backend) #Must occur before importing pyplot 

    global plt #Style choice: Can also be placed at the top of the function 
    import matplotlib.pyplot as plt #Must occur after setting backend 
    return args 

そうでない場合は、あなたが他の変数を扱うのと同じ方法で、関数の戻り値を通じて付属のライブラリの参照を渡すことができます。

def main(): 
    plt, args = handleCommandLine() # CHANGED HERE 
    fig, ax = plt.subplots(1) 
    # ...  

def handleCommandLine(): 
    p = argparse.ArgumentParser() 
    p.add_argument('--backend', '-b', default=None, help='Specify plotting backend') 
    args = p.parse_args() 

    if args.backend != None: 
     matplotlib.use(args.backend) #Must occur before importing pyplot 

    import matplotlib.pyplot as plt #Must occur after setting backend, if desired 
    return plt, args # CHANGED HERE 

を

Answer 2

できません。

pltをhandleCommandLineから返すか、インポートをmainに移動して解決します。

Answer 3

handleCommandLine()の先頭にglobal pltを入れます。

Answer 4

import文はずっと割り当てのようなものです - あなたが明示的にglobalとして宣言しない限り、それは、ローカル名に割り当てられます。グローバル名前空間に次のインポートplt：

def handleCommandLine(): 
    global plt 
    ... 
    import matplotlib.pyplot as plt