122 lines
4.0 KiB
Python
122 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
# IMPORTS
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
from src.local.io import *;
|
|
from src.local.misc import *;
|
|
from src.local.system import *;
|
|
from src.local.typing import *;
|
|
|
|
from src.core.utils import StripAnsi;
|
|
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
# GLOBAL VARIABLES
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
_logging_prefix: str = '';
|
|
_quietmode: bool = False;
|
|
_debugmode: bool = False;
|
|
_ansimode: bool = False;
|
|
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
# METHOD get/set quiet mode, logging depth, timer
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
def GetQuietMode() -> bool:
|
|
return _quietmode;
|
|
|
|
def SetQuietMode(mode: bool):
|
|
global _quietmode;
|
|
_quietmode = mode;
|
|
return;
|
|
|
|
def GetDebugMode() -> bool:
|
|
return _debugmode;
|
|
|
|
def SetDebugMode(mode: bool):
|
|
global _debugmode;
|
|
_debugmode = mode;
|
|
return;
|
|
|
|
def GetAnsiMode() -> bool:
|
|
return _ansimode;
|
|
|
|
def SetAnsiMode(mode: bool):
|
|
global _ansimode;
|
|
_ansimode = mode;
|
|
return;
|
|
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
# Logging
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
def logGeneric(tag: str, *lines: Any, file: io.TextIOWrapper, force: bool = False, tag_all: bool = True):
|
|
if not force and _quietmode:
|
|
return;
|
|
tag = '' if tag == '' else tag + ' ';
|
|
file = file or sys.stdout;
|
|
for line in lines:
|
|
line = '{}{}{}'.format('', tag, line);
|
|
if not _ansimode:
|
|
line = StripAnsi(line);
|
|
print(line, file=file);
|
|
if not tag_all:
|
|
tag = '';
|
|
return;
|
|
|
|
def logPlain(*lines: Any, force: bool = False, file: Any = None):
|
|
logGeneric('', *lines, force=force, file=file or sys.stdout);
|
|
|
|
def logInfo(*lines: Any, force: bool = False, tag_all: bool = True, file: Any = None):
|
|
logGeneric('[\033[94;1mINFO\033[0m]', *lines, force=force, tag_all=tag_all, file=file or sys.stdout);
|
|
|
|
def logDebug(*lines: Any, force: bool = False, tag_all: bool = True, file: Any = None):
|
|
if not _debugmode:
|
|
return;
|
|
logGeneric('\033[2m[\033[96;1mDEBUG\033[0m\033[2m]\033[0m',
|
|
*[ '\033[2m{}\033[0m'.format(line) for line in lines ],
|
|
force=force, tag_all=tag_all, file=file or sys.stdout
|
|
);
|
|
|
|
def logWarn(*lines: Any, force: bool = False, tag_all: bool = False, file: Any = None):
|
|
logGeneric('[\033[93;1mWARNING\033[0m]', *lines, force=force, tag_all=tag_all, file=file or sys.stdout);
|
|
|
|
def logError(*lines: Any, force: bool = False, tag_all: bool = False, file: Any = None):
|
|
logGeneric('[\033[91;1mERROR\033[0m]', *lines, force=force, tag_all=tag_all, file=file or sys.stderr);
|
|
|
|
def logFatal(*lines: Any, force: bool = False, tag_all: bool = False, file: Any = None):
|
|
logGeneric('[\033[91;1mFATAL\033[0m]', *lines, force=force, tag_all=tag_all, file=file or sys.stderr);
|
|
exit(1);
|
|
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
# User Input
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
def askUserInput(message: str, expectedformat: Callable) -> Union[str, None]:
|
|
answer = None;
|
|
while True:
|
|
try:
|
|
answer = input('{}{}'.format(_logging_prefix, message));
|
|
## Meta+C erkennen:
|
|
except KeyboardInterrupt:
|
|
logPlain('');
|
|
return None;
|
|
## Meta+D erkennen:
|
|
except EOFError:
|
|
logPlain('');
|
|
return None;
|
|
except:
|
|
continue;
|
|
if expectedformat(answer):
|
|
break;
|
|
return answer;
|
|
|
|
def askConfirmation(message: str, default: bool = False) -> bool:
|
|
answer = askUserInput(message, lambda x: not not re.match(r'^(y|yes|j|ja|n|no|nein)$', x));
|
|
if isinstance(answer, str):
|
|
return True if re.match(r'^(y|yes|j|ja)$', answer) else False;
|
|
return default;
|