os.walk tanpa membuka direktori di bawah ini

Bagaimana cara membatasi os.walkuntuk hanya mengembalikan file di direktori yang saya sediakan?

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        for f in files:
            if os.path.splitext(f)[1] in whitelist:
                outputList.append(os.path.join(root, f))
            else:
                self._email_to_("ignore")
    return outputList

Gunakan walklevelfungsinya.

import os

def walklevel(some_dir, level=1):
    some_dir = some_dir.rstrip(os.path.sep)
    assert os.path.isdir(some_dir)
    num_sep = some_dir.count(os.path.sep)
    for root, dirs, files in os.walk(some_dir):
        yield root, dirs, files
        num_sep_this = root.count(os.path.sep)
        if num_sep + level <= num_sep_this:
            del dirs[:]

Ini berfungsi seperti os.walk, tetapi Anda dapat meneruskannya ke levelparameter yang menunjukkan seberapa dalam rekursi akan berjalan.

Jangan gunakan os.walk.

Contoh:

import os

root = "C:\\"
for item in os.listdir(root):
    if os.path.isfile(os.path.join(root, item)):
        print item

Menurut saya solusinya sebenarnya sangat sederhana.

menggunakan

break

untuk hanya melakukan iterasi pertama pada loop for, harus ada cara yang lebih elegan.

for root, dirs, files in os.walk(dir_name):
    for f in files:
        ...
        ...
    break
...

Pertama kali Anda memanggil os.walk, ia mengembalikan tulip untuk direktori saat ini, kemudian pada perulangan berikutnya isi direktori berikutnya.

Ambil skrip asli dan tambahkan jeda .

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        for f in files:
            if os.path.splitext(f)[1] in whitelist:
                outputList.append(os.path.join(root, f))
            else:
                self._email_to_("ignore")
        break
    return outputList

Saran untuk digunakan listdiritu bagus. Jawaban langsung untuk pertanyaan Anda dengan Python 2 adalah root, dirs, files = os.walk(dir_name).next().

Sintaks setara Python 3 adalah root, dirs, files = next(os.walk(dir_name))

Anda bisa menggunakan os.listdir()yang mengembalikan daftar nama (untuk file dan direktori) di direktori tertentu. Jika Anda perlu membedakan antara file dan direktori, panggil os.stat()setiap nama.

Jika Anda memiliki persyaratan yang lebih kompleks daripada hanya direktori teratas (misalnya mengabaikan direktori VCS, dll.), Anda juga dapat mengubah daftar direktori untuk mencegah os.walk berulang kali melewatinya.

yaitu:

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        dirs[:] = [d for d in dirs if is_good(d)]
        for f in files:
            do_stuff()

Catatan - hati-hati untuk mengubah daftar, bukan hanya mengulangnya. Jelas os.walk tidak tahu tentang rebinding eksternal.

for path, dirs, files in os.walk('.'):
    print path, dirs, files
    del dirs[:] # go only one level deep

Ide yang sama dengan listdir, tetapi lebih pendek:

[f for f in os.listdir(root_dir) if os.path.isfile(os.path.join(root_dir, f))]

Merasa ingin membuang 2 pence saya.

baselevel = len(rootdir.split("\\"))
for subdirs, dirs, files in os.walk(rootdir):
    curlevel = len(subdirs.split("\\"))
    if curlevel <= baselevel + 1:
        [do stuff]

Dengan Python 3, saya bisa melakukan ini:

import os
dir = "/path/to/files/"

#List all files immediately under this folder:
print ( next( os.walk(dir) )[2] )

#List all folders immediately under this folder:
print ( next( os.walk(dir) )[1] )

Sejak Python 3.5 Anda dapat menggunakan os.scandirsebagai pengganti os.listdir. Alih-alih string, Anda mendapatkan iterator DirEntryobjek sebagai gantinya. Dari dokumen:

Menggunakan scandir()alih-alih listdir()dapat secara signifikan meningkatkan kinerja kode yang juga memerlukan jenis file atau informasi atribut file, karena DirEntryobjek mengekspos informasi ini jika sistem operasi menyediakannya saat memindai direktori. Semua DirEntrymetode dapat melakukan panggilan sistem, tetapi is_dir()dan is_file()biasanya hanya memerlukan panggilan sistem untuk tautan simbolik; DirEntry.stat()selalu membutuhkan panggilan sistem di Unix tetapi hanya memerlukan satu panggilan untuk tautan simbolik di Windows.

Anda dapat mengakses nama objek DirEntry.nameyang kemudian setara dengan output darios.listdir

Anda juga bisa melakukan hal berikut:

for path, subdirs, files in os.walk(dir_name):
    for name in files:
        if path == ".": #this will filter the files in the current directory
             #code here

Beginilah cara saya menyelesaikannya

if recursive:
    items = os.walk(target_directory)
else:
    items = [next(os.walk(target_directory))]

...

Ada kendala saat menggunakan listdir. Os.path.isdir (pengenal) harus berupa jalur absolut. Untuk memilih subdirektori yang Anda lakukan:

for dirname in os.listdir(rootdir):
  if os.path.isdir(os.path.join(rootdir, dirname)):
     print("I got a subdirectory: %s" % dirname)

Alternatifnya adalah mengubah ke direktori untuk melakukan pengujian tanpa os.path.join ().

Anda dapat menggunakan potongan ini

for root, dirs, files in os.walk(directory):
    if level > 0:
        # do some stuff
    else:
        break
    level-=1

buat daftar pengecualian, gunakan fnmatch untuk melewati struktur direktori dan lakukan prosesnya

excludes= ['a\*\b', 'c\d\e']
for root, directories, files in os.walk('Start_Folder'):
    if not any(fnmatch.fnmatch(nf_root, pattern) for pattern in excludes):
        for root, directories, files in os.walk(nf_root):
            ....
            do the process
            ....

sama seperti untuk 'termasuk':

if **any**(fnmatch.fnmatch(nf_root, pattern) for pattern in **includes**):

Mengapa tidak menggunakan a rangedan os.walkdikombinasikan dengan zip? Bukan solusi terbaik, tapi akan berhasil juga.

Contohnya seperti ini:

# your part before
for count, (root, dirs, files) in zip(range(0, 1), os.walk(dir_name)):
    # logic stuff
# your later part

Bekerja untuk saya di python 3.

Juga: A breaklebih sederhana juga btw. (Lihat jawaban dari @Pieter)

Sedikit perubahan pada jawaban Alex, tetapi menggunakan __next__():

print(next(os.walk('d:/'))[2]) atau print(os.walk('d:/').__next__()[2])

dengan [2]menjadi filedi root, dirs, filedisebutkan dalam jawaban lainnya

folder root berubah untuk setiap direktori yang ditemukan os.walk. Saya memecahkan memeriksa apakah root == direktori

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        if root == dir_name: #This only meet parent folder
            for f in files:
                if os.path.splitext(f)[1] in whitelist:
                    outputList.append(os.path.join(root, f))
                else:
                    self._email_to_("ignore")
    return outputList

import os

def listFiles(self, dir_name):
    names = []
    for root, directory, files in os.walk(dir_name):
        if root == dir_name:
            for name in files:
                names.append(name)
    return names