133 lines
4.8 KiB
Python
133 lines
4.8 KiB
Python
from common.log.module_logger import ModuleLogger
|
|
|
|
from typing import List, Dict, Optional
|
|
|
|
from common.config.app_settings import app_settings
|
|
|
|
import httpx
|
|
import asyncio
|
|
|
|
|
|
async def fetch_with_retry(url, method="GET", retries=3, backoff=1.0, **kwargs):
|
|
"""
|
|
A generic function for making HTTP requests with retry logic.
|
|
|
|
Parameters:
|
|
url (str): The endpoint URL.
|
|
method (str): HTTP method ('GET', 'POST', etc.).
|
|
retries (int): Number of retry attempts.
|
|
backoff (float): Backoff time in seconds.
|
|
kwargs: Additional arguments for the request.
|
|
|
|
Returns:
|
|
httpx.Response: The response object.
|
|
"""
|
|
for attempt in range(retries):
|
|
try:
|
|
async with httpx.AsyncClient(timeout=httpx.Timeout(10.0)) as client:
|
|
if method.upper() == "GET":
|
|
response = await client.get(url, **kwargs)
|
|
elif method.upper() == "POST":
|
|
response = await client.post(url, **kwargs)
|
|
response.raise_for_status() # Check for HTTP errors
|
|
return response
|
|
except (httpx.ReadTimeout, httpx.RequestError) as exc:
|
|
if attempt < retries - 1:
|
|
await asyncio.sleep(backoff * (2**attempt)) # Exponential backoff
|
|
continue
|
|
else:
|
|
raise exc
|
|
|
|
|
|
class CodeDepotService:
|
|
def __init__(self) -> None:
|
|
self.depot_endpoint = (
|
|
app_settings.DEVSVC_WEBAPI_URL_BASE.rstrip("/") + "/depot/"
|
|
)
|
|
self.module_logger = ModuleLogger(sender_id="CodeDepotService")
|
|
|
|
async def check_depot_name_availabe(self, code_depot_name: str) -> bool:
|
|
api_url = self.depot_endpoint + "check-depot-name-available/" + code_depot_name
|
|
response = await fetch_with_retry(api_url)
|
|
return response.json()
|
|
|
|
async def create_code_depot(self, product_id, code_depot_name) -> Optional[str]:
|
|
api_url = self.depot_endpoint + "create-code-depot"
|
|
response = await fetch_with_retry(
|
|
api_url,
|
|
method="POST",
|
|
json={"product_id": product_id, "code_depot_name": code_depot_name},
|
|
)
|
|
return response.json()
|
|
|
|
async def get_depot_ssh_url(self, code_depot_name: str) -> str:
|
|
api_url = self.depot_endpoint + "get-depot-ssh-url/" + code_depot_name
|
|
response = await fetch_with_retry(api_url)
|
|
return response.json()
|
|
|
|
async def get_depot_http_url(self, code_depot_name: str) -> str:
|
|
api_url = self.depot_endpoint + "get-depot-http-url/" + code_depot_name
|
|
response = await fetch_with_retry(api_url)
|
|
return response.json()
|
|
|
|
async def get_depot_http_url_with_user_name(
|
|
self, code_depot_name: str, user_name: str
|
|
) -> str:
|
|
api_url = (
|
|
self.depot_endpoint
|
|
+ "get-depot-http-url-with-user-name/"
|
|
+ code_depot_name
|
|
+ "/"
|
|
+ user_name
|
|
)
|
|
response = await fetch_with_retry(api_url)
|
|
return response.json()
|
|
|
|
async def get_depot_users(self, code_depot_name: str) -> List[str]:
|
|
api_url = self.depot_endpoint + "get-depot-users/" + code_depot_name
|
|
response = await fetch_with_retry(api_url)
|
|
return response.json()
|
|
|
|
async def update_depot_user_password(self, user_name: str, password: str) -> bool:
|
|
api_url = self.depot_endpoint + "update-depot-password-for-user"
|
|
response = await fetch_with_retry(
|
|
api_url,
|
|
method="POST",
|
|
json={"user_name": user_name, "password": password},
|
|
)
|
|
return response.json()
|
|
|
|
async def create_depot_user(
|
|
self, user_name: str, password: str, email: str
|
|
) -> bool:
|
|
api_url = self.depot_endpoint + "create-depot-user"
|
|
response = await fetch_with_retry(
|
|
api_url,
|
|
method="POST",
|
|
json={"user_name": user_name, "password": password, "email": email},
|
|
)
|
|
return response.json()
|
|
|
|
async def grant_user_depot_access(
|
|
self, user_name: str, code_depot_name: str
|
|
) -> bool:
|
|
api_url = self.depot_endpoint + "grant-user-depot-access"
|
|
response = await fetch_with_retry(
|
|
api_url,
|
|
method="POST",
|
|
json={"user_name": user_name, "code_depot_name": code_depot_name},
|
|
)
|
|
return response.json()
|
|
|
|
async def generate_statistic_result(
|
|
self, code_depot_name: str
|
|
) -> Optional[Dict[str, any]]:
|
|
api_url = self.depot_endpoint + "generate-statistic-result/" + code_depot_name
|
|
response = await fetch_with_retry(api_url)
|
|
return response.json()
|
|
|
|
async def fetch_code_depot(self, code_depot_id: str) -> Optional[Dict[str, any]]:
|
|
api_url = self.depot_endpoint + "fetch-code-depot/" + code_depot_id
|
|
response = await fetch_with_retry(api_url)
|
|
return response.json()
|