Spaces:
No application file
No application file
| from colorama import Fore, init as colorama_init | |
| import decimal | |
| import platform | |
| import sys | |
| colorama_init() | |
| class ProgressBarStyle: | |
| def __init__(self, advance_char='█', empty_advance_char='▓', progressbar_length=30): | |
| self.advance_char = advance_char | |
| self.empty_advance_char = empty_advance_char | |
| self.progressbar_length = progressbar_length | |
| DEFAULT_STYLE = ProgressBarStyle() | |
| DEFAULT_RICH_STYLE = ProgressBarStyle(f'{Fore.GREEN}━{Fore.RESET}', f'{Fore.LIGHTBLACK_EX}━{Fore.RESET}', 41) | |
| CLASSIC_STYLE = ProgressBarStyle(f'{Fore.GREEN}█{Fore.RESET}', f'{Fore.LIGHTBLACK_EX}▓{Fore.RESET}') | |
| DRACULA_STYLE = ProgressBarStyle(f'{Fore.RED}█{Fore.RESET}', f'{Fore.LIGHTRED_EX}▓{Fore.RESET}') | |
| GIRL_STYLE = ProgressBarStyle(f'{Fore.LIGHTMAGENTA_EX}█{Fore.RESET}', f'{Fore.MAGENTA}▓{Fore.RESET}') | |
| DARK_STYLE = ProgressBarStyle(f'{Fore.LIGHTBLACK_EX}█{Fore.RESET}', ' ') | |
| RAINBOW_STYLE = ProgressBarStyle(f'{Fore.RED}█{Fore.CYAN}█{Fore.YELLOW}█{Fore.GREEN}█{Fore.BLUE}█{Fore.MAGENTA}█{Fore.RESET}', '', 10) | |
| class ProgressBar: | |
| def __init__(self, total, description: str, progress_bar_style=DEFAULT_STYLE): | |
| self.advance = 0 | |
| self.total = total | |
| self.description = description | |
| self.progressbar_length = progress_bar_style.progressbar_length | |
| self.advance_char = progress_bar_style.advance_char | |
| self.empty_advance_char = progress_bar_style.empty_advance_char | |
| self.advance_char_coef = round(self.total/self.progressbar_length, 2) | |
| def is_finished(self): | |
| return self.advance == self.total | |
| def force_finish(self): | |
| self.advance = self.total | |
| def render(self): | |
| if self.is_finished: | |
| advance_char_count = self.progressbar_length | |
| else: | |
| advance_char_count = int(self.advance/self.advance_char_coef) | |
| advance_percent = round(decimal.Decimal(self.advance/self.total), 2)*100 | |
| if platform.release() == '7' and sys.platform.startswith('win'): # disable rendering for windows 7 (cmd.exe does not support ASCII control characters) | |
| pass | |
| else: | |
| print(f'{self.description}{self.advance_char*advance_char_count}{self.empty_advance_char*(self.progressbar_length-advance_char_count)} {advance_percent}%') | |
| print('\033[F', end='') | |
| if self.is_finished: | |
| print() | |
| def update(self, count): | |
| self.advance += count |