Abrufen von Dateien aus Verzeichnisargument, Sortieren nach Größe

Question

Ich versuche, ein Programm zu schreiben, das ein Befehlszeilenargument verwendet, den durch das Argument bereitgestellten Verzeichnisbaum durchsucht und eine Liste aller Dateien im Verzeichnis erstellt und sortiert nach Länge der Dateien.

Ich bin nicht viel von einem Skript-Kerl - aber das ist, was ich habe und es funktioniert nicht:

import sys 
import os 
from os.path import getsize 

file_list = [] 

#Get dirpath 
dirpath = os.path.abspath(sys.argv[0]) 
if os.path.isdir(dirpath): 
    #Get all entries in the directory 
    for root, dirs, files in os.walk(dirpath): 
     for name in files: 
      file_list.append(name) 
     file_list = sorted(file_list, key=getsize) 
     for item in file_list: 
      sys.stdout.write(str(file) + '\n') 

else: 
    print "not found" 

jemand mich in die richtige Richtung zeigen kann?

Answer 1

Hoffentlich hilft Ihnen diese Funktion (ich benutze Python 2.7):

Answer 2

Sie extrahieren den Befehl und nicht das erste Argument mit argv[0]; verwenden argv[1] dafür:

dirpath = sys.argv[1] # argv[0] contains the command itself. 

Aus Performance-Gründen schlage ich vor, Sie die Dateigrößen Prefetch anstelle der OS über die Größe der gleichen Datei mehrmals während des Sortierens zu fragen (wie durch Koffein vorgeschlagen, os.walk ist der Weg zu go):

files_list = [] 
for path, dirs, files in os.walk(dirpath)): 
    files_list.extend([(os.path.join(path, file), getsize(os.path.join(path, file))) for file in files]) 

Sie nicht die unsortierte Liste benötigen Unter der Annahme, werden wir die Vor-Ort-Art() -Methode:

files_list.sort(key=operator.itemgetter(1)) 

Answer 3

Dies ist ein Ansatz mit Generatoren. Sollte für große Anzahl von Dateien schneller ...

Dies ist der Beginn der beiden Beispiele:

import os, operator, sys 
dirpath = os.path.abspath(sys.argv[0]) 
# make a generator for all file paths within dirpath 
all_files = (os.path.join(basedir, filename) for basedir, dirs, files in os.walk(dirpath) for filename in files ) 

Wenn Sie nur eine Liste der Dateien, ohne die Größe möchten, können Sie diese verwenden:

sorted_files = sorted(all_files, key = os.path.getsize) 

Aber wenn Sie Dateien und Pfade in einer Liste möchten, können Sie diese verwenden:

# make a generator for tuples of file path and size: ('/Path/to/the.file', 1024) 
files_and_sizes = ((path, os.path.getsize(path)) for path in all_files) 
sorted_files_with_size = sorted(files_and_sizes, key = operator.itemgetter(1))