mirror of
https://github.com/netbox-community/Device-Type-Library-Import.git
synced 2025-06-08 19:07:01 +02:00
8832370aba
* Remove Repo init from __init__ Initializing the repository from init, before the path has been updated, will lead to various errors when the script is not run directly from the repository. * Implemented fix that will allow for running script from any directory --------- Co-authored-by: Daniel W. Anner <daniel.anner@danstechsupport.com>
46 lines
1.9 KiB
Python
46 lines
1.9 KiB
Python
from git import Repo, exc
|
|
from sys import exit as system_exit
|
|
import settings
|
|
import os
|
|
|
|
class GitCMD:
|
|
def __new__(cls, *args, **kwargs):
|
|
return super().__new__(cls)
|
|
|
|
def __init__(self, args, repo_path):
|
|
self.url = args.url
|
|
self.repo_path = repo_path
|
|
self.branch = args.branch
|
|
# self.repo = Repo()
|
|
self.cwd = os.getcwd()
|
|
|
|
if os.path.isdir(self.repo_path):
|
|
self.pull_repo()
|
|
else:
|
|
self.clone_repo()
|
|
|
|
|
|
def pull_repo(self):
|
|
try:
|
|
print("Package devicetype-library is already installed, "
|
|
+ f"updating {os.path.join(self.cwd, self.repo_path)}")
|
|
self.repo = Repo(self.repo_path)
|
|
if not self.repo.remotes.origin.url.endswith('.git'):
|
|
settings.handle.exception("GitInvalidRepositoryError", self.repo.remotes.origin.url, f"Origin URL {self.repo.remotes.origin.url} does not end with .git")
|
|
self.repo.remotes.origin.pull()
|
|
self.repo.git.checkout(self.branch)
|
|
print(f"Pulled Repo {self.repo.remotes.origin.url}")
|
|
except exc.GitCommandError as git_error:
|
|
settings.handle.exception("GitCommandError", self.repo.remotes.origin.url, git_error)
|
|
except Exception as git_error:
|
|
settings.handle.exception("Exception", 'Git Repository Error', git_error)
|
|
|
|
def clone_repo(self):
|
|
try:
|
|
self.repo = Repo.clone_from(self.url, os.path.join(self.cwd, self.repo_path), branch=self.branch)
|
|
print(f"Package Installed {self.repo.remotes.origin.url}")
|
|
except exc.GitCommandError as git_error:
|
|
settings.handle.exception("GitCommandError", self.url, git_error)
|
|
except Exception as git_error:
|
|
settings.handle.exception("Exception", 'Git Repository Error', git_error)
|