| import os |
| import sys |
| import traceback |
|
|
| import time |
| import git |
|
|
| from modules import paths, shared |
|
|
| extensions = [] |
| extensions_dir = os.path.join(paths.data_path, "extensions") |
| extensions_builtin_dir = os.path.join(paths.script_path, "extensions-builtin") |
|
|
| if not os.path.exists(extensions_dir): |
| os.makedirs(extensions_dir) |
|
|
| def active(): |
| return [x for x in extensions if x.enabled] |
|
|
|
|
| class Extension: |
| def __init__(self, name, path, enabled=True, is_builtin=False): |
| self.name = name |
| self.path = path |
| self.enabled = enabled |
| self.status = '' |
| self.can_update = False |
| self.is_builtin = is_builtin |
| self.version = '' |
|
|
| 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: |
| try: |
| self.remote = next(repo.remote().urls, None) |
| self.status = 'unknown' |
| head = repo.head.commit |
| ts = time.asctime(time.gmtime(repo.head.commit.committed_date)) |
| self.version = f'{head.hexsha[:8]} ({ts})' |
|
|
| except Exception: |
| self.remote = None |
|
|
| 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 fetch_and_reset_hard(self): |
| repo = git.Repo(self.path) |
| |
| |
| repo.git.fetch('--all') |
| repo.git.reset('--hard', 'origin') |
|
|
|
|
| def list_extensions(): |
| extensions.clear() |
|
|
| if not os.path.isdir(extensions_dir): |
| return |
|
|
| paths = [] |
| for dirname in [extensions_dir, extensions_builtin_dir]: |
| if not os.path.isdir(dirname): |
| return |
|
|
| for extension_dirname in sorted(os.listdir(dirname)): |
| path = os.path.join(dirname, extension_dirname) |
| if not os.path.isdir(path): |
| continue |
|
|
| paths.append((extension_dirname, path, dirname == extensions_builtin_dir)) |
|
|
| for dirname, path, is_builtin in paths: |
| extension = Extension(name=dirname, path=path, enabled=dirname not in shared.opts.disabled_extensions, is_builtin=is_builtin) |
| extensions.append(extension) |
|
|
|
|
