Biblioteka wielokrotnego użytku, aby uzyskać czytelną dla człowieka wersję rozmiaru pliku?

238

W sieci dostępne są różne fragmenty, które umożliwiłyby przywrócenie czytelnego dla człowieka rozmiaru z wielkości bajtów:

>>> human_readable(2048)
'2 kilobytes'
>>>

Ale czy istnieje biblioteka Python, która to zapewnia?

Sridhar Ratnakumar
źródło
2
Myślę, że mieści się to w tytule „zbyt małe zadanie, aby wymagać biblioteki”. Jeśli spojrzysz na źródło pośpiechu. Rozmiar pliku, jest tylko jedna funkcja z kilkanaście wierszy kodu. I nawet to można by zagęścić.
Ben Blank
8
Zaletą korzystania z biblioteki jest to, że jest ona zwykle testowana (zawiera testy, które można uruchomić w przypadku, gdy edycja wprowadzi błąd). Jeśli dodasz testy, to nie będzie już „tuzin wierszy kodu” :-)
Sridhar Ratnakumar
Ponowne wynalezienie koła w społeczności Pythona jest szalone i śmieszne. Wystarczy ls -h /path/to/file.ext wykona zadanie. To powiedziawszy, zaakceptowana odpowiedź robi dobrą robotę. Kudo.
Edward Aung,

Odpowiedzi:

522

Rozwiązanie powyższego problemu „zbyt małe zadanie, aby wymagać biblioteki” przez prostą implementację:

def sizeof_fmt(num, suffix='B'):
    for unit in ['','Ki','Mi','Gi','Ti','Pi','Ei','Zi']:
        if abs(num) < 1024.0:
            return "%3.1f%s%s" % (num, unit, suffix)
        num /= 1024.0
    return "%.1f%s%s" % (num, 'Yi', suffix)

Obsługuje:

  • wszystkie obecnie znane prefiksy binarne
  • liczby ujemne i dodatnie
  • liczby większe niż 1000 Yobibytes
  • dowolne jednostki (może lubisz liczyć w Gibibits!)

Przykład:

>>> sizeof_fmt(168963795964)
'157.4GiB'

autor: Fred Cirera

Sridhar Ratnakumar
źródło
4
Pomiędzy liczbą a jednostką powinna być spacja. Jeśli wyprowadzasz html lub lateks, powinna to być przestrzeń niełamliwa.
josch
3
tylko myśl, ale dla dowolnego (?) przyrostka innego niż B(tj. dla jednostek innych niż bajty) chciałbyś, aby czynnik był 1000.0raczej niż 1024.0nie?
Anentropic
5
Jeśli chcesz zwiększyć dokładność składnika dziesiętnego, zmień 1wiersze 4 i 6 na dowolną dokładność.
Matthew G,
44
na pewno byłoby miło, gdyby cała ta iteracja tego „zbyt małego zadania” została przechwycona i zamknięta w bibliotece z testami.
fess.
6
@ MD004 Odwrotnie. Prefiksy są zdefiniowane w taki sposób, że 1 KB = 1000 B i 1 KiB = 1024 B.
'18
116

Jest to biblioteka, która ma wszystkie funkcje, których szukasz humanize. humanize.naturalsize()wydaje się robić wszystko, czego szukasz.

Pyrocater
źródło
9
Niektóre przykłady wykorzystujące dane z PO: humanize.naturalsize(2048) # => '2.0 kB' ,humanize.naturalsize(2048, binary=True) # => '2.0 KiB' humanize.naturalsize(2048, gnu=True) # => '2.0K'
RubenLaguna
32

Oto moja wersja. Nie używa pętli for. Ma stałą złożoność, O ( 1 ), i jest teoretycznie bardziej wydajny niż odpowiedzi tutaj, które używają pętli for.

