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

1import logging 7 ctx1abcdefg

2from typing import Annotated, Any 7 ctx1abcdefg

3 

4import typer 7 ctx1abcdefg

5from pydantic import BaseModel 7 ctx1abcdefg

6from rich.markup import escape 7 ctx1abcdefg

7from rich.table import Table 7 ctx1abcdefg

8from rich_toolkit import RichToolkit 7 ctx1abcdefg

9 

10from fastapi_cloud_cli.api import APIClient 7 ctx1abcdefg

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

12from fastapi_cloud_cli.commands.teams._app import teams_app 7 ctx1abcdefg

13from fastapi_cloud_cli.commands.teams.get import Team, _get_team_dashboard_url 7 ctx1abcdefg

14from fastapi_cloud_cli.config import Settings 7 ctx1abcdefg

15from fastapi_cloud_cli.utils.execution import JsonOutputOption 7 ctx1abcdefg

16 

17logger = logging.getLogger(__name__) 7 ctx1abcdefg

18 

19DEFAULT_LIMIT = 100 7 ctx1abcdefg

20DEFAULT_OFFSET = 0 7 ctx1abcdefg

21 

22 

23class TeamsListAPIResponse(BaseModel): 7 ctx1abcdefg

24 data: list[Team] 7 ctx1abcdefg

25 count: int 7 ctx1abcdefg

26 

27 

28class TeamsListOutput(BaseModel): 7 ctx1abcdefg

29 teams: list[Team] 7 ctx1abcdefg

30 total_count: int 7 ctx1abcdefg

31 limit: int 7 ctx1abcdefg

32 offset: int 7 ctx1abcdefg

33 

34 

35def _get_teams(client: APIClient, *, limit: int, offset: int) -> TeamsListOutput: 7 ctx1abcdefg

36 response = client.get( 28 ctx1vCkowDhpxElqyFirzGmsAHntBIju

37 "/teams/", 

38 params={ 

39 "limit": limit, 

40 "skip": offset, 

41 }, 

42 ) 

43 response.raise_for_status() 28 ctx1vCkowDhpxElqyFirzGmsAHntBIju

44 

45 data = TeamsListAPIResponse.model_validate(response.json()) 21 ctx1vkowhpxlqyirzmsAntBju

46 

47 return TeamsListOutput( 21 ctx1vkowhpxlqyirzmsAntBju

48 teams=data.data, 

49 total_count=data.count, 

50 limit=limit, 

51 offset=offset, 

52 ) 

53 

54 

55def _render_teams_list_output(data: TeamsListOutput, toolkit: RichToolkit) -> None: 10 ctx1abhcdiefgj

56 toolkit.print_title("teams") 14 ctx1kohplqirmsntju

57 toolkit.print_line() 14 ctx1kohplqirmsntju

58 

59 if not data.teams: 14 ctx1kohplqirmsntju

60 toolkit.print("No teams found.", bullet=False) 7 ctx1opqrstu

61 return 7 ctx1opqrstu

62 

63 settings = Settings.get() 7 ctx1khlimnj

64 

65 table = Table.grid(padding=(0, 2), pad_edge=False) 7 ctx1khlimnj

66 table.add_column("Name") 7 ctx1khlimnj

67 table.add_column("ID") 7 ctx1khlimnj

68 table.add_row("[bold]Name[/bold]", "[bold]ID[/bold]") 7 ctx1khlimnj

69 table.add_row("", "") 7 ctx1khlimnj

70 

71 for team in data.teams: 7 ctx1khlimnj

72 table.add_row( 7 ctx1khlimnj

73 f"[link={_get_team_dashboard_url(team, settings=settings)}]{escape(team.name)}[/link]", 

74 team.id, 

75 ) 

76 

77 toolkit.print(table, bullet=False) 7 ctx1khlimnj

78 

79 

80@teams_app.command("list", cls=UserCommand) 7 ctx1abcdefg

81def list_teams( 10 ctx1abJcdKefgL

82 ctx: typer.Context, 

83 limit: Annotated[ 

84 int, 

85 typer.Option( 

86 "--limit", 

87 help="Maximum number of teams to return.", 

88 min=1, 

89 ), 

90 ] = DEFAULT_LIMIT, 

91 offset: Annotated[ 

92 int, 

93 typer.Option( 

94 "--offset", 

95 help="Offset into the team result set.", 

96 min=0, 

97 ), 

98 ] = DEFAULT_OFFSET, 

99 json_output: JsonOutputOption = False, 

100) -> Any: 

101 """ 

102 List FastAPI Cloud teams. 

103 """ 

104 

105 toolkit = get_user_command_context(ctx).toolkit 28 ctx1vCkowDhpxElqyFirzGmsAHntBIju

106 

107 with APIClient() as client: 28 ctx1vCkowDhpxElqyFirzGmsAHntBIju

108 with toolkit.progress( 28 ctx1vCkowDhpxElqyFirzGmsAHntBIju

109 title="Fetching teams", 

110 transient=True, 

111 ) as progress: 

112 with client.handle_http_errors( 28 ctx1vCkowDhpxElqyFirzGmsAHntBIju

113 progress, 

114 default_message="Error fetching teams. Please try again later.", 

115 toolkit=toolkit, 

116 ): 

117 result = _get_teams(client, limit=limit, offset=offset) 28 ctx1vCkowDhpxElqyFirzGmsAHntBIju

118 

119 toolkit.success(result, render_output=_render_teams_list_output) 21 ctx1vkowhpxlqyirzmsAntBju