Make update check commandline option, give response on all paths.

1 files changed, 19 insertions, 10 deletions

diff --git a/launch.py b/launch.py
index 8484d7c1..9d9dd1b6 100644
--- a/launch.py
+++ b/launch.py
@@ -86,7 +86,24 @@ def git_clone(url, dir, name, commithash=None):
     if commithash is not None:

         run(f'"{git}" -C {dir} checkout {commithash}', None, "Couldn't checkout {name}'s hash: {commithash}")

 

+        

+def version_check(commit):

+    try:

+        import requests

+        commits = requests.get('https://api.github.com/repos/AUTOMATIC1111/stable-diffusion-webui/branches/master').json()

+        if commit != "<none>" and commits['commit']['sha'] != commit:

+            print("--------------------------------------------------------")

+            print("| You are not up to date with the most recent release. |")

+            print("| Consider running `git pull` to update.               |")

+            print("--------------------------------------------------------")

+        elif commits['commit']['sha'] == commit:

+            print("You are up to date with the most recent release.")

+        else:

+            print("Not a git clone, can't perform version check.")

+    except Exception as e:

+        print("versipm check failed",e)

 

+        

 def prepare_enviroment():

     torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113")

     requirements_file = os.environ.get('REQS_FILE', "requirements_versions.txt")

@@ -165,16 +182,8 @@ def prepare_enviroment():
 

     sys.argv += args

 

-    try:

-        import requests

-        commits = requests.get('https://api.github.com/repos/AUTOMATIC1111/stable-diffusion-webui/branches/master').json()

-        if commit != "<none>" and commits['commit']['sha'] != commit:

-            print("--------------------------------------------------------")

-            print("| You are not up to date with the most recent release. |")

-            print("| Consider running `git pull` to update.               |")

-            print("--------------------------------------------------------")

-    except Exception as e:

-        pass

+    if '--update-check' in args:

+        version_check(commit)

     

     if "--exit" in args:

         print("Exiting because of --exit argument")