use extension name for determining an extension is installed in the index

1 files changed, 8 insertions, 6 deletions

diff --git a/modules/ui_extensions.py b/modules/ui_extensions.py
index c0a73b57..b6708881 100644
--- a/modules/ui_extensions.py
+++ b/modules/ui_extensions.py
@@ -335,6 +335,11 @@ def normalize_git_url(url):
     return url

 

 

+def get_extension_dirname_from_url(url):

+    *parts, last_part = url.split('/')

+    return normalize_git_url(last_part)

+

+

 def install_extension_from_url(dirname, url, branch_name=None):

     check_access()

 

@@ -346,10 +351,7 @@ def install_extension_from_url(dirname, url, branch_name=None):
     assert url, 'No URL specified'

 

     if dirname is None or dirname == "":

-        *parts, last_part = url.split('/')

-        last_part = normalize_git_url(last_part)

-

-        dirname = last_part

+        dirname = get_extension_dirname_from_url(url)

 

     target_dir = os.path.join(extensions.extensions_dir, dirname)

     assert not os.path.exists(target_dir), f'Extension directory already exists: {target_dir}'

@@ -449,7 +451,7 @@ def get_date(info: dict, key):
 

 def refresh_available_extensions_from_data(hide_tags, sort_column, filter_text=""):

     extlist = available_extensions["extensions"]

-    installed_extension_urls = {normalize_git_url(extension.remote): extension.name for extension in extensions.extensions}

+    installed_extensions = {extension.name for extension in extensions.extensions}

 

     tags = available_extensions.get("tags", {})

     tags_to_hide = set(hide_tags)

@@ -482,7 +484,7 @@ def refresh_available_extensions_from_data(hide_tags, sort_column, filter_text="
         if url is None:

             continue

 

-        existing = installed_extension_urls.get(normalize_git_url(url), None)

+        existing = get_extension_dirname_from_url(url) in installed_extensions

         extension_tags = extension_tags + ["installed"] if existing else extension_tags

 

         if any(x for x in extension_tags if x in tags_to_hide):