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

Bitbucket server, WIP

Browse Source
This commit is contained in:
Ori Kotek
2023-08-24 16:33:51 +03:00
parent 123741faf3
commit 5079daa4ad
8 changed files with 145 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pr_agent/git_providers/bitbucket_provider.py
View File

@ -5,6 +5,7 @@ from urllib.parse import urlparse
import requests
from atlassian.bitbucket import Cloud
from starlette_context import context
from ..config_loader import get_settings
from .git_provider import FilePatchInfo, GitProvider
@ -13,7 +14,11 @@ from .git_provider import FilePatchInfo, GitProvider
class BitbucketProvider(GitProvider):
def __init__(self, pr_url: Optional[str] = None, incremental: Optional[bool] = False):
s = requests.Session()
s.headers['Authorization'] = f'Bearer {get_settings().get("BITBUCKET.BEARER_TOKEN", None)}'
try:
bearer = context.get("bitbucket_bearer_token", None)
s.headers['Authorization'] = f'Bearer {bearer}'
except Exception:
s.headers['Authorization'] = f'Bearer {get_settings().get("BITBUCKET.BEARER_TOKEN", None)}'
s.headers['Content-Type'] = 'application/json'
self.headers = s.headers
self.bitbucket_client = Cloud(session=s)