from math import log
unit_list = zip(['bytes', 'kB', 'MB', 'GB', 'TB', 'PB'], [0, 0, 1, 2, 2, 2])
def sizeof_fmt(num):
    """Human friendly file size"""
    if num > 1:
        exponent = min(int(log(num, 1024)), len(unit_list) - 1)
        quotient = float(num) / 1024**exponent
        unit, num_decimals = unit_list[exponent]
        format_string = '{:.%sf} {}' % (num_decimals)
        return format_string.format(quotient, unit)
    if num == 0:
        return '0 bytes'
    if num == 1:
        return '1 byte'

Aby wyjaśnić, co się dzieje, możemy pominąć kod formatowania łańcucha. Oto linie, które faktycznie wykonują pracę:

exponent = int(log(num, 1024))
quotient = num / 1024**exponent
unit_list[exponent]
żart
źródło
2
skoro mówisz o optymalizacji tak krótkiego kodu, dlaczego nie użyć if / elif / else? Ostatni test num == 1 nie jest konieczny, chyba że oczekujesz ujemnych rozmiarów plików. W przeciwnym razie: dobra robota, podoba mi się ta wersja.
ted
2
Mój kod może być z pewnością bardziej zoptymalizowany. Chciałem jednak wykazać, że zadanie to można rozwiązać ze stałą złożonością.
żartuje
37
Odpowiedzi na for dla pętli są również O (1), ponieważ dla pętli for są ograniczone - ich czas obliczeń nie jest skalowany wraz z rozmiarem danych wejściowych (nie mamy nieograniczonych prefiksów SI).
Thomas Minor
1
prawdopodobnie powinien dodać przecinek do formatowania, więc 1000pokaże się jako 1,000 bytes.
iTayb
3
Zauważ, że gdy używasz Python 3, zip zwraca iterator, więc musisz owinąć go list (). unit_list = list(zip(['bytes', 'kB', 'MB', 'GB', 'TB', 'PB'], [0, 0, 1, 2, 2, 2]))
donarb
30

Poniższe działa w Pythonie 3.6+, jest, moim zdaniem, najłatwiejszą do zrozumienia odpowiedzią tutaj i pozwala dostosować liczbę miejsc dziesiętnych.

def human_readable_size(size, decimal_places=3):
    for unit in ['B','KiB','MiB','GiB','TiB']:
        if size < 1024.0:
            break
        size /= 1024.0
    return f"{size:.{decimal_places}f}{unit}"
wp-overwatch.com
źródło
26

Chociaż wiem, że to pytanie jest pradawne, niedawno opracowałem wersję, która pozwala uniknąć pętli, używając log2do określenia kolejności rozmiarów, która podwaja się jako przesunięcie i indeks do listy sufiksów:

from math import log2

_suffixes = ['bytes', 'KiB', 'MiB', 'GiB', 'TiB', 'PiB', 'EiB', 'ZiB', 'YiB']

def file_size(size):
    # determine binary order in steps of size 10 
    # (coerce to int, // still returns a float)
    order = int(log2(size) / 10) if size else 0
    # format file size
    # (.4g results in rounded numbers for exact matches and max 3 decimals, 
    # should never resort to exponent values)
    return '{:.4g} {}'.format(size / (1 << (order * 10)), _suffixes[order])

Można go jednak uznać za mało mityczny ze względu na jego czytelność :)

akaIDIOT
źródło
1
Chociaż podoba mi się log2, powinieneś poradzić sobie z rozmiarem == 0!
Marti Nito,
Trzeba owinąć albo sizeczy (1 << (order * 10)w float()w ostatnim wierszu (dla Pythona 2).
Harvey
FYI: niektórzy mogą import mathtam potrzebować .
monsto
@monsto true, dodane :)
akaIDIOT
Uwielbiam, jak kompaktowy jest! Dziękuję za udostępnienie.
John Crawford,
17

Zawsze musi być jeden z tych facetów. Cóż, dzisiaj to ja. Oto rozwiązanie jednowierszowe - lub dwie linie, jeśli policzymy podpis funkcji.

