ファイルオブジェクトまたはパスを受け入れるPython関数

Question

パスを文字列またはファイルオブジェクトとして受け入れる関数を記述したいと思います。これまでのところ私が持っている：

do_stuffが開かれたファイルオブジェクトを受け取り

def awesome_parse(path_or_file): 
    if isinstance(path_or_file, basestring): 
     f = open(path_or_file, 'rb') 
    else: 
     f = path_or_file 
    with f as f: 
     return do_stuff(f) 

。

これを行うより良い方法はありますか？ with f as f:には何らかの反響がありますか？

ありがとうございます！

Answer 1

あなたのコードに関して奇妙なことは、開いているファイルが渡された場合、それを閉じることです。これは良くありません。ファイルを開いたコードが何であれ、ファイルを閉じる責任があります。

def awesome_parse(path_or_file): 
    if isinstance(path_or_file, basestring): 
     f = file_to_close = open(path_or_file, 'rb') 
    else: 
     f = path_or_file 
     file_to_close = None 
    try: 
     return do_stuff(f) 
    finally: 
     if file_to_close: 
      file_to_close.close() 

することはでき独自のコンテキストマネージャ書き込むことによって、この抽象化：これは、しかし、もう少し複雑な関数を作る

@contextlib.contextmanager 
def awesome_open(path_or_file): 
    if isinstance(path_or_file, basestring): 
     f = file_to_close = open(path_or_file, 'rb') 
    else: 
     f = path_or_file 
     file_to_close = None 

    try: 
     yield f 
    finally: 
     if file_to_close: 
      file_to_close.close() 

def awesome_parse(path_or_file): 
    with awesome_open(path_or_file) as f: 
     return do_stuff(f) 

Answer 2

あなたは行うことができます：

def awesome_parse(do_stuff): 
    """Decorator to open a filename passed to a function 
     that requires an open file object""" 
    def parse_path_or_file(path_or_file): 
     """Use a ternary expression to either open the file from the filename 
      or just pass the extant file object on through""" 
     with (open(path_or_file, 'rb') 
       if isinstance(path_or_file, basestring) 
       else path_or_file) as f: 
      return do_stuff(f) 
    return parse_path_or_file 

そして、あなたが開いているファイルオブジェクト上のものを行う任意の関数を宣言します。

@awesome_parse 
def do_things(open_file_object): 
    """This will always get an open file object even if passed a string""" 
    pass 

@awesome_parse 
def do_stuff(open_file_object): 
    """So will this""" 
    pass 

編集2：デコレータに関するより詳細な情報を。