Calculer la taille d'un répertoire à l'aide de Python ?

Question

Avant de réinventer cette roue particulière, quelqu'un a-t-il une routine sympa pour calculer la taille d'un répertoire en utilisant Python ? Ce serait très bien si la routine pouvait formater joliment la taille en Mb/Gb, etc.

Answer 1

Cette opération parcourt tous les sous-répertoires, en additionnant les tailles des fichiers :

import os

def get_size(start_path = '.'):
    total_size = 0
    for dirpath, dirnames, filenames in os.walk(start_path):
        for f in filenames:
            fp = os.path.join(dirpath, f)
            # skip if it is symbolic link
            if not os.path.islink(fp):
                total_size += os.path.getsize(fp)

    return total_size

print(get_size(), 'bytes')

Et un oneliner pour le plaisir en utilisant os.listdir ( N'inclut pas les sous-répertoires ) :

import os
sum(os.path.getsize(f) for f in os.listdir('.') if os.path.isfile(f))

Référence :

• os.path.getsize - Donne la taille en octets
• os.walk
• os.path.islink

Mise à jour de Pour utiliser os.path.getsize Cette méthode est plus claire que l'utilisation de la méthode os.stat().st_size.

Merci à ghostdog74 de nous l'avoir signalé !

os.stat - st_size Donne la taille en octets. Peut également être utilisé pour obtenir la taille du fichier et d'autres informations relatives au fichier.

import os

nbytes = sum(d.stat().st_size for d in os.scandir('.') if d.is_file())

Mise à jour 2018

Si vous utilisez Python 3.4 ou une version antérieure, vous pouvez envisager d'utiliser l'option plus efficace walk fournie par la tierce partie scandir paquet. À partir de la version 3.5 de Python, ce paquetage a été intégré à la bibliothèque standard et à la bibliothèque de l'utilisateur. os.walk a reçu l'augmentation correspondante des performances.

Mise à jour 2019

Récemment, j'ai utilisé pathlib de plus en plus, voici un pathlib solution :

from pathlib import Path

root_directory = Path('.')
sum(f.stat().st_size for f in root_directory.glob('**/*') if f.is_file())

Answer 2

Certaines des approches suggérées jusqu'à présent mettent en œuvre une récursion, d'autres utilisent un shell ou ne produisent pas de résultats bien formatés. Lorsque votre code est unique pour les plateformes Linux, vous pouvez obtenir le formatage habituel, récursion incluse, en une seule ligne. A l'exception du print dans la dernière ligne, cela fonctionnera pour les versions actuelles de python2 y python3 :

du.py
-----
#!/usr/bin/python3
import subprocess

def du(path):
    """disk usage in human readable format (e.g. '2,1GB')"""
    return subprocess.check_output(['du','-sh', path]).split()[0].decode('utf-8')

if __name__ == "__main__":
    print(du('.'))

est simple, efficace et fonctionne pour les fichiers et les répertoires multi-niveaux :

$ chmod 750 du.py
$ ./du.py
2,9M

Answer 3

Voici une fonction récursive (elle additionne récursivement la taille de tous les sous-dossiers et de leurs fichiers respectifs) qui renvoie exactement les mêmes octets qu'en exécutant "du -sb ." sous linux (où le "." signifie "le dossier courant") :

import os

def getFolderSize(folder):
    total_size = os.path.getsize(folder)
    for item in os.listdir(folder):
        itempath = os.path.join(folder, item)
        if os.path.isfile(itempath):
            total_size += os.path.getsize(itempath)
        elif os.path.isdir(itempath):
            total_size += getFolderSize(itempath)
    return total_size

print "Size: " + str(getFolderSize("."))

Answer 4

La réponse de monknut est bonne mais elle échoue sur un lien symbolique cassé, vous devez donc également vérifier si ce chemin existe vraiment

if os.path.exists(fp):
    total_size += os.stat(fp).st_size

Answer 5

La réponse acceptée ne prend pas en compte les liens matériels ou logiciels, et compterait ces fichiers deux fois. Il faudrait garder la trace des inodes que vous avez vus, et ne pas ajouter la taille de ces fichiers.

import os
def get_size(start_path='.'):
    total_size = 0
    seen = {}
    for dirpath, dirnames, filenames in os.walk(start_path):
        for f in filenames:
            fp = os.path.join(dirpath, f)
            try:
                stat = os.stat(fp)
            except OSError:
                continue

            try:
                seen[stat.st_ino]
            except KeyError:
                seen[stat.st_ino] = True
            else:
                continue

            total_size += stat.st_size

    return total_size

print get_size()