def human_size(bytes, units=[' bytes','KB','MB','GB','TB', 'PB', 'EB']):
    """ Returns a human readable string reprentation of bytes"""
    return str(bytes) + units[0] if bytes < 1024 else human_size(bytes>>10, units[1:])

>>> human_size(123)
123 bytes
>>> human_size(123456789)
117GB
wp-overwatch.com
źródło
1
Do Twojej wiadomości, wynik zawsze będzie zaokrąglany w dół.
wp-overwatch.com
1
czy nie lepiej byłoby przypisać domyślną listę do jednostek w metodzie, aby uniknąć używania listy jako domyślnego argumentu? (i units=Nonezamiast tego używa )
Imanol,
3
@ImanolEizaguirre Najlepsze praktyki wskazują, że dobrym pomysłem jest robienie tego, co sugerowałeś, abyś nie wprowadził przypadkowo błędów do programu. Jednak ta funkcja w formie, w jakiej jest napisana, jest bezpieczna, ponieważ lista jednostek nigdy nie jest modyfikowana. Gdyby został zmanipulowany, zmiany byłyby trwałe, a wszelkie kolejne wywołania funkcji otrzymałyby zmanipulowaną wersję listy jako domyślny argument dla argumentu jednostek.
wp-overwatch.com
W przypadku Pythona 3, jeśli chcesz przecinek dziesiętny, użyj tego zamiast tego: `` def human_size (fsize, units = ['bytes', 'KB', 'MB', 'GB', 'TB', 'PB' 'EB']): return "{: .2f} {}". Format (zmiennoprzecinkowy (fsize), jednostki [0]), jeśli fsize <1024 inny human_size (fsize / 1024, jednostki [1:]) ``
Omer
15

Jeśli korzystasz z zainstalowanego Django, możesz także wypróbować format rozmiaru pliku :

from django.template.defaultfilters import filesizeformat
filesizeformat(1073741824)

=>

"1.0 GB"
Jon Tirsen
źródło
1
Dla mnie jedną wadą jest to, że używa GB zamiast GiB, mimo że dzieli przez 1024.
Pepedou
9

Jedną z takich bibliotek jest pośpiech . Rozmiar pliku .

>>> from hurry.filesize import alternative
>>> size(1, system=alternative)
'1 byte'
>>> size(10, system=alternative)
'10 bytes'
>>> size(1024, system=alternative)
'1 KB'
Sridhar Ratnakumar
źródło
3
Jednak ta biblioteka nie jest bardzo konfigurowalna. >>> z pośpiechu. rozmiar importu rozmiar >>> rozmiar (1031053) >>> rozmiar (3033053) „2M” Spodziewam się, że pokaże na przykład „2.4M” lub „2423K” ... zamiast rażąco przybliżonego ” 2M ”.
Sridhar Ratnakumar
Zauważ też, że bardzo łatwo jest po prostu pobrać kod bez pośpiechu. Rozmiar pliku i umieścić go bezpośrednio we własnym kodzie, jeśli masz do czynienia z systemami zależności i tym podobnymi. Jest on tak krótki, jak fragmenty, które ludzie tu udostępniają.
mlissner
@SridharRatnakumar, aby nieco inteligentnie rozwiązać problem nadmiernego przybliżenia, zobacz mój matematyczny hack . Czy można jeszcze ulepszyć to podejście?
Acumenus
9

Użycie mocy 1000 lub kibibajtów byłoby bardziej przyjazne dla standardów:

def sizeof_fmt(num, use_kibibyte=True):
    base, suffix = [(1000.,'B'),(1024.,'iB')][use_kibibyte]
    for x in ['B'] + map(lambda x: x+suffix, list('kMGTP')):
        if -base < num < base:
            return "%3.1f %s" % (num, x)
        num /= base
    return "%3.1f %s" % (num, x)

PS Nigdy nie ufaj bibliotece, która drukuje tysiące z sufiksem K (wielkie litery) :)

