github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-15 01:23:44 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix log_info/log_err printing with progress bars (#349)

Browse source
This commit is contained in:
Auroir 2019-08-17 00:20:10 -07:00 • committed by iperov
parent c4e68ef539
commit a906f24a4d
1 changed files with 19 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

22
interact/interact.py
View file

@ -33,6 +33,7 @@ class InteractBase(object):
self.key_events = {} self.key_events = {}
self.pg_bar = None self.pg_bar = None
self.focus_wnd_name = None self.focus_wnd_name = None
self.error_log_line_prefix = '/!\\ '
def is_support_windows(self): def is_support_windows(self):
return False return False
@ -65,10 +66,22 @@ class InteractBase(object):
raise NotImplemented raise NotImplemented
def log_info(self, msg, end='\n'): def log_info(self, msg, end='\n'):
if self.pg_bar is not None:
try: # Attempt print before the pb
tqdm.write(msg)
return
except:
pass #Fallback to normal print upon failure
print (msg, end=end) print (msg, end=end)
def log_err(self, msg, end='\n'): def log_err(self, msg, end='\n'):
print (msg, end=end) if self.pg_bar is not None:
try: # Attempt print before the pb
tqdm.write(f'{self.error_log_line_prefix}{msg}')
return
except:
pass #Fallback to normal print upon failure
print (f'{self.error_log_line_prefix}{msg}', end=end)
def named_window(self, wnd_name): def named_window(self, wnd_name):
if wnd_name not in self.named_windows: if wnd_name not in self.named_windows:
@ -150,9 +163,12 @@ class InteractBase(object):
else: print("progress_bar not set.") else: print("progress_bar not set.")
def progress_bar_generator(self, data, desc, leave=True): def progress_bar_generator(self, data, desc, leave=True):
for x in tqdm( data, desc=desc, leave=leave, ascii=True ): self.pg_bar = tqdm( data, desc=desc, leave=leave, ascii=True )
for x in self.pg_bar:
yield x yield x
self.pg_bar.close()
self.pg_bar = None
def process_messages(self, sleep_time=0): def process_messages(self, sleep_time=0):
self.on_process_messages(sleep_time) self.on_process_messages(sleep_time)