xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-17 11:00:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' into hl/gitlab_fix

Browse Source
This commit is contained in:
Hussam Lawen
2023-07-14 22:48:13 +03:00
committed by GitHub
parent 3f75b14ba3 ae9cedd50d
commit 4a6bf4c55a
14 changed files with 155 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/git_providers/git_provider.py
View File

@ -25,6 +25,10 @@ class GitProvider(ABC):
def get_diff_files(self) -> list[FilePatchInfo]:
pass
@abstractmethod
def publish_description(self, pr_title: str, pr_body: str):
pass
@abstractmethod
def publish_comment(self, pr_comment: str, is_temporary: bool = False):
pass

6
pr_agent/git_providers/github_provider.py
View File

@ -28,8 +28,6 @@ class GithubProvider:
self.pr = self._get_pr()
def get_files(self):
if hasattr(self.pr, 'files'):
return self.pr.files
return self.pr.get_files()
def get_diff_files(self) -> list[FilePatchInfo]:
@ -42,6 +40,10 @@ class GithubProvider:
self.diff_files = diff_files
return diff_files
def publish_description(self, pr_title: str, pr_body: str):
self.pr.edit(title=pr_title, body=pr_body)
# self.pr.create_issue_comment(pr_comment)
def publish_comment(self, pr_comment: str, is_temporary: bool = False):
response = self.pr.create_issue_comment(pr_comment)
if hasattr(response, "user") and hasattr(response.user, "login"):

4
pr_agent/git_providers/gitlab_provider.py
View File

@ -71,6 +71,10 @@ class GitLabProvider(GitProvider):
def get_files(self):
return [change['new_path'] for change in self.mr.changes()['changes']]
def publish_description(self, pr_title: str, pr_body: str):
logging.exception("Not implemented yet")
pass
def publish_comment(self, mr_comment: str, is_temporary: bool = False):
comment = self.mr.notes.create({'body': mr_comment})
if is_temporary: