Update bitbucket_server_provider.py

This commit is contained in:
Boston008 2025-08-13 12:46:50 +08:00 committed by GitHub
parent 01f16f57ca
commit e0a4be1397
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -41,28 +41,36 @@ class BitbucketServerProvider(GitProvider):
# Get username and password from settings
username = get_settings().get("BITBUCKET_SERVER.USERNAME", None)
password = get_settings().get("BITBUCKET_SERVER.PASSWORD", None)
self.bitbucket_server_url = self._parse_bitbucket_server(url=pr_url)
if not self.bitbucket_server_url:
# If bitbucket_client is provided, try to extract the server URL from its configuration
if bitbucket_client:
self.bitbucket_client = bitbucket_client
try:
# Get the base URL from the existing client
self.bitbucket_server_url = bitbucket_client.url
except AttributeError:
# If we can't get the URL from the client, try parsing it from PR URL
self.bitbucket_server_url = self._parse_bitbucket_server(url=pr_url) if pr_url else None
else:
# Parse server URL from PR URL
self.bitbucket_server_url = self._parse_bitbucket_server(url=pr_url) if pr_url else None
# Validate the server URL unless we have a pre-configured client
if not bitbucket_client and not self.bitbucket_server_url:
raise ValueError("Invalid or missing Bitbucket Server URL parsed from PR URL.")
# If bearer token is provided, use it to authenticate, otherwise use username and password
try:
# If no client provided, create one with the appropriate authentication
if not bitbucket_client:
if self.bearer_token:
self.bitbucket_client = bitbucket_client or Bitbucket(
self.bitbucket_client = Bitbucket(
url=self.bitbucket_server_url,
token=self.bearer_token
)
else:
if not username or not password:
raise ValueError("Bitbucket authentication requires either 'BITBUCKET_SERVER.BEARER_TOKEN' or both 'BITBUCKET_SERVER.USERNAME' and 'BITBUCKET_SERVER.PASSWORD'.")
self.bitbucket_client = bitbucket_client or Bitbucket(
self.bitbucket_client = Bitbucket(
url=self.bitbucket_server_url,
username=username,
password=password
)
except Exception as e:
get_logger().error(f"Failed to initialize Bitbucket client for {self.bitbucket_server_url}: {e}")
raise
try:
self.bitbucket_api_version = parse_version(self.bitbucket_client.get("rest/api/1.0/application-properties").get('version'))
except Exception: