diff options
Diffstat (limited to '')
| -rw-r--r-- | cgitize/bitbucket.py | 6 | ||||
| -rw-r--r-- | cgitize/config.py | 30 | ||||
| -rw-r--r-- | cgitize/github.py | 6 |
3 files changed, 30 insertions, 12 deletions
diff --git a/cgitize/bitbucket.py b/cgitize/bitbucket.py index eaa8bb5..052d2c1 100644 --- a/cgitize/bitbucket.py +++ b/cgitize/bitbucket.py @@ -8,6 +8,8 @@ import logging from atlassian.bitbucket.cloud import Cloud from requests.exceptions import HTTPError +from cgitize.repo import Repo + class Bitbucket: def __init__(self, username=None, password=None): @@ -33,3 +35,7 @@ class Bitbucket: except HTTPError: logging.error("Couldn't fetch user repositories: %s", user.name) raise + + @staticmethod + def convert_repo(repo, *args, **kwargs): + return Repo.from_bitbucket(repo, *args, **kwargs) diff --git a/cgitize/config.py b/cgitize/config.py index 9f515c9..036b855 100644 --- a/cgitize/config.py +++ b/cgitize/config.py @@ -119,6 +119,10 @@ class User: def dir(self): return self._impl.get('dir') + @property + def skip(self): + return self._impl.get('skip', []) + class HostedRepo: def __init__(self, impl): @@ -155,23 +159,25 @@ class Config: for r in self.repositories.enum_repositories(): yield Repo.from_config(r, self) - def _parse_github_repositories(self): - github = GitHub(self.github.access_token) - for r in self.github.repositories.enum_repositories(): + def _parse_hosted_repositories(self, cfg, api): + for r in cfg.repositories.enum_repositories(): r = HostedRepo(r) - yield Repo.from_github(github.get_repo(r), self, r.dir) - for u in self.github.users.enum_users(): + yield api.convert_repo(api.get_repo(r), self, r.dir) + for u in cfg.users.enum_users(): u = User(u) - yield from (Repo.from_github(r, self, u.dir) for r in github.get_user_repos(u)) + for r in api.get_user_repos(u): + r = api.convert_repo(r, self, u.dir) + if r.name in u.skip: + continue + yield r + + def _parse_github_repositories(self): + github = GitHub(self.github.access_token) + return self._parse_hosted_repositories(self.github, github) def _parse_bitbucket_repositories(self): bitbucket = Bitbucket(self.bitbucket.username, self.bitbucket.app_password) - for r in self.bitbucket.repositories.enum_repositories(): - r = HostedRepo(r) - yield Repo.from_bitbucket(bitbucket.get_repo(r), self, r.dir) - for u in self.bitbucket.users.enum_users(): - u = User(u) - yield from (Repo.from_bitbucket(r, self, u.dir) for r in bitbucket.get_user_repos(u)) + return self._parse_hosted_repositories(self.bitbucket, bitbucket) def parse_repositories(self): yield from self._parse_explicit_repositories() diff --git a/cgitize/github.py b/cgitize/github.py index a5754af..761b2b7 100644 --- a/cgitize/github.py +++ b/cgitize/github.py @@ -7,6 +7,8 @@ import logging from github import Github, GithubException +from cgitize.repo import Repo + class GitHub: def __init__(self, access_token): @@ -25,3 +27,7 @@ class GitHub: except GithubException: logging.error("Couldn't fetch user repositories: %s", user.name) raise + + @staticmethod + def convert_repo(repo, *args, **kwargs): + return Repo.from_github(repo, *args, **kwargs) |