Browse Source

fix: git_helper's tqdm encoding error

pull/175/head
Dr.Lt.Data 1 year ago
parent
commit
332e4e2856
  1. 15
      __init__.py
  2. 2
      git_helper.py
  3. 2
      prestartup_script.py

15
__init__.py

@ -11,14 +11,23 @@ import subprocess # don't remove this
from tqdm.auto import tqdm
version = "V0.39.2"
version = "V0.39.3"
print(f"### Loading: ComfyUI-Manager ({version})")
def handle_stream(stream, prefix):
stream.reconfigure(encoding=locale.getpreferredencoding(), errors='replace')
for line in stream:
print(prefix, line, end="")
for msg in stream:
if prefix == '[!]' and ('it/s]' in msg or 's/it]' in msg) and ('%|' in msg or 'it [' in msg):
if msg.startswith('100%'):
print('\r' + msg, end="", file=sys.stderr),
else:
print('\r' + msg[:-1], end="", file=sys.stderr),
else:
if prefix == '[!]':
print(prefix, msg, end="", file=sys.stderr)
else:
print(prefix, msg, end="")
def run_script(cmd, cwd='.'):

2
git_helper.py

@ -16,7 +16,7 @@ working_directory = os.getcwd()
class GitProgress(RemoteProgress):
def __init__(self):
super().__init__()
self.pbar = tqdm()
self.pbar = tqdm(ascii=True)
def update(self, op_code, cur_count, max_count=None, message=''):
self.pbar.total = max_count

2
prestartup_script.py

@ -28,7 +28,7 @@ git_script_path = os.path.join(comfyui_manager_path, "git_helper.py")
def handle_stream(stream, prefix):
stream.reconfigure(encoding=locale.getpreferredencoding(), errors='replace')
for msg in stream:
if prefix == '[!]' and ('it/s]' or 's/it]') in msg and ('%|' in msg or 'it [' in msg):
if prefix == '[!]' and ('it/s]' in msg or 's/it]' in msg) and ('%|' in msg or 'it [' in msg):
if msg.startswith('100%'):
print('\r' + msg, end="", file=sys.stderr),
else:

Loading…
Cancel
Save