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

1from typing import Annotated, Any 7 ctx1abcdefg

2 

3import typer 7 ctx1abcdefg

4from pydantic import BaseModel, Field 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._shared import ( 7 ctx1abcdefg

11 _find_custom_domain, 

12 _select_custom_domain, 

13) 

14from fastapi_cloud_cli.commands.domains.rendering import render_custom_domain_details 7 ctx1abcdefg

15from fastapi_cloud_cli.utils.apps import resolve_app_id_or_fail 7 ctx1abcdefg

16from fastapi_cloud_cli.utils.execution import JsonOutputOption 7 ctx1abcdefg

17 

18 

19class CustomDomainGetOutput(BaseModel): 7 ctx1abcdefg

20 app_id: str 7 ctx1abcdefg

21 domain: CustomDomain 7 ctx1abcdefg

22 show_title: Annotated[bool, Field(exclude=True)] = True 7 ctx1abcdefg

23 

24 

25def _render_custom_domain_get_output( 10 ctx1abocdpefgq

26 data: CustomDomainGetOutput, 

27 toolkit: RichToolkit, 

28) -> None: 

29 if data.show_title: 42 ctx1rshtuvBCjDoEFGkHIJKLlMpNwxiyzAOPmQRSTUnVqW

30 toolkit.print_title("custom domains") 35 ctx1rstuvBCDoEFGHIJKLMpNwxyzAOPQRSTUVqW

31 toolkit.print_line() 35 ctx1rstuvBCDoEFGHIJKLMpNwxyzAOPQRSTUVqW

32 

33 render_custom_domain_details(data.domain, toolkit) 42 ctx1rshtuvBCjDoEFGkHIJKLlMpNwxiyzAOPmQRSTUnVqW

34 

35 

36@domains_app.command("get", cls=UserCommand) 7 ctx1abcdefg

37def get_domain( 10 ctx1ab=cd?efg@

38 ctx: typer.Context, 

39 domain: Annotated[ 

40 str | None, 

41 typer.Argument( 

42 help="Hostname or ID of the custom domain to return.", 

43 ), 

44 ] = None, 

45 app_id: Annotated[ 

46 str | None, 

47 typer.Option( 

48 "--app-id", 

49 help="ID of the app that owns the custom domain.", 

50 ), 

51 ] = None, 

52 json_output: JsonOutputOption = False, 

53) -> Any: 

54 """ 

55 Get a custom domain for an app. 

56 """ 

57 toolkit = get_user_command_context(ctx).toolkit 70 ctx1r+shtu64XvB,CjDo%7ZEF-GkHI'80JK.LlMp(91Nw/xiyz!5YAO:PmQR)#2ST;UnVq*$3W

58 app_id = resolve_app_id_or_fail(toolkit, app_id=app_id) 70 ctx1r+shtu64XvB,CjDo%7ZEF-GkHI'80JK.LlMp(91Nw/xiyz!5YAO:PmQR)#2ST;UnVq*$3W

59 domain_was_provided = domain is not None 70 ctx1r+shtu64XvB,CjDo%7ZEF-GkHI'80JK.LlMp(91Nw/xiyz!5YAO:PmQR)#2ST;UnVq*$3W

60 

61 if domain is None and toolkit.mode == "json": 70 ctx1r+shtu64XvB,CjDo%7ZEF-GkHI'80JK.LlMp(91Nw/xiyz!5YAO:PmQR)#2ST;UnVq*$3W

62 toolkit.fail( 7 ctx1+,-./:;

63 "missing_required_input", 

64 "Custom domain is required.", 

65 hint="Pass DOMAIN to choose a custom domain.", 

66 ) 

67 

68 with APIClient() as client: 63 ctx1rshtu64XvBCjDo%7ZEFGkHI'80JKLlMp(91Nwxiyz!5YAOPmQR)#2STUnVq*$3W

69 with ( 18 ctx1rshtu64Xvwxiyz!5YA

70 toolkit.progress( 

71 title="Fetching custom domains", 

72 transient=True, 

73 ) as progress, 

74 client.handle_http_errors( 

75 progress, 

76 default_message=( 

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

78 ), 

79 not_found_message="App not found.", 

80 toolkit=toolkit, 

81 ), 

82 ): 

83 domains = client.get_custom_domains(app_id=app_id).data 63 ctx1rshtu64XvBCjDo%7ZEFGkHI'80JKLlMp(91Nwxiyz!5YAOPmQR)#2STUnVq*$3W

84 

85 selected_domain: CustomDomain | None 

86 

87 if domain is None: 63 ctx1rshtu64XvBCjDo%7ZEFGkHI'80JKLlMp(91Nwxiyz!5YAOPmQR)#2STUnVq*$3W

88 toolkit.print_title("custom domains") 14 ctx1hXjZk0l1iYm2n3

89 toolkit.print_line() 14 ctx1hXjZk0l1iYm2n3

90 

91 if not domains: 14 ctx1hXjZk0l1iYm2n3

92 toolkit.print("No custom domains found.", bullet=False) 7 ctx1XZ01Y23

93 return 7 ctx1XZ01Y23

94 

95 selected_domain = _select_custom_domain( 7 ctx1hjklimn

96 toolkit, 

97 domains, 

98 prompt="Select the custom domain to get:", 

99 ) 

100 toolkit.print_line() 7 ctx1hjklimn

101 else: 

102 if (selected_domain := _find_custom_domain(domains, domain)) is None: 49 ctx1rstu64vBCDo%7EFGHI'8JKLMp(9Nwxyz!5AOPQR)#STUVq*$W

103 toolkit.fail( 7 ctx16%'(!)*

104 "not_found", 

105 f"Custom domain {domain} not found.", 

106 hint=( 

107 "Run `fastapi cloud domains list` to see available custom domains." 

108 ), 

109 ) 

110 

111 assert selected_domain is not None 49 ctx1rshtu4vBCjDo7EFGkHI8JKLlMp9Nwxiyz5AOPmQR#STUnVq$W

112 

113 toolkit.success( 49 ctx1rshtu4vBCjDo7EFGkHI8JKLlMp9Nwxiyz5AOPmQR#STUnVq$W

114 CustomDomainGetOutput( 

115 app_id=app_id, 

116 domain=selected_domain, 

117 show_title=domain_was_provided, 

118 ), 

119 render_output=_render_custom_domain_get_output, 

120 )