Refactored code
This commit is contained in:
		
							
								
								
									
										26
									
								
								src_old/sh_edraft/console/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										26
									
								
								src_old/sh_edraft/console/__init__.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,26 @@ | ||||
| # -*- coding: utf-8 -*- | ||||
|  | ||||
| """ | ||||
| sh_edraft.console  | ||||
| ~~~~~~~~~~~~~~~~~~~ | ||||
|  | ||||
|  | ||||
|  | ||||
| :copyright: (c) 2020 sh-edraft.de | ||||
| :license: MIT, see LICENSE for more details. | ||||
|  | ||||
| """ | ||||
|  | ||||
| __title__ = 'sh_edraft.console' | ||||
| __author__ = 'Sven Heidemann' | ||||
| __license__ = 'MIT' | ||||
| __copyright__ = 'Copyright (c) 2020 sh-edraft.de' | ||||
| __version__ = '2020.12.9' | ||||
|  | ||||
| from collections import namedtuple | ||||
|  | ||||
| # imports: | ||||
| from .console import Console | ||||
|  | ||||
| VersionInfo = namedtuple('VersionInfo', 'major minor micro') | ||||
| version_info = VersionInfo(major=2020, minor=12, micro=9) | ||||
							
								
								
									
										205
									
								
								src_old/sh_edraft/console/console.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										205
									
								
								src_old/sh_edraft/console/console.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,205 @@ | ||||
| import os | ||||
| from typing import Union, Optional | ||||
|  | ||||
| import pyfiglet | ||||
| from tabulate import tabulate | ||||
| from termcolor import colored | ||||
|  | ||||
| from sh_edraft.console.model.background_color import BackgroundColor | ||||
| from sh_edraft.console.model.foreground_color import ForegroundColor | ||||
|  | ||||
|  | ||||
| class Console: | ||||
|     _is_first_write = True | ||||
|  | ||||
|     _background_color: BackgroundColor = BackgroundColor.default | ||||
|     _foreground_color: ForegroundColor = ForegroundColor.default | ||||
|     _x: Optional[int] = None | ||||
|     _y: Optional[int] = None | ||||
|     _disabled: bool = False | ||||
|  | ||||
|     """ | ||||
|         Properties | ||||
|     """ | ||||
|  | ||||
|     @classmethod | ||||
|     @property | ||||
|     def background_color(cls) -> str: | ||||
|         return str(cls._background_color.value) | ||||
|  | ||||
|     @classmethod | ||||
|     @property | ||||
|     def foreground_color(cls) -> str: | ||||
|         return str(cls._foreground_color.value) | ||||
|  | ||||
|     """ | ||||
|         Settings | ||||
|     """ | ||||
|  | ||||
|     @classmethod | ||||
|     def set_background_color(cls, color: Union[BackgroundColor, str]): | ||||
|         if type(color) is str: | ||||
|             cls._background_color = BackgroundColor[color] | ||||
|         else: | ||||
|             cls._background_color = color | ||||
|  | ||||
|     @classmethod | ||||
|     def set_foreground_color(cls, color: Union[ForegroundColor, str]): | ||||
|  | ||||
|         if type(color) is str: | ||||
|             cls._foreground_color = ForegroundColor[color] | ||||
|         else: | ||||
|             cls._foreground_color = color | ||||
|  | ||||
|     @classmethod | ||||
|     def reset_cursor_position(cls): | ||||
|         cls._x = None | ||||
|         cls._y = None | ||||
|  | ||||
|     @classmethod | ||||
|     def set_cursor_position(cls, x: int, y: int): | ||||
|         cls._x = x | ||||
|         cls._y = y | ||||
|  | ||||
|     """ | ||||
|         Useful protected methods | ||||
|     """ | ||||
|  | ||||
|     @classmethod | ||||
|     def _output(cls, string: str, x: int = None, y: int = None, end='\n'): | ||||
|         if cls._is_first_write: | ||||
|             cls._is_first_write = False | ||||
|  | ||||
|         args = [] | ||||
|         colored_args = [] | ||||
|  | ||||
|         if x is not None and y is not None: | ||||
|             args.append(f'\033[{x};{y}H') | ||||
|         elif cls._x is not None and cls._y is not None: | ||||
|             args.append(f'\033[{cls._x};{cls._y}H') | ||||
|  | ||||
|         colored_args.append(string) | ||||
|         if cls._foreground_color != ForegroundColor.default and cls._background_color == BackgroundColor.default: | ||||
|             colored_args.append(cls._foreground_color.value) | ||||
|         elif cls._foreground_color == ForegroundColor.default and cls._background_color != BackgroundColor.default: | ||||
|             colored_args.append(cls._background_color.value) | ||||
|         elif cls._foreground_color != ForegroundColor.default and cls._background_color != BackgroundColor.default: | ||||
|             colored_args.append(cls._foreground_color.value) | ||||
|             colored_args.append(cls._background_color.value) | ||||
|  | ||||
|         args.append(colored(*colored_args)) | ||||
|         print(*args, end=end) | ||||
|  | ||||
|     """ | ||||
|         Useful public methods | ||||
|     """ | ||||
|  | ||||
|     @classmethod | ||||
|     def banner(cls, string: str): | ||||
|         if cls._disabled: | ||||
|             return | ||||
|  | ||||
|         ascii_banner = pyfiglet.figlet_format(string) | ||||
|         cls.write_line(ascii_banner) | ||||
|  | ||||
|     @classmethod | ||||
|     def clear(cls): | ||||
|         os.system('cls' if os.name == 'nt' else 'clear') | ||||
|  | ||||
|     @classmethod | ||||
|     def close(cls): | ||||
|         if cls._disabled: | ||||
|             return | ||||
|  | ||||
|         Console.reset() | ||||
|         Console.write('\n\n\nPress any key to continue...') | ||||
|         Console.read_line() | ||||
|         exit() | ||||
|  | ||||
|     @classmethod | ||||
|     def disable(cls): | ||||
|         cls._disabled = True | ||||
|  | ||||
|     @classmethod | ||||
|     def error(cls, string: str, tb: str = None): | ||||
|         if cls._disabled: | ||||
|             return | ||||
|  | ||||
|         cls.set_foreground_color('red') | ||||
|         if tb is not None: | ||||
|             cls.write_line(f'{string} -> {tb}') | ||||
|         else: | ||||
|             cls.write_line(string) | ||||
|         cls.set_foreground_color('default') | ||||
|  | ||||
|     @classmethod | ||||
|     def enable(cls): | ||||
|         cls._disabled = False | ||||
|  | ||||
|     @classmethod | ||||
|     def read(cls, output: str = None) -> str: | ||||
|         if output is not None: | ||||
|             cls.write(output) | ||||
|  | ||||
|         return input()[0] | ||||
|  | ||||
|     @classmethod | ||||
|     def read_line(cls, output: str = None) -> str: | ||||
|         if cls._disabled: | ||||
|             return '' | ||||
|  | ||||
|         if output is not None: | ||||
|             cls.write(output) | ||||
|  | ||||
|         return input() | ||||
|  | ||||
|     @classmethod | ||||
|     def reset(cls): | ||||
|         cls._background_color = BackgroundColor.default | ||||
|         cls._foreground_color = ForegroundColor.default | ||||
|  | ||||
|     @classmethod | ||||
|     def table(cls, header: list[str], values: list[list[str]]): | ||||
|         if cls._disabled: | ||||
|             return | ||||
|  | ||||
|         table = tabulate(values, headers=header) | ||||
|  | ||||
|         Console.write_line(table) | ||||
|         Console.write('\n') | ||||
|  | ||||
|     @classmethod | ||||
|     def write(cls, *args): | ||||
|         if cls._disabled: | ||||
|             return | ||||
|  | ||||
|         string = ' '.join(map(str, args)) | ||||
|         cls._output(string, end='') | ||||
|  | ||||
|     @classmethod | ||||
|     def write_at(cls, x: int, y: int, *args): | ||||
|         if cls._disabled: | ||||
|             return | ||||
|  | ||||
|         string = ' '.join(map(str, args)) | ||||
|         cls._output(string, x, y, end='') | ||||
|  | ||||
|     @classmethod | ||||
|     def write_line(cls, *args): | ||||
|         if cls._disabled: | ||||
|             return | ||||
|  | ||||
|         string = ' '.join(map(str, args)) | ||||
|         if not cls._is_first_write: | ||||
|             cls._output('') | ||||
|         cls._output(string, end='') | ||||
|  | ||||
|     @classmethod | ||||
|     def write_line_at(cls, x: int, y: int, *args): | ||||
|         if cls._disabled: | ||||
|             return | ||||
|  | ||||
|         string = ' '.join(map(str, args)) | ||||
|         if not cls._is_first_write: | ||||
|             cls._output('', end='') | ||||
|         cls._output(string, x, y, end='') | ||||
							
								
								
									
										27
									
								
								src_old/sh_edraft/console/model/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								src_old/sh_edraft/console/model/__init__.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,27 @@ | ||||
| # -*- coding: utf-8 -*- | ||||
|  | ||||
| """ | ||||
| sh_edraft.console.model  | ||||
| ~~~~~~~~~~~~~~~~~~~ | ||||
|  | ||||
|  | ||||
|  | ||||
| :copyright: (c) 2020 sh-edraft.de | ||||
| :license: MIT, see LICENSE for more details. | ||||
|  | ||||
| """ | ||||
|  | ||||
| __title__ = 'sh_edraft.console.model' | ||||
| __author__ = 'Sven Heidemann' | ||||
| __license__ = 'MIT' | ||||
| __copyright__ = 'Copyright (c) 2020 sh-edraft.de' | ||||
| __version__ = '2020.12.9' | ||||
|  | ||||
| from collections import namedtuple | ||||
|  | ||||
| # imports: | ||||
| from .background_color import BackgroundColor | ||||
| from .foreground_color import ForegroundColor | ||||
|  | ||||
| VersionInfo = namedtuple('VersionInfo', 'major minor micro') | ||||
| version_info = VersionInfo(major=2020, minor=12, micro=9) | ||||
							
								
								
									
										14
									
								
								src_old/sh_edraft/console/model/background_color.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								src_old/sh_edraft/console/model/background_color.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | ||||
| from enum import Enum | ||||
|  | ||||
|  | ||||
| class BackgroundColor(Enum): | ||||
|  | ||||
|     default = 'on_default' | ||||
|     grey = 'on_grey' | ||||
|     red = 'on_red' | ||||
|     green = 'on_green' | ||||
|     yellow = 'on_yellow' | ||||
|     blue = 'on_blue' | ||||
|     magenta = 'on_magenta' | ||||
|     cyan = 'on_cyan' | ||||
|     white = 'on_white' | ||||
							
								
								
									
										14
									
								
								src_old/sh_edraft/console/model/foreground_color.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								src_old/sh_edraft/console/model/foreground_color.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | ||||
| from enum import Enum | ||||
|  | ||||
|  | ||||
| class ForegroundColor(Enum): | ||||
|  | ||||
|     default = 'default' | ||||
|     grey = 'grey' | ||||
|     red = 'red' | ||||
|     green = 'green' | ||||
|     yellow = 'yellow' | ||||
|     blue = 'blue' | ||||
|     magenta = 'magenta' | ||||
|     cyan = 'cyan' | ||||
|     white = 'white' | ||||
		Reference in New Issue
	
	Block a user