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

Support repo-specific configuration file

Browse Source
This commit is contained in:
Ori Kotek
2023-08-01 17:44:08 +03:00
parent 696e2bd6ff
commit e12874b696
3 changed files with 22 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
pr_agent/git_providers/gitlab_provider.py
View File

@ -1,7 +1,5 @@
import logging
import os
import re
import tempfile
from typing import Optional, Tuple
from urllib.parse import urlparse
@ -37,17 +35,6 @@ class GitLabProvider(GitProvider):
self.RE_HUNK_HEADER = re.compile(
r"^@@ -(\d+)(?:,(\d+))? \+(\d+)(?:,(\d+))? @@[ ]?(.*)")
self.incremental = incremental
if get_settings().config.use_repo_settings_file:
repo_settings = self.get_repo_settings()
if repo_settings:
repo_settings_file = None
try:
fd, repo_settings_file = tempfile.mkstemp(suffix='.toml')
os.write(fd, repo_settings.encode())
get_settings().load_file(repo_settings_file)
finally:
if repo_settings_file:
os.remove(repo_settings_file)
def is_supported(self, capability: str) -> bool:
if capability in ['get_issue_comments', 'create_inline_comment', 'publish_inline_comments']:
@ -268,7 +255,7 @@ class GitLabProvider(GitProvider):
def get_repo_settings(self):
try:
contents = self.gl.projects.get(self.id_project).files.get(file_path='.pr_agent.toml', ref=self.mr.source_branch).decode()
contents = self.gl.projects.get(self.id_project).files.get(file_path='.pr_agent.toml', ref=self.mr.source_branch)
return contents
except Exception:
return ""