Giancarlo Sportelli
źródło
P.S. Never trust a library that prints thousands with the K (uppercase) suffix :)Dlaczego nie? Kod może być doskonale poprawny, a autor po prostu nie wziął pod uwagę obudowy na kilogram. Wydaje się, że automatyczne odrzucenie dowolnego kodu w oparciu o twoją regułę wydaje się niesłychanie ...
Douglas Gaskell,
7

To zrobi to, czego potrzebujesz w prawie każdej sytuacji, można ją dostosować za pomocą opcjonalnych argumentów i, jak widać, jest w zasadzie samodokumentujące:

from math import log
def pretty_size(n,pow=0,b=1024,u='B',pre=['']+[p+'i'for p in'KMGTPEZY']):
    pow,n=min(int(log(max(n*b**pow,1),b)),len(pre)-1),n*b**pow
    return "%%.%if %%s%%s"%abs(pow%(-pow-1))%(n/b**float(pow),pre[pow],u)

Przykładowe dane wyjściowe:

>>> pretty_size(42)
'42 B'

>>> pretty_size(2015)
'2.0 KiB'

>>> pretty_size(987654321)
'941.9 MiB'

>>> pretty_size(9876543210)
'9.2 GiB'

>>> pretty_size(0.5,pow=1)
'512 B'

>>> pretty_size(0)
'0 B'

Zaawansowane dostosowania:

>>> pretty_size(987654321,b=1000,u='bytes',pre=['','kilo','mega','giga'])
'987.7 megabytes'

>>> pretty_size(9876543210,b=1000,u='bytes',pre=['','kilo','mega','giga'])
'9.9 gigabytes'

Ten kod jest zgodny zarówno z Python 2, jak i Python 3. Zgodność z PEP8 jest ćwiczeniem dla czytelnika. Pamiętaj, że wynik jest ładny.

Aktualizacja:

Jeśli potrzebujesz tysięcy przecinków, po prostu zastosuj oczywiste rozszerzenie:

def prettier_size(n,pow=0,b=1024,u='B',pre=['']+[p+'i'for p in'KMGTPEZY']):
    r,f=min(int(log(max(n*b**pow,1),b)),len(pre)-1),'{:,.%if} %s%s'
    return (f%(abs(r%(-r-1)),pre[r],u)).format(n*b**pow/b**float(r))

Na przykład:

>>> pretty_units(987654321098765432109876543210)
'816,968.5 YiB'
gojomo
źródło
7

Powinieneś użyć „humanizować”.

>>> humanize.naturalsize(1000000)
'1.0 MB'
>>> humanize.naturalsize(1000000, binary=True)
'976.6 KiB'
>>> humanize.naturalsize(1000000, gnu=True)
'976.6K'

Odniesienie:

https://pypi.org/project/humanize/

Saeed Zahedian Abroodi
źródło
6

Riffing na fragmencie udostępnionym jako alternatywa dla hurry.filesize (), oto fragment, który daje różne liczby precyzji w zależności od użytego prefiksu. Nie jest tak zwięzły jak niektóre fragmenty, ale lubię wyniki.

def human_size(size_bytes):
    """
    format a size in bytes into a 'human' file size, e.g. bytes, KB, MB, GB, TB, PB
    Note that bytes/KB will be reported in whole numbers but MB and above will have greater precision
    e.g. 1 byte, 43 bytes, 443 KB, 4.3 MB, 4.43 GB, etc
    """
    if size_bytes == 1:
        # because I really hate unnecessary plurals
        return "1 byte"

    suffixes_table = [('bytes',0),('KB',0),('MB',1),('GB',2),('TB',2), ('PB',2)]

    num = float(size_bytes)
    for suffix, precision in suffixes_table:
        if num < 1024.0:
            break
        num /= 1024.0

    if precision == 0:
        formatted_size = "%d" % num
    else:
        formatted_size = str(round(num, ndigits=precision))

    return "%s %s" % (formatted_size, suffix)
