From 505ee9f32a86975a43637dbe9aa99957d325eb7f Mon Sep 17 00:00:00 2001 From: "Cooper, Trevor" Date: Thu, 15 Apr 2021 08:05:06 -0700 Subject: [PATCH] add SetYes to profile delete command --- internal/app/wwctl/profile/delete/main.go | 25 +++++++++++++---------- internal/app/wwctl/profile/delete/root.go | 5 +++++ 2 files changed, 19 insertions(+), 11 deletions(-) diff --git a/internal/app/wwctl/profile/delete/main.go b/internal/app/wwctl/profile/delete/main.go index 48bd74f1..651d07b5 100644 --- a/internal/app/wwctl/profile/delete/main.go +++ b/internal/app/wwctl/profile/delete/main.go @@ -48,19 +48,22 @@ func CobraRunE(cmd *cobra.Command, args []string) error { } if count > 0 { - q := fmt.Sprintf("Are you sure you want to delete %d profile(s) (%d groups, %d nodes)", count, numGroups, numNodes) - - prompt := promptui.Prompt{ - Label: q, - IsConfirm: true, - } - - result, _ := prompt.Run() - - if result == "y" || result == "yes" { + if SetYes == true { nodeDB.Persist() - } + } else { + q := fmt.Sprintf("Are you sure you want to delete %d profile(s) (%d groups, %d nodes)", count, numGroups, numNodes) + prompt := promptui.Prompt{ + Label: q, + IsConfirm: true, + } + + result, _ := prompt.Run() + + if result == "y" || result == "yes" { + nodeDB.Persist() + } + } } else { wwlog.Printf(wwlog.INFO, "No groups found\n") } diff --git a/internal/app/wwctl/profile/delete/root.go b/internal/app/wwctl/profile/delete/root.go index 9a9c028b..ad16efe4 100644 --- a/internal/app/wwctl/profile/delete/root.go +++ b/internal/app/wwctl/profile/delete/root.go @@ -10,8 +10,13 @@ var ( RunE: CobraRunE, Args: cobra.MinimumNArgs(1), } + SetYes bool ) +func init() { +baseCmd.PersistentFlags().BoolVarP(&SetYes, "yes", "y", false, "Set 'yes' to all questions asked") +} + // GetRootCommand returns the root cobra.Command for the application. func GetCommand() *cobra.Command { return baseCmd