python count size spezifischer Typ Datei txt in einem Verzeichnis

Question

In meinem Ordner gibt es zwei Arten von Dateien: html und txt. Ich möchte die Gesamtgröße der txt Dateien wissen.

Ich habe diesen Code gefunden, aber wie kann ich ihn für meine Bedürfnisse anwenden?

import os 
from os.path import join, getsize 
size = 0 
count = 0 
for root, dirs, files in os.walk(path): 
    size += sum(getsize(join(root, name)) for name in files) 
    count += len(files) 
print count, size 

Answer 1

Sie können, welche Dateien qualifizieren, indem sie ein if den Comprehensions hinzufügen:

for root, dirs, files in os.walk(path): 
    size += sum(getsize(join(root, name)) for name in files if name.endswith('.txt')) 
    count += sum(1 for name in files if name.endswith('.txt')) 
print count, size 

Answer 2

bessere Nutzung glob (https://docs.python.org/3/library/glob.html) statt os Ihre Dateien zu finden. das macht es imho lesbarer.

import glob 
import os 

path = '/tmp' 
files = glob.glob(path + "/**/*.txt") 
total_size = 0 
for file in files: 
    total_size += os.path.getsize(os.path.join(path, file)) 
print len(files), total_size