Coverage for src/fastapi_cloud_cli/commands/domains/list.py: 100%

1from typing import Annotated, Any 7 ctx1abcdefg

2 

3import typer 7 ctx1abcdefg

4from pydantic import BaseModel 7 ctx1abcdefg

5from rich_toolkit import RichToolkit 7 ctx1abcdefg

6 

7from fastapi_cloud_cli.api import APIClient, CustomDomain 7 ctx1abcdefg

8from fastapi_cloud_cli.commands._auth import UserCommand, get_user_command_context 7 ctx1abcdefg

9from fastapi_cloud_cli.commands.domains._app import domains_app 7 ctx1abcdefg

10from fastapi_cloud_cli.commands.domains.rendering import get_custom_domains_table 7 ctx1abcdefg

11from fastapi_cloud_cli.utils.apps import resolve_app_id_or_fail 7 ctx1abcdefg

12from fastapi_cloud_cli.utils.execution import JsonOutputOption 7 ctx1abcdefg

13 

14 

15class CustomDomainsListOutput(BaseModel): 7 ctx1abcdefg

16 app_id: str 7 ctx1abcdefg

17 domains: list[CustomDomain] 7 ctx1abcdefg

18 total_count: int 7 ctx1abcdefg

19 

20 

21def _render_custom_domains_list_output( 10 ctx1abjcdkefgl

22 data: CustomDomainsListOutput, 

23 toolkit: RichToolkit, 

24) -> None: 

25 toolkit.print_title("custom domains") 14 ctx1hmnjotpkiqrusl

26 toolkit.print_line() 14 ctx1hmnjotpkiqrusl

27 

28 if not data.domains: 14 ctx1hmnjotpkiqrusl

29 toolkit.print("No custom domains found.", bullet=False) 7 ctx1hnopirs

30 return 7 ctx1hnopirs

31 

32 toolkit.print(get_custom_domains_table(data.domains), bullet=False) 7 ctx1mjtkqul

33 

34 

35@domains_app.command("list", cls=UserCommand) 7 ctx1abcdefg

36def list_domains( 10 ctx1abXcdYefgZ

37 ctx: typer.Context, 

38 app_id: Annotated[ 

39 str | None, 

40 typer.Option( 

41 "--app-id", 

42 help="ID of the app whose custom domains should be listed.", 

43 ), 

44 ] = None, 

45 json_output: JsonOutputOption = False, 

46) -> Any: 

47 """ 

48 List custom domains for an app. 

49 """ 

50 toolkit = get_user_command_context(ctx).toolkit 42 ctx1hQvxymnRzGHjoSAIJtpTBKLkiUwCDqrVEMNusWFOPl

51 app_id = resolve_app_id_or_fail(toolkit, app_id=app_id) 42 ctx1hQvxymnRzGHjoSAIJtpTBKLkiUwCDqrVEMNusWFOPl

52 

53 with APIClient() as client: 35 ctx1hvxymnzGHjoAIJtpBKLkiwCDqrEMNusFOPl

54 with ( 10 ctx1hvxymiwCDq

55 toolkit.progress( 

56 title="Fetching custom domains", 

57 transient=True, 

58 ) as progress, 

59 client.handle_http_errors( 

60 progress, 

61 default_message=( 

62 "Error fetching custom domains. Please try again later." 

63 ), 

64 not_found_message="App not found.", 

65 toolkit=toolkit, 

66 ), 

67 ): 

68 response = client.get_custom_domains(app_id=app_id) 35 ctx1hvxymnzGHjoAIJtpBKLkiwCDqrEMNusFOPl

69 

70 toolkit.success( 21 ctx1hvmnzjoAtpBkiwqrEusFl

71 CustomDomainsListOutput( 

72 app_id=app_id, 

73 domains=response.data, 

74 total_count=response.count, 

75 ), 

76 render_output=_render_custom_domains_list_output, 

77 )