如何嵌套任意數量的Python文件上下文管理器？

Question

我想採取任意數量的代表嵌套tar檔案的路徑，並對最內層的檔案執行操作。麻煩的是，嵌套可以是任意的，所以我需要的上下文管理器的數量也是任意的。

舉個例子來說：

ARCHIVE_PATH = "path/to/archive.tar" 

INNER_PATHS = (
    "nested/within/archive/one.tar", 
    "nested/within/archive/two.tar", 
    # Arbitary number of these 
) 

def list_inner_contents(archive_path, inner_paths): 
    with TarFile(archive_path) as tf1: 
     with TarFile(fileobj=tf1.extractfile(inner_paths[0])) as tf2: 
      with TarFile(fileobj=tf2.extractfile(inner_paths[1])) as tf3: 
       # ...arbitary level of these! 
       return tfX.getnames() 

contents = list_inner_contents(ARCHIVE_PATH, INNER_PATHS)) 

我不能使用with語句的nesting syntax，因爲可能有任何數量的水平巢。打開第二時，如果有異常拋出

...使用nested()打開這兩個文件是一個編程錯誤作爲第一個文件不會被及時關閉：我不能使用contextlib.nested因爲文檔說就在那裏文件。

有沒有辦法使用語言結構來做到這一點，還是我需要手動管理我自己的打開文件對象堆棧？

Answer 1

對於這種情況，您可以使用遞歸。這種感覺是最自然的情況下（當然如果有在Python無特殊治療方法還）：

ARCHIVE_PATH = "path/to/archive.tar" 

INNER_PATHS = [ 
    "nested/within/archive/one.tar", 
    "nested/within/archive/two.tar", 
    # Arbitary number of these 
] 

def list_inner_contents(archive_path, inner_paths): 
    def rec(tf, rest_paths): 
     if not rest_paths: 
      return tf.getnames() 

     with TarFile(fileobj=tf.extractfile(rest_paths[0])) as tf2: 
      return rec(tf2, rest_paths[1:]) 

    with TarFile(archive_path) as tf: 
     try: 
      return rec(tf, inner_paths) 
     except RuntimeError: 
      # We come here in case the inner_paths list is too long 
      # and we go too deeply in the recursion 
      return None