Add & use modules.errors.print_error where currently printing exception info by hand

1 files changed, 3 insertions, 6 deletions

diff --git a/modules/ui_extensions.py b/modules/ui_extensions.py
index 515ec262..cadf56be 100644
--- a/modules/ui_extensions.py
+++ b/modules/ui_extensions.py
@@ -1,10 +1,8 @@
 import json

 import os.path

-import sys

 import threading

 import time

 from datetime import datetime

-import traceback

 

 import git

 

@@ -14,6 +12,7 @@ import shutil
 import errno

 

 from modules import extensions, shared, paths, config_states

+from modules.errors import print_error

 from modules.paths_internal import config_states_dir

 from modules.call_queue import wrap_gradio_gpu_call

 

@@ -46,8 +45,7 @@ def apply_and_restart(disable_list, update_list, disable_all):
         try:

             ext.fetch_and_reset_hard()

         except Exception:

-            print(f"Error getting updates for {ext.name}:", file=sys.stderr)

-            print(traceback.format_exc(), file=sys.stderr)

+            print_error(f"Error getting updates for {ext.name}", exc_info=True)

 

     shared.opts.disabled_extensions = disabled

     shared.opts.disable_all_extensions = disable_all

@@ -113,8 +111,7 @@ def check_updates(id_task, disable_list):
             if 'FETCH_HEAD' not in str(e):

                 raise

         except Exception:

-            print(f"Error checking updates for {ext.name}:", file=sys.stderr)

-            print(traceback.format_exc(), file=sys.stderr)

+            print_error(f"Error checking updates for {ext.name}", exc_info=True)

 

         shared.state.nextjob()