ファイルが.gzであるかどうかに応じてファイルを開く

Question

私は、pythonファイルを開く最良の方法がそのタイプに基づいているかを調べようとしています。

例えば、私はこれのような基本的なものを持っていますが、私には 'pythonic'のようには見えませんし、何らかの形でリファクタリングしてより洗練された感じがします。

def openfile(filename): 
    if read_file_from_top: 
     if not filename.endswith('.gz'): 
      with open(filename, 'r') as infile: 
       for line in infile: 
        # do something 
     else: 
      with gzip.open(filename, 'r') as infile: 
       for line in infile: 
        # do something 
    elif read_file_from_bottom: 
     if not filename.endswith('.gz'): 
      with open(filename, 'r') as infile: 
       for line in infile: 
        # do something 
     else: 
      with gzip.open(filename, 'r') as infile: 
       for line in infile: 
        # do something 

これを行うには、発電機を使用する方が良いでしょうか？ありがとう。

Answer 1

あなたが開口部と読書区切る必要があります。重要な機能の定義済みリストを使用して

def openfile(filename, mode='r'): 
    if filename.endswith('.gz'): 
     return gzip.open(filename, mode): 
    else: 
     return open(filename, mode) 

with openfile(filename, 'r') as infile: 
    for line in infile: 
     # do something 

Answer 2

私はこのようなものは、少なくとも少し優れていると思う：

import gzip 


def file_line_gen(filename): 
    if filename.endswith('.gz'): 
     open_fn = gzip.open 
    else: 
     open_fn = open 

    with open_fn(filename, 'r') as f: 
     for line in f: 
      yield line 


for line in file_line_gen('data.gz'): 
    # do something here 
    print repr(line) 

Answer 3

ショートソリューションを：

def processFile(filepath): 
    with [open, gzip.open][0 if not filepath.endswith('.gz') else 1](filepath, 'r') as fh: 
     if read_file_from_top: 
      # do something 
     elif read_file_from_bottom: 
      # do something