Как отображать mb/s вместо b/s в progressbar и Python?

Question

Я пытаюсь изменить способ ProgressBar (версия 2) отображает переданную/общий размер файла в Python 3. Вот код:

import progressbar 
import requests 

url = 'http://download.openbricks.org/sample/H264/big_buck_bunny_1080p_H264_AAC_25fps_7200K_short.MP4' 

request = requests.get(url, stream=True) 

file   = open('test.mp4', 'wb') 
file_size = int(request.headers['Content-Length']) 
file_size_mb = round(file_size/1024/1024,2) 

chunk_sz = 512 

widgets = [progressbar.Bar(marker="#",left="[",right="]"), 
      progressbar.Percentage()," | ", 
      progressbar.FileTransferSpeed()," | ", 
      progressbar.SimpleProgress()," | ", 
      progressbar.ETA()] 

bar = progressbar.ProgressBar(widgets=widgets, maxval=file_size).start() 

i = 0 

for chunk in request.iter_content(chunk_size=chunk_sz): 
    file.write(chunk) 
    i += len(chunk) 
    bar.update(i) 

bar.finish() 
print('File size: {0} MB'.format(file_size_mb)) 

file.close() 

Выход:

[#### ##################### 91% | 741,3 KiB/s | 3397632 из 3714474 | Время: 00:00:08

Размер файла: 3,54 MB

Я хочу, чтобы иметь возможность сделать « из » будет отображаться в формате МБ (например, в file_size_mb), а не в байтах, поскольку по умолчанию он отображается по умолчанию.

Я прочитал документы в Progress Bar’s documentation, но я не смог найти ответ на свой вопрос в любом из приведенных здесь примеров.

Answer 1

вы можете получить этот вид дисплея сравнительно легко с DataSize виджетом:

[######         ] 17% | 256.0 KiB/s | 631.0 KiB/ 3.5 MiB | ETA: 0:00:11 

вы можете сделать это путем замены SimpleProgress виджетов два DataSize виджетов, как показано ниже:

widgets = [progressbar.Bar(marker="#",left="[",right="]"), 
      progressbar.Percentage(), " | ", 
      progressbar.FileTransferSpeed(), " | ", 
      progressbar.DataSize(), "/", 
      progressbar.DataSize(variable="max_value"), " | ", 
      progressbar.ETA()] 

По default, виджет DataSize отображает текущее значение, но вы можете указать, что вы хотите, чтобы он отображал максимальное значение, передав его «max_value» как переменная параметр.

Проблема в том, что она не задокументирована, и вам необходимо проверить код, чтобы найти эту информацию. Таким образом, это может быть изменено в один прекрасный день и больше работать с более поздней версией модуля.

Учитывая, что the DataSize doc, вы можете изменить формат по умолчанию, передав настраиваемое значение в формат.По умолчанию используется формат

format='%(scaled)5.1f %(prefix)s%(unit)s' 

Я думаю, что вы могли бы получить два десятичных числа, указав формат:

format='%(scaled)5.2f %(prefix)s%(unit)s' 

Что касается отображения блока, док упоминает параметр префиксов, который по умолчанию

prefixes=('', 'Ki', 'Mi', 'Gi', 'Ti', 'Pi', 'Ei', 'Zi', 'Yi') 

Возможно, вы можете передать другой кортеж с префиксами, которые вы хотите. Но подумайте об этом article discussing about binary prefixes, прежде чем делать это. Если вы используете полномочия 1024 для преобразования ваших единиц, и если вы хотите быть точным, то вы, вероятно, захотите использовать бинарные префиксы IEC с символом «i».

---

Другим, но больше, способ добиться того, что вы хотите бы вывести SimpleProgress класс, чтобы тот, который подходит вам нужно.

Answer 2

Из progressbar документации:

class progressbar.widgets.SimpleProgress(format=’%(value)d of %(max_value)d’, 
max_width=None, **kwargs) 

Bases: progressbar.widgets.FormatWidgetMixin, progressbar.widgets.WidgetBase 

Returns progress as a count of the total (e.g.: “5 of 47”) 

Так что я предполагаю, что вы можете использовать параметр format делать то, что вам нужно.

widgets = [progressbar.Bar(marker="#",left="[",right="]"), 
      progressbar.Percentage()," | ", 
      progressbar.FileTransferSpeed()," | ", 
      progressbar.SimpleProgress(format='%(value/1024/1024)d of %(max_value/1024/1024)d')," | ", 
      progressbar.ETA()] 

Другой способ состоит в том, чтобы получать на выходе из simpleProgress, разобрать его, чтобы получить значения, и делать математику вручную.