mirror of
https://github.com/qodo-ai/pr-agent.git
synced 2025-07-04 21:00:40 +08:00
Extract repeated token retrieval logic into a helper function to reduce code duplication
This commit is contained in:
@ -34,15 +34,17 @@ class BitbucketProvider(GitProvider):
|
|||||||
try:
|
try:
|
||||||
self.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")
|
||||||
|
|
||||||
|
def get_token(token_name, auth_type_name):
|
||||||
|
token = context.get(f"bitbucket_{token_name}", None) or get_settings().get(f"BITBUCKET.{token_name.upper()}", None)
|
||||||
|
if not token:
|
||||||
|
raise ValueError(f"{auth_type_name} auth requires a token")
|
||||||
|
return token
|
||||||
|
|
||||||
if self.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 = get_token("basic_token", "Basic")
|
||||||
if not self.basic_token:
|
|
||||||
raise ValueError("Basic auth requires a token")
|
|
||||||
s.headers["Authorization"] = f"Basic {self.basic_token}"
|
s.headers["Authorization"] = f"Basic {self.basic_token}"
|
||||||
elif self.auth_type == "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 = get_token("bearer_token", "Bearer")
|
||||||
if not self.bearer_token:
|
|
||||||
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:
|
else:
|
||||||
raise ValueError(f"Unsupported auth_type: {self.auth_type}")
|
raise ValueError(f"Unsupported auth_type: {self.auth_type}")
|
||||||
|
Reference in New Issue
Block a user