Python - Ottieni il percorso relativo di tutti i file e le sottocartelle in una directory

Question

Sto cercando un buon modo per ottenere percorsi relativi di file e (sotto) cartelle all'interno di una cartella specifica.

Per il mio approccio attuale sto usando os.walk(). Funziona ma non mi sembra "pitonico":

myFolder = "myfolder" 
fileSet = set() # yes, I need a set() 

for root, dirs, files in os.walk(myFolder): 
    for fileName in files: 
     fileSet.add(root.replace(myFolder, "") + os.sep + fileName) 

Altri suggerimenti?

Grazie

Answer 1

Utilizzare os.path.relpath(). Questo è esattamente il suo uso previsto.

import os 
rootDir = "myfolder" 
fileSet = set() 

for dir_, _, files in os.walk(rootDir): 
    for fileName in files: 
     relDir = os.path.relpath(dir_, rootDir) 
     relFile = os.path.join(relDir, fileName) 
     fileSet.add(relFile) 

Nota che os.path.relpath() è stato aggiunto in Python 2.6 ed è supportato su Windows e Unix.

Answer 2

Questo è probabilmente il modo migliore per essere onesti: è possibile utilizzare glob di andare un certo numero di strati verso il basso, ma se avete bisogno di essere ricorsivo bisogna walk.

Answer 3

Quello che state facendo è perfettamente ragione e penso che dovrebbe essere fatto in questo modo, ma solo per il gusto di alternativa, qui è un tentativo

import os 

def getFiles(myFolder): 
    old = os.getcwd() 
    os.chdir(myFolder) 

    fileSet = set() 

    for root, dirs, files in os.walk(""): 
     for f in files: 
      fileSet.add(os.path.join(root, f)) 

    os.chdir(old) 
    return fileSet 

Answer 4

Penso os.walk è la scelta giusta qui.
forse root.replace(myFolder, "") dovrebbe cambiare a root.replace(myFolder, "", 1) per evitare potenziale sth. sai.
Se si dispone già dei file e delle (sotto) cartelle, vale la pena dare un'occhiata anche a os.path.commonprefix.

Answer 5

myFolder = "myfolder" 
fileSet = set() 

for root, dirs, files in os.walk(myFolder): 
    for fileName in files: 
     fileSet.add(os.path.join(root[len(myFolder):], fileName)) 

Answer 6

È inoltre possibile utilizzare os.listdir() se sono solo alla ricerca di un'alternativa alla soluzione.

Ma fondamentalmente la logica rimarrà la stessa: scorrere i file - se la directory, scorrere la sottodirectory.