|
| 1 | +import httpx |
| 2 | +from pydantic import TypeAdapter, validate_call |
| 3 | + |
| 4 | +from ...models import Client |
| 5 | +from ..models import Error |
| 6 | +from ..models.v3 import FilesystemFile |
| 7 | + |
| 8 | + |
| 9 | +@validate_call() |
| 10 | +def _build_request( |
| 11 | + client: Client, |
| 12 | + storage: str, |
| 13 | + glob: str = "", |
| 14 | + size_min: str = "", |
| 15 | + size_max: str = "", |
| 16 | + lastmod_start: str = "", |
| 17 | + lastmod_end: str = "", |
| 18 | + sort: str = "", |
| 19 | + order: str = "", |
| 20 | + core_id: str = None, |
| 21 | + retries: int = None, |
| 22 | + timeout: float = None, |
| 23 | +): |
| 24 | + """_summary_ |
| 25 | + Args: |
| 26 | + glob (str): glob pattern for file names |
| 27 | + sort (str): none, name, size or lastmod |
| 28 | + order (str): asc or desc |
| 29 | + """ |
| 30 | + if not retries: |
| 31 | + retries = client.retries |
| 32 | + if not timeout: |
| 33 | + timeout = client.timeout |
| 34 | + return { |
| 35 | + "method": "get", |
| 36 | + "url": f"{client.base_url}/api/v3/cluster/fs/{storage}" |
| 37 | + + f"?glob={glob}&size_min={size_min}&size_max={size_max}" |
| 38 | + + f"&lastmod_start={lastmod_start}&lastmod_end={lastmod_end}" |
| 39 | + + f"&sort={sort}&order={order}", |
| 40 | + "headers": client.headers, |
| 41 | + "timeout": timeout, |
| 42 | + "data": None, |
| 43 | + "json": None, |
| 44 | + "params": {"core_id": core_id}, |
| 45 | + }, retries |
| 46 | + |
| 47 | + |
| 48 | +def _build_response(response: httpx.Response): |
| 49 | + if response.status_code == 200: |
| 50 | + response_200 = TypeAdapter(list[FilesystemFile]).validate_python(response.json()) |
| 51 | + return response_200 |
| 52 | + else: |
| 53 | + response_error = TypeAdapter(Error).validate_python(response.json()) |
| 54 | + return response_error |
| 55 | + |
| 56 | + |
| 57 | +def sync(client: Client, **kwargs): |
| 58 | + request, retries = _build_request(client, **kwargs) |
| 59 | + transport = httpx.HTTPTransport(retries=retries) |
| 60 | + httpx_client = httpx.Client(transport=transport, http2=True) |
| 61 | + response = httpx_client.request(**request) |
| 62 | + return _build_response(response=response) |
| 63 | + |
| 64 | + |
| 65 | +async def asyncio(client: Client, **kwargs): |
| 66 | + request, retries = _build_request(client, **kwargs) |
| 67 | + transport = httpx.AsyncHTTPTransport(retries=retries) |
| 68 | + async with httpx.AsyncClient(transport=transport) as httpx_client: |
| 69 | + response = await httpx_client.request(**request) |
| 70 | + return _build_response(response=response) |
0 commit comments