diff options
| author | random-thoughtss <116161560+random-thoughtss@users.noreply.github.com> | 2022-11-03 15:55:54 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-03 15:55:54 -0700 |
| commit | 243253ff4a8ae944ba142abe9c1e78a92dd14ebe (patch) | |
| tree | c40402e18a29ca9a9b167a2f9e47dab39dce0943 /modules/extensions.py | |
| parent | d9e4e4d7a09d4aee8ce249a3c8e91ce165b10fa5 (diff) | |
| parent | 20a860b525cb7a319a42994f75a94bbca9a54d89 (diff) | |
Merge branch 'AUTOMATIC1111:master' into master
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)
|