aljaz/Interlace
1
1
Fork 0
mirror of https://github.com/codingo/Interlace.git synced 2025-12-17 23:04:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fixed the sober flag

Browse Source
This commit is contained in:
prodigysml
2019-06-03 22:26:46 -07:00
parent f699668cf2
commit 20bf16305d
2 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Interlace/lib/core/input.py
View File

@@ -355,7 +355,7 @@ class InputParser(object):
) )
parser.add_argument( parser.add_argument(
'--no-bar', '--sober', dest='sober', action='store_true', default=True, '--no-bar', '--sober', dest='sober', action='store_true', default=False,
help='If set then progress bar will be stripped out' help='If set then progress bar will be stripped out'
) )

21
Interlace/lib/threader.py
View File

@@ -16,17 +16,22 @@ class Worker(object):
try: try:
# get task from queue # get task from queue
task = self.queue.pop(0) task = self.queue.pop(0)
if self.tqdm: if isinstance(self.tqdm, tqdm):
self.tqdm.update(1) self.tqdm.update(1)
# run task # run task
self.run_task(task, self.tqdm) self.run_task(task, self.tqdm)
else:
self.run_task(task)
except IndexError: except IndexError:
break break
@staticmethod @staticmethod
def run_task(task, t): def run_task(task, t=False):
s = subprocess.Popen(task, shell=True, stdout=subprocess.PIPE) if t:
t.write(s.stdout.readline().decode("utf-8")) s = subprocess.Popen(task, shell=True, stdout=subprocess.PIPE)
t.write(s.stdout.readline().decode("utf-8"))
else:
subprocess.Popen(task, shell=True)
class Pool(object): class Pool(object):
@@ -48,10 +53,10 @@ class Pool(object):
self.output = output self.output = output
self.max_workers = max_workers self.max_workers = max_workers
if progress_bar: if not progress_bar:
self.tqdm = tqdm(total=len(queue)) self.tqdm = tqdm(total=len(queue))
else: else:
self.tqdm = False self.tqdm = True
def run(self): def run(self):