markltbaker
źródło
6

Pomaga w tym projekt HumanFriendly .

import humanfriendly
humanfriendly.format_size(1024)

Powyższy kod da 1 KB jako odpowiedź.
Przykłady można znaleźć tutaj .

arumuga abinesh
źródło
4

Czerpiąc ze wszystkich poprzednich odpowiedzi, oto moje zdanie na ten temat. Jest to obiekt, który przechowa rozmiar pliku w bajtach jako liczbę całkowitą. Ale kiedy próbujesz wydrukować obiekt, automatycznie otrzymujesz wersję czytelną dla człowieka.

class Filesize(object):
    """
    Container for a size in bytes with a human readable representation
    Use it like this::

        >>> size = Filesize(123123123)
        >>> print size
        '117.4 MB'
    """

    chunk = 1024
    units = ['bytes', 'KB', 'MB', 'GB', 'TB', 'PB']
    precisions = [0, 0, 1, 2, 2, 2]

    def __init__(self, size):
        self.size = size

    def __int__(self):
        return self.size

    def __str__(self):
        if self.size == 0: return '0 bytes'
        from math import log
        unit = self.units[min(int(log(self.size, self.chunk)), len(self.units) - 1)]
        return self.format(unit)

    def format(self, unit):
        if unit not in self.units: raise Exception("Not a valid file size unit: %s" % unit)
        if self.size == 1 and unit == 'bytes': return '1 byte'
        exponent = self.units.index(unit)
        quotient = float(self.size) / self.chunk**exponent
        precision = self.precisions[exponent]
        format_string = '{:.%sf} {}' % (precision)
        return format_string.format(quotient, unit)
xApple
źródło
3

Podoba mi się stała precyzja wersji dziesiętnej senderle , więc oto rodzaj hybrydy z powyższą odpowiedzią Joctee (czy wiesz, że możesz wziąć dzienniki z bazami niecałkowitymi?):

from math import log
def human_readable_bytes(x):
    # hybrid of https://stackoverflow.com/a/10171475/2595465
    #      with https://stackoverflow.com/a/5414105/2595465
    if x == 0: return '0'
    magnitude = int(log(abs(x),10.24))
    if magnitude > 16:
        format_str = '%iP'
        denominator_mag = 15
    else:
        float_fmt = '%2.1f' if magnitude % 3 == 1 else '%1.2f'
        illion = (magnitude + 1) // 3
        format_str = float_fmt + ['', 'K', 'M', 'G', 'T', 'P'][illion]
    return (format_str % (x * 1.0 / (1024 ** illion))).lstrip('0')
HST
źródło
2

DiveIntoPython3 mówi również o tej funkcji.

Sridhar Ratnakumar
źródło
2

Co powiesz na prosty 2 liniowiec:

def humanizeFileSize(filesize):
    p = int(math.floor(math.log(filesize, 2)/10))
    return "%.3f%s" % (filesize/math.pow(1024,p), ['B','KiB','MiB','GiB','TiB','PiB','EiB','ZiB','YiB'][p])

Oto, jak to działa pod maską:

  1. Oblicza log 2 (rozmiar pliku)
  2. Dzieli go przez 10, aby uzyskać najbliższą jednostkę. (np. jeśli rozmiar wynosi 5000 bajtów, najbliższą jednostką jest Kb, więc odpowiedzią powinno być X KiB)
  3. Zwraca file_size/value_of_closest_unitwraz z jednostką.

Nie działa jednak, jeśli rozmiar pliku wynosi 0 lub jest ujemny (ponieważ log jest niezdefiniowany dla liczb 0 i -ve). Możesz dodać dla nich dodatkowe kontrole:

def humanizeFileSize(filesize):
    filesize = abs(filesize)
    if (filesize==0):
        return "0 Bytes"
    p = int(math.floor(math.log(filesize, 2)/10))
    return "%0.2f %s" % (filesize/math.pow(1024,p), ['Bytes','KiB','MiB','GiB','TiB','PiB','EiB','ZiB','YiB'][p])

