diff options
author | Joe Robinson <joe@mumsnet.com> | 2014-07-20 16:13:16 +0100 |
---|---|---|
committer | Joe Robinson <joe@mumsnet.com> | 2014-07-20 16:13:16 +0100 |
commit | 9c64140e68d236b6b8bbe90157b60cb9eb35506d (patch) | |
tree | 0c0c9832f3219452c63dafaee36d1100f76163c5 /blavote.go | |
parent | 2b2d6732707ff24a50f672466f8b9fbb40f90653 (diff) |
Added poll removal option
Diffstat (limited to 'blavote.go')
-rw-r--r-- | blavote.go | 13 |
1 files changed, 12 insertions, 1 deletions
@@ -13,7 +13,7 @@ var version string func main() { - version = "0.1" + version = "0.2" //Command line arguments var opts struct { @@ -21,6 +21,7 @@ func main() { Add string `short:"a" long:"add" description:"Title for a new poll"` New string `short:"n" long:"new" description:"Title for a new poll"` Username string `short:"u" long:"username" description:"Username of user adding poll"` + Remove int `short:"r" long:"remove" description:"ID of a poll to delete"` Args struct { Rest []string @@ -73,6 +74,16 @@ func main() { } else { fmt.Println(err) } + } else if (opts.Remove > 0) { + deletePoll(db, opts.Remove) + + if (err == nil) { + fmt.Print("Poll removed with ID ") + fmt.Println(opts.Remove) + } else { + fmt.Println(err) + } + } else { fmt.Println(opts.Args.Rest) vote(db, opts.Username, opts.Args.Rest) |