diff options
| author | Martin Cairns <4314538+MartinCairnsSQL@users.noreply.github.com> | 2022-11-01 08:34:39 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-01 08:34:39 +0000 |
| commit | b88505925b58290f78a13f267d395ff3c2ca276c (patch) | |
| tree | 29be172302344c8d739cde2b3900dc5027a30028 /modules/extensions.py | |
| parent | 6c9e427c0eb6e48e367ea5c63c49ed8ac837b85e (diff) | |
| parent | dd02889124b2f89aa1d9acafedfa8774e53c98cb (diff) | |
| download | stable-diffusion-webui-gfx803-b88505925b58290f78a13f267d395ff3c2ca276c.tar.gz stable-diffusion-webui-gfx803-b88505925b58290f78a13f267d395ff3c2ca276c.tar.bz2 stable-diffusion-webui-gfx803-b88505925b58290f78a13f267d395ff3c2ca276c.zip | |
Merge branch 'AUTOMATIC1111:master' into adjust-ddim-uniform-steps
Diffstat (limited to 'modules/extensions.py')
| -rw-r--r-- | modules/extensions.py | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/modules/extensions.py b/modules/extensions.py new file mode 100644 index 00000000..897af96e --- /dev/null +++ b/modules/extensions.py @@ -0,0 +1,83 @@ +import os
+import sys
+import traceback
+
+import git
+
+from modules import paths, shared
+
+
+extensions = []
+extensions_dir = os.path.join(paths.script_path, "extensions")
+
+
+def active():
+ return [x for x in extensions if x.enabled]
+
+
+class Extension:
+ def __init__(self, name, path, enabled=True):
+ self.name = name
+ self.path = path
+ self.enabled = enabled
+ self.status = ''
+ self.can_update = False
+
+ repo = None
+ try:
+ if os.path.exists(os.path.join(path, ".git")):
+ repo = git.Repo(path)
+ except Exception:
+ print(f"Error reading github repository info from {path}:", file=sys.stderr)
+ print(traceback.format_exc(), file=sys.stderr)
+
+ if repo is None or repo.bare:
+ self.remote = None
+ else:
+ self.remote = next(repo.remote().urls, None)
+ self.status = 'unknown'
+
+ def list_files(self, subdir, extension):
+ from modules import scripts
+
+ dirpath = os.path.join(self.path, subdir)
+ if not os.path.isdir(dirpath):
+ return []
+
+ res = []
+ for filename in sorted(os.listdir(dirpath)):
+ res.append(scripts.ScriptFile(self.path, filename, os.path.join(dirpath, filename)))
+
+ res = [x for x in res if os.path.splitext(x.path)[1].lower() == extension and os.path.isfile(x.path)]
+
+ return res
+
+ def check_updates(self):
+ repo = git.Repo(self.path)
+ for fetch in repo.remote().fetch("--dry-run"):
+ if fetch.flags != fetch.HEAD_UPTODATE:
+ self.can_update = True
+ self.status = "behind"
+ return
+
+ self.can_update = False
+ self.status = "latest"
+
+ def pull(self):
+ repo = git.Repo(self.path)
+ repo.remotes.origin.pull()
+
+
+def list_extensions():
+ extensions.clear()
+
+ if not os.path.isdir(extensions_dir):
+ return
+
+ for dirname in sorted(os.listdir(extensions_dir)):
+ path = os.path.join(extensions_dir, dirname)
+ if not os.path.isdir(path):
+ continue
+
+ extension = Extension(name=dirname, path=path, enabled=dirname not in shared.opts.disabled_extensions)
+ extensions.append(extension)
|