открыть файл в зависимости от того, является ли он .gz или нет

Question

Я пытаюсь выяснить, какой лучший способ открыть файл python основан на его типе.

Например, у меня есть что-то основное, но это просто не кажется мне «пифоническим», и я чувствую, что он каким-то образом может быть реорганизован и написан более чистым;

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