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

Update pr_agent/git_providers/bitbucket_provider.py

Browse Source 
Co-authored-by: Prateek <110811408+Prateikx@users.noreply.github.com>
This commit is contained in:
ChunTing Wu
2025-04-11 14:55:06 +08:00
committed by GitHub
parent ca95e876eb
commit c0c307503f
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pr_agent/git_providers/bitbucket_provider.py
View File

@ -32,18 +32,20 @@ class BitbucketProvider(GitProvider):
s.headers["Content-Type"] = "application/json" s.headers["Content-Type"] = "application/json"
try: try:
auth_type = context.get("bitbucket_auth_type", None) or get_settings().get("BITBUCKET.AUTH_TYPE", "bearer") self.auth_type = context.get("bitbucket_auth_type", None) or get_settings().get("BITBUCKET.AUTH_TYPE", "bearer")
if auth_type == "basic": if self.auth_type == "basic":
self.basic_token = context.get("bitbucket_basic_token", None) or get_settings().get("BITBUCKET.BASIC_TOKEN", None) self.basic_token = context.get("bitbucket_basic_token", None) or get_settings().get("BITBUCKET.BASIC_TOKEN", None)
if not self.basic_token: if not self.basic_token:
raise ValueError("Basic auth requires a token") raise ValueError("Basic auth requires a token")
s.headers["Authorization"] = f"Basic {self.basic_token}" s.headers["Authorization"] = f"Basic {self.basic_token}"
else: # default to bearer elif self.auth_type == "bearer":
self.bearer_token = context.get("bitbucket_bearer_token", None) or get_settings().get("BITBUCKET.BEARER_TOKEN", None) self.bearer_token = context.get("bitbucket_bearer_token", None) or get_settings().get("BITBUCKET.BEARER_TOKEN", None)
if not self.bearer_token: if not self.bearer_token:
raise ValueError("Bearer token is required for bearer auth") raise ValueError("Bearer token is required for bearer auth")
s.headers["Authorization"] = f"Bearer {self.bearer_token}" s.headers["Authorization"] = f"Bearer {self.bearer_token}"
else:
raise ValueError(f"Unsupported auth_type: {self.auth_type}")
except Exception as e: except Exception as e:
get_logger().exception(f"Failed to initialize Bitbucket authentication: {e}") get_logger().exception(f"Failed to initialize Bitbucket authentication: {e}")