Przykłady:

>>> humanizeFileSize(538244835492574234)
'478.06 PiB'
>>> humanizeFileSize(-924372537)
'881.55 MiB'
>>> humanizeFileSize(0)
'0 Bytes'

UWAGA - Istnieje różnica między Kb a KiB. KB oznacza 1000 bajtów, podczas gdy KiB oznacza 1024 bajty. KB, MB, GB są wielokrotnościami 1000, podczas gdy KiB, MiB, GiB itp. Są wielokrotnościami 1024. Więcej o tym tutaj

jerrymouse
źródło
1
def human_readable_data_quantity(quantity, multiple=1024):
    if quantity == 0:
        quantity = +0
    SUFFIXES = ["B"] + [i + {1000: "B", 1024: "iB"}[multiple] for i in "KMGTPEZY"]
    for suffix in SUFFIXES:
        if quantity < multiple or suffix == SUFFIXES[-1]:
            if suffix == SUFFIXES[0]:
                return "%d%s" % (quantity, suffix)
            else:
                return "%.1f%s" % (quantity, suffix)
        else:
            quantity /= multiple
Matt Joiner
źródło
1

To, co zaraz znajdziesz, nie jest w żaden sposób najbardziej wydajnym ani najkrótszym rozwiązaniem spośród już opublikowanych. Zamiast tego skupia się na jednym konkretnym problemie , którego brakuje w wielu innych odpowiedziach.

Mianowicie przypadek, gdy podano dane wejściowe 999_995:

Python 3.6.1 ...
...
>>> value = 999_995
>>> base = 1000
>>> math.log(value, base)
1.999999276174054

które, skrócone do najbliższej liczby całkowitej i zastosowane z powrotem do wejścia, daje

>>> order = int(math.log(value, base))
>>> value/base**order
999.995

Wydaje się, że dokładnie tego oczekujemy, dopóki nie będziemy musieli kontrolować precyzji wyjściowej . I wtedy zaczyna się robić trochę trudniej.

Przy precyzji ustawionej na 2 cyfry otrzymujemy:

>>> round(value/base**order, 2)
1000 # K

zamiast 1M.

Jak możemy temu przeciwdziałać?

Oczywiście możemy to wyraźnie sprawdzić:

if round(value/base**order, 2) == base:
    order += 1

Ale czy możemy zrobić lepiej? Czy możemy dowiedzieć się, w którą stronę ordernależy wyciąć, zanim zrobimy ostatni krok?

Okazuje się, że możemy.

Przy założeniu reguły zaokrąglania dziesiętnego 0,5 powyższy ifwarunek przekłada się na:

wprowadź opis zdjęcia tutaj

powodując

def abbreviate(value, base=1000, precision=2, suffixes=None):
    if suffixes is None:
        suffixes = ['', 'K', 'M', 'B', 'T']

    if value == 0:
        return f'{0}{suffixes[0]}'

    order_max = len(suffixes) - 1
    order = log(abs(value), base)
    order_corr = order - int(order) >= log(base - 0.5/10**precision, base)
    order = min(int(order) + order_corr, order_max)

    factored = round(value/base**order, precision)

    return f'{factored:,g}{suffixes[order]}'

dający

>>> abbreviate(999_994)
'999.99K'
>>> abbreviate(999_995)
'1M'
>>> abbreviate(999_995, precision=3)
'999.995K'
>>> abbreviate(2042, base=1024)
'1.99K'
>>> abbreviate(2043, base=1024)
'2K'
ayorgo
źródło
0

odnieść się do Sridhar Ratnakumarodpowiedzi, zaktualizowano do:

