62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
import logging
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
|
|||
|
|
def config_logging():
|
|||
|
|
str_format = '%(asctime)s | %(levelname)s | %(message)s'
|
|||
|
|
logging.basicConfig(format=str_format, level=logging.INFO)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ProcessBar:
|
|||
|
|
def __init__(self, progress, max_iter, prefix='Progress',
|
|||
|
|
suffix='downloading', completed_suffix='completed',
|
|||
|
|
bar_length=50):
|
|||
|
|
self.progress = progress
|
|||
|
|
self.max_iter = max_iter
|
|||
|
|
|
|||
|
|
self.bar_length = bar_length
|
|||
|
|
|
|||
|
|
self.prefix = prefix
|
|||
|
|
self.suffix = suffix
|
|||
|
|
|
|||
|
|
self.completed_suffix = completed_suffix
|
|||
|
|
|
|||
|
|
def display(self):
|
|||
|
|
progress_rate = self.progress / self.max_iter
|
|||
|
|
|
|||
|
|
percent = 100 * progress_rate
|
|||
|
|
|
|||
|
|
filled_length = round(self.bar_length * progress_rate)
|
|||
|
|
bar = '#' * filled_length + '-' * (self.bar_length - filled_length)
|
|||
|
|
|
|||
|
|
sys.stdout.write(
|
|||
|
|
'\r{}: |{}| {:.1f}% {}'.format(self.prefix, bar, percent,
|
|||
|
|
self.suffix))
|
|||
|
|
|
|||
|
|
if self.progress == self.max_iter:
|
|||
|
|
sys.stdout.write(
|
|||
|
|
'\r{}: |{}| {:.1f}% {}'.format(self.prefix, bar, percent,
|
|||
|
|
self.completed_suffix))
|
|||
|
|
sys.stdout.write('\n')
|
|||
|
|
|
|||
|
|
sys.stdout.flush()
|
|||
|
|
|
|||
|
|
def update(self):
|
|||
|
|
self.progress += 1
|
|||
|
|
|
|||
|
|
self.display()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def display_speed(start_time, fetch_end_time, task_end_time, target_mp4_file_path):
|
|||
|
|
download_time = fetch_end_time - start_time
|
|||
|
|
total_time = task_end_time - start_time
|
|||
|
|
|
|||
|
|
if download_time < 0.01:
|
|||
|
|
download_speed = os.path.getsize(target_mp4_file_path) / 1024
|
|||
|
|
else:
|
|||
|
|
download_speed = os.path.getsize( target_mp4_file_path) / download_time / 1024
|
|||
|
|
|
|||
|
|
logging.info( "download successfully！ take {:.2f}s, average download speed is {:.2f}KB/s".format( total_time, download_speed))
|