Skip to content

chore(CG-10447): add github_type param for remote repo operator #58

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
christinewangcw merged 2 commits into from
Jan 24, 2025
Merged

chore(CG-10447): add github_type param for remote repo operator #58

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
5a6919f
chore(CG-10447): add github_type param for remote repo operator
christinewangcw Jan 24, 2025
aa2c128
add github type in git repo client
christinewangcw Jan 24, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 6 additions & 4 deletions src/codegen/git/repo_operator/remote_repo_operator.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ class RemoteRepoOperator(RepoOperator):
# __init__ attributes
repo_config: RepoConfig
base_dir: str
github_type: GithubType

# lazy attributes
_remote_git_repo: GitRepoClient | None = None
Expand All @@ -43,6 +44,7 @@ def __init__(
github_type: GithubType = GithubType.GithubEnterprise,
) -> None:
super().__init__(repo_config=repo_config, base_dir=base_dir)
self.github_type = github_type
self.setup_repo_dir(setup_option=setup_option, shallow=shallow)

####################################################################################################################
Expand All @@ -52,7 +54,7 @@ def __init__(
@property
def remote_git_repo(self) -> GitRepoClient:
if not self._remote_git_repo:
self._remote_git_repo = GitRepoClient(self.repo_config, access_scope=GithubScope.WRITE)
self._remote_git_repo = GitRepoClient(self.repo_config, github_type=self.github_type, access_scope=GithubScope.WRITE)
return self._remote_git_repo

@property
Expand All @@ -74,17 +76,17 @@ def codeowners_parser(self) -> CodeOwnersParser | None:
@override
def pull_repo(self) -> None:
"""Pull the latest commit down to an existing local repo"""
pull_repo(repo=self.repo_config, path=self.base_dir)
pull_repo(repo=self.repo_config, path=self.base_dir, github_type=self.github_type)

def clone_repo(self, shallow: bool = True) -> None:
clone_repo(repo=self.repo_config, path=self.base_dir, shallow=shallow)
clone_repo(repo=self.repo_config, path=self.base_dir, shallow=shallow, github_type=self.github_type)

def clone_or_pull_repo(self, shallow: bool = True) -> None:
"""If repo exists, pulls changes. otherwise, clones the repo."""
# TODO(CG-7804): if repo is not valid we should delete it and re-clone. maybe we can create a pull_repo util + use the existing clone_repo util
if self.repo_exists():
self.clean_repo()
clone_or_pull_repo(self.repo_config, path=self.base_dir, shallow=shallow)
clone_or_pull_repo(self.repo_config, path=self.base_dir, shallow=shallow, github_type=self.github_type)

def setup_repo_dir(self, setup_option: SetupOption = SetupOption.PULL_OR_CLONE, shallow: bool = True) -> None:
os.makedirs(self.base_dir, exist_ok=True)
Expand Down
Loading