def formatSize(sizeInBytes, decimalNum=1, isUnitWithI=False, sizeUnitSeperator=""):
  """format size to human readable string"""
  # https://en.wikipedia.org/wiki/Binary_prefix#Specific_units_of_IEC_60027-2_A.2_and_ISO.2FIEC_80000
  # K=kilo, M=mega, G=giga, T=tera, P=peta, E=exa, Z=zetta, Y=yotta
  sizeUnitList = ['','K','M','G','T','P','E','Z']
  largestUnit = 'Y'

  if isUnitWithI:
    sizeUnitListWithI = []
    for curIdx, eachUnit in enumerate(sizeUnitList):
      unitWithI = eachUnit
      if curIdx >= 1:
        unitWithI += 'i'
      sizeUnitListWithI.append(unitWithI)

    # sizeUnitListWithI = ['','Ki','Mi','Gi','Ti','Pi','Ei','Zi']
    sizeUnitList = sizeUnitListWithI

    largestUnit += 'i'

  suffix = "B"
  decimalFormat = "." + str(decimalNum) + "f" # ".1f"
  finalFormat = "%" + decimalFormat + sizeUnitSeperator + "%s%s" # "%.1f%s%s"
  sizeNum = sizeInBytes
  for sizeUnit in sizeUnitList:
      if abs(sizeNum) < 1024.0:
        return finalFormat % (sizeNum, sizeUnit, suffix)
      sizeNum /= 1024.0
  return finalFormat % (sizeNum, largestUnit, suffix)

a przykładowy wynik to:

def testKb():
  kbSize = 3746
  kbStr = formatSize(kbSize)
  print("%s -> %s" % (kbSize, kbStr))

def testI():
  iSize = 87533
  iStr = formatSize(iSize, isUnitWithI=True)
  print("%s -> %s" % (iSize, iStr))

def testSeparator():
  seperatorSize = 98654
  seperatorStr = formatSize(seperatorSize, sizeUnitSeperator=" ")
  print("%s -> %s" % (seperatorSize, seperatorStr))

def testBytes():
  bytesSize = 352
  bytesStr = formatSize(bytesSize)
  print("%s -> %s" % (bytesSize, bytesStr))

def testMb():
  mbSize = 76383285
  mbStr = formatSize(mbSize, decimalNum=2)
  print("%s -> %s" % (mbSize, mbStr))

def testTb():
  tbSize = 763832854988542
  tbStr = formatSize(tbSize, decimalNum=2)
  print("%s -> %s" % (tbSize, tbStr))

def testPb():
  pbSize = 763832854988542665
  pbStr = formatSize(pbSize, decimalNum=4)
  print("%s -> %s" % (pbSize, pbStr))


def demoFormatSize():
  testKb()
  testI()
  testSeparator()
  testBytes()
  testMb()
  testTb()
  testPb()

  # 3746 -> 3.7KB
  # 87533 -> 85.5KiB
  # 98654 -> 96.3 KB
  # 352 -> 352.0B
  # 76383285 -> 72.84MB
  # 763832854988542 -> 694.70TB
  # 763832854988542665 -> 678.4199PB
Crifan
źródło
0

To rozwiązanie może ci się spodobać, w zależności od tego, jak działa twój umysł:

from pathlib import Path    

def get_size(path = Path('.')):
    """ Gets file size, or total directory size """
    if path.is_file():
        size = path.stat().st_size
    elif path.is_dir():
        size = sum(file.stat().st_size for file in path.glob('*.*'))
    return size

def format_size(path, unit="MB"):
    """ Converts integers to common size units used in computing """
    bit_shift = {"B": 0,
            "kb": 7,
            "KB": 10,
            "mb": 17,
            "MB": 20,
            "gb": 27,
            "GB": 30,
            "TB": 40,}
    return "{:,.0f}".format(get_size(path) / float(1 << bit_shift[unit])) + " " + unit

# Tests and test results
>>> get_size("d:\\media\\bags of fun.avi")
'38 MB'
>>> get_size("d:\\media\\bags of fun.avi","KB")
'38,763 KB'
>>> get_size("d:\\media\\bags of fun.avi","kb")
'310,104 kb'
Peter F.
źródło