Làm thế nào để giả lập os.walk trong python với một hệ thống tập tin tạm thời?

Question

Tôi đang thử nghiệm một số mã sử dụng os.walk. Tôi muốn tạo một hệ thống tập tin tạm thời, trong bộ nhớ mà tôi có thể cư trú với các tệp mẫu (trống) và các thư mục mà os.walk sẽ trở lại. Điều này sẽ giúp tôi tiết kiệm được sự phức tạp của các cuộc gọi os.walk giả lập để mô phỏng đệ quy.

Cụ thể, mã Tôi muốn thử nghiệm là:

if recursive: 
    log.debug("Recursively searching for files under %s" % path) 

    for (dir_path, dirs, files) in os.walk(path): 
     log.debug("Found %d files in %s: %s" % (len(files), path, files)) 
     for f in [os.path.join(dir_path, f) for f in files 
        if not re.search(exclude, f)]: 
      yield f 
else: 
    log.debug("Non-recursively searching for files under %s" % path) 

    for (dir_path, dirs, files) in os.walk(path): 
     log.debug("Found %d files in %s: %s" % (len(files), path, files)) 
     for f in [os.path.join(dir_path, f) for f in files 
        if not re.search(exclude, f)]: 
      yield f 

Đây có phải là có thể trong python?

Answer 1

Số os.walk() được xây dựng hoàn toàn xung quanh os.listdir(), với sự hỗ trợ của và os.path.isdir(). Đây chủ yếu là các cuộc gọi hệ thống, vì vậy bạn phải giả lập hệ thống tệp ở cấp hệ thống. Trừ khi bạn muốn viết một FUSE plugin, điều này sẽ không dễ dàng giả lập.

Tất cả os.walk() cần phải trả lại là danh sách các bộ dữ liệu, thực sự. Trừ khi bạn đang thử nghiệm thao tác thành phần dirs, nó không thể đơn giản hơn:

with mock.patch('os.walk') as mockwalk: 
    mockwalk.return_value = [ 
     ('/foo', ('bar',), ('baz',)), 
     ('/foo/bar',(), ('spam', 'eggs')), 
    ]