refined work of previous commit

This commit is contained in:
Kjistóf 2017-02-04 18:38:08 +01:00
parent 3974c5aa1c
commit a39bb01460

View File

@ -32,15 +32,15 @@ class File(Enum):
FRACTION = 3
OUTPUT = 4
def call_verbose(before_message=''):
def wrap(f):
def call_verbose(before_message='', after_message='Done!'):
def tag(f):
@wraps(f)
def wrapper(*args, **kwargs):
print_opt(before_message+' ', end='', flush=True)
print_opt(before_message, end='', flush=True)
f(*args, **kwargs)
print_opt('Done!')
print_opt(after_message)
return wrapper
return wrap
return tag
def print_opt(*args, **kwargs):
if VERBOSE: