import os.path class Repo: DEFAULT_OWNER = None @staticmethod def extract_repo_name(repo_id): return os.path.basename(repo_id) def __init__(self, repo_id, clone_url, owner=None, desc=None, homepage=None): self.repo_id = repo_id self.repo_name = self.extract_repo_name(repo_id) self.clone_url = clone_url if owner is None: owner = Repo.DEFAULT_OWNER self.owner = owner if desc is None: if homepage is not None: desc = homepage elif clone_url is not None: desc = clone_url else: desc = self.repo_name self.desc = desc self.homepage = homepage class GithubRepo(Repo): DEFAULT_USER = None def __init__(self, repo_id, clone_url=None, owner=None, desc=None, homepage=None, user=DEFAULT_USER, via_ssh=True): if user is None: if GithubRepo.DEFAULT_USER is None: raise RuntimeError('neither explicit or default GitHub username was provided') user = GithubRepo.DEFAULT_USER name = Repo.extract_repo_name(repo_id) if clone_url is None: if via_ssh: clone_url = self.build_clone_url_ssh(user, name) else: clone_url = self.build_clone_url_https(user, name) if homepage is None: homepage = self.build_homepage_url(user, name) super().__init__(repo_id, clone_url, owner=owner, desc=desc, homepage=homepage) @staticmethod def build_clone_url_ssh(user, name): return f'ssh://git@github.com/{user}/{name}.git' @staticmethod def build_clone_url_https(user, name): return f'https://github.com/{user}/{name}.git' @staticmethod def build_homepage_url(user, name): return f'https://github.com/{user}/{name}' class BitbucketRepo(Repo): DEFAULT_USER = None def __init__(self, repo_id, clone_url=None, owner=None, desc=None, homepage=None, user=DEFAULT_USER, via_ssh=True): if user is None: if BitbucketRepo.DEFAULT_USER is None: raise RuntimeError('neither explicit or default Bitbucket username was provided') user = BitbucketRepo.DEFAULT_USER name = Repo.extract_repo_name(repo_id) if clone_url is None: if via_ssh: clone_url = self.build_clone_url_ssh(user, name) else: clone_url = self.build_clone_url_https(user, name) if homepage is None: homepage = self.build_homepage_url(user, name) super().__init__(repo_id, clone_url, owner=owner, desc=desc, homepage=homepage) @staticmethod def build_clone_url_ssh(user, name): return f'ssh://git@bitbucket.org/{user}/{name}.git' @staticmethod def build_clone_url_https(user, name): return f'https://bitbucket.org/{user}/{name}.git' @staticmethod def build_homepage_url(user, name): return f'https://bitbucket.org/{user}/{name.lower()}'