| import subprocess | |
| from pathlib import Path | |
| new_extensions = set() | |
| def clone_or_pull_repository(github_url): | |
| global new_extensions | |
| repository_folder = Path("extensions") | |
| repo_name = github_url.rstrip("/").split("/")[-1].split(".")[0] | |
| # Check if the repository folder exists | |
| if not repository_folder.exists(): | |
| repository_folder.mkdir(parents=True) | |
| repo_path = repository_folder / repo_name | |
| # Check if the repository is already cloned | |
| if repo_path.exists(): | |
| yield f"Updating {github_url}..." | |
| # Perform a 'git pull' to update the repository | |
| try: | |
| pull_output = subprocess.check_output(["git", "-C", repo_path, "pull"], stderr=subprocess.STDOUT) | |
| yield "Done." | |
| return pull_output.decode() | |
| except subprocess.CalledProcessError as e: | |
| return str(e) | |
| # Clone the repository | |
| try: | |
| yield f"Cloning {github_url}..." | |
| clone_output = subprocess.check_output(["git", "clone", github_url, repo_path], stderr=subprocess.STDOUT) | |
| new_extensions.add(repo_name) | |
| yield f"The extension `{repo_name}` has been downloaded.\n\nPlease close the the web UI completely and launch it again to be able to load it." | |
| return clone_output.decode() | |
| except subprocess.CalledProcessError as e: | |
| return str(e) | |