Coverage for src/fastapi_cloud_cli/utils/teams.py: 100%

1from pathlib import Path 7 ctx1hijklmn

2 

3from pydantic import BaseModel 7 ctx1hijklmn

4from rich_toolkit.menu import Option 7 ctx1hijklmn

5 

6from fastapi_cloud_cli.api import APIClient 7 ctx1hijklmn

7from fastapi_cloud_cli.utils.apps import get_app_config 7 ctx1hijklmn

8from fastapi_cloud_cli.utils.cli import FastAPIRichToolkit 7 ctx1hijklmn

9 

10 

11class Team(BaseModel): 7 ctx1hijklmn

12 id: str 7 ctx1hijklmn

13 slug: str 7 ctx1hijklmn

14 name: str 7 ctx1hijklmn

15 

16 

17def get_teams(client: APIClient) -> list[Team]: 7 ctx1hijklmn

18 response = client.get("/teams/") 35 ctx1aovwxbpyzAcqBCDdrEFGesHIJftKLMguNOP

19 response.raise_for_status() 35 ctx1aovwxbpyzAcqBCDdrEFGesHIJftKLMguNOP

20 

21 return [Team.model_validate(team) for team in response.json()["data"]] 35 ctx1aovwxbpyzAcqBCDdrEFGesHIJftKLMguNOP

22 

23 

24def select_team( 7 ctx1hijklmn

25 toolkit: FastAPIRichToolkit, 

26 client: APIClient, 

27 *, 

28 empty_hint: str, 

29) -> Team: 

30 with toolkit.progress( 35 ctx1aovwxbpyzAcqBCDdrEFGesHIJftKLMguNOP

31 title="Fetching teams", 

32 transient=True, 

33 ) as progress: 

34 with client.handle_http_errors( 35 ctx1aovwxbpyzAcqBCDdrEFGesHIJftKLMguNOP

35 progress, 

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

37 toolkit=toolkit, 

38 ): 

39 teams = get_teams(client) 35 ctx1aovwxbpyzAcqBCDdrEFGesHIJftKLMguNOP

40 

41 if not teams: 35 ctx1aovwxbpyzAcqBCDdrEFGesHIJftKLMguNOP

42 toolkit.fail( 14 ctx1oxpAqDrGsJtMuP

43 "missing_required_input", 

44 "No teams found.", 

45 hint=empty_hint, 

46 ) 

47 

48 return toolkit.ask( 21 ctx1avwbyzcBCdEFeHIfKLgNO

49 "Select the team:", 

50 options=[ 

51 Option({"name": team.name, "value": team}) 

52 for team in sorted(teams, key=lambda team: team.name.lower()) 

53 ], 

54 allow_filtering=True, 

55 bullet=False, 

56 ) 

57 

58 

59def resolve_team_id( 7 ctx1hijklmn

60 toolkit: FastAPIRichToolkit, 

61 client: APIClient, 

62 *, 

63 team_id: str | None = None, 

64 path: Path | None = None, 

65 empty_hint: str, 

66) -> str: 

67 if team_id is not None: 49 ctx145Q6aoX78R9bpY!#S$cqZ%'T(dr0)*U+es1,-V.ft2/:W;gu3

68 return team_id 21 ctx1456789!#$%'()*+,-./:;

69 

70 if app_config := get_app_config(path or Path.cwd()): 28 ctx1QaoXRbpYScqZTdr0Ues1Vft2Wgu3

71 return app_config.team_id 7 ctx1XYZ0123

72 

73 if toolkit.mode == "json": 21 ctx1QaoRbpScqTdrUesVftWgu

74 toolkit.fail( 7 ctx1QRSTUVW

75 "missing_required_input", 

76 "Team ID is required.", 

77 hint="Pass --team-id or run the command from a linked app.", 

78 ) 

79 

80 team = select_team(toolkit, client, empty_hint=empty_hint) 14 ctx1aobpcqdresftgu

81 toolkit.print_line() 7 ctx1abcdefg

82 

83 return team.id 7 ctx1abcdefg