Python： "with"を使用してzipファイルまたはunzippedファイルを開く

Question

"with"を使用してファイルを開くソリューションを探しています。任意のアイデアを

if f.endswith(".gz"): 
    with gzip.open() as f: 
else: 
    with open() as f: 
for line in f: 

：Hoeverは、私は、ファイルがgzip圧縮されたかのどちらかであるかどうかをチェックして、いずれかのようなgzip.openまたはちょうど開いたものを使用したいと思いますか？

Answer 1

別のコンテキストマネージャを作成する代わりに、単にあなたのwithステートメント内の関数を選択することができます。

with (gzip.open if filename.endswith(".gz") else open)(filename) as f: 
    for line in f: 
     ... 

（.gzで終わることは必ずしもファイルgzipファイルを作成していないことを認識しているものの）。

Answer 2

あなたがこれを行う方法の1つは、それを行う独自のコンテキストマネージャを作成することです。

from contextlib import contextmanager 
def open_file(filename): 
    if filename.endswith(".gz"): 
     with gzip.open(filename) as f: 
      yield f 
    else: 
     with open(filename) as f: 
      yield f 

次に、代わりにこれを使用してファイルを開くことができます。

with open_file("filename") as f: 
    # f will be either the result of `gzip.open` or `open`, depending on its filename. 
    for line in f: 
     pass 

これはあなたを呼び出すように機能するかを選択する柔軟性を与えながら、with句を使用することの利点を維持することができます。