Coverage for src/fastapi_cloud_cli/commands/tokens/delete.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 7 ctx1abcdefg

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

9from fastapi_cloud_cli.commands.tokens._app import tokens_app 7 ctx1abcdefg

10from fastapi_cloud_cli.utils.apps import resolve_app_id_or_fail 7 ctx1abcdefg

11from fastapi_cloud_cli.utils.execution import JsonOutputOption 7 ctx1abcdefg

12 

13 

14class DeployTokenDeleteOutput(BaseModel): 7 ctx1abcdefg

15 token_id: str 7 ctx1abcdefg

16 deleted: bool = True 7 ctx1abcdefg

17 

18 

19def _delete_deploy_token(client: APIClient, *, app_id: str, token_id: str) -> bool: 7 ctx1abcdefg

20 response = client.delete(f"/apps/{app_id}/tokens/{token_id}") 28 ctx1kopqhrstluvwixyzmABCnDEFjGHI

21 

22 if response.status_code == 404: 28 ctx1kopqhrstluvwixyzmABCnDEFjGHI

23 return False 7 ctx1oruxADG

24 

25 response.raise_for_status() 21 ctx1kpqhstlvwiyzmBCnEFjHI

26 

27 return True 21 ctx1kpqhstlvwiyzmBCnEFjHI

28 

29 

30def _render_deploy_token_delete_output( 10 ctx1abhcdiefgj

31 data: DeployTokenDeleteOutput, toolkit: RichToolkit 

32) -> None: 

33 toolkit.print( 7 ctx1khlimnj

34 f"Deleted deploy token [bold]{data.token_id}[/bold]", 

35 bullet=False, 

36 ) 

37 

38 

39@tokens_app.command("delete", cls=UserCommand) 7 ctx1abcdefg

40def delete_token( 10 ctx1abQcdRefgS

41 ctx: typer.Context, 

42 token_id: Annotated[ 

43 str, 

44 typer.Argument( 

45 help="ID of the deploy token to delete.", 

46 ), 

47 ], 

48 app_id: Annotated[ 

49 str | None, 

50 typer.Option( 

51 "--app-id", 

52 help="ID of the app that owns the deploy token.", 

53 ), 

54 ] = None, 

55 json_output: JsonOutputOption = False, 

56) -> Any: 

57 """ 

58 Delete a deploy token for an app. 

59 """ 

60 

61 toolkit = get_user_command_context(ctx).toolkit 35 ctx1kopqJhrstKluvwLixyzMmABCNnDEFOjGHIP

62 

63 target_app_id = resolve_app_id_or_fail(toolkit, app_id=app_id) 35 ctx1kopqJhrstKluvwLixyzMmABCNnDEFOjGHIP

64 

65 with APIClient() as client: 28 ctx1kopqhrstluvwixyzmABCnDEFjGHI

66 with toolkit.progress( 28 ctx1kopqhrstluvwixyzmABCnDEFjGHI

67 title="Deleting deploy token", 

68 transient=True, 

69 ) as progress: 

70 with client.handle_http_errors( 28 ctx1kopqhrstluvwixyzmABCnDEFjGHI

71 progress, 

72 default_message="Error deleting deploy token. Please try again later.", 

73 not_found_message="Deploy token not found.", 

74 toolkit=toolkit, 

75 ): 

76 deleted = _delete_deploy_token( 28 ctx1kopqhrstluvwixyzmABCnDEFjGHI

77 client, 

78 app_id=target_app_id, 

79 token_id=token_id, 

80 ) 

81 

82 if not deleted: 28 ctx1kopqhrstluvwixyzmABCnDEFjGHI

83 message = ( 7 ctx1oruxADG

84 f"Deploy token {token_id} not found." 

85 if toolkit.mode == "json" 

86 else "Deploy token not found." 

87 ) 

88 toolkit.fail( 7 ctx1oruxADG

89 "not_found", 

90 message, 

91 hint="Run `fastapi cloud tokens list` to see available deploy tokens.", 

92 ) 

93 

94 toolkit.success( 21 ctx1kpqhstlvwiyzmBCnEFjHI

95 DeployTokenDeleteOutput(token_id=token_id), 

96 render_output=_render_deploy_token_delete_output, 

97 )