From 0913485975baf0a5db4939ab3112ed7b8883d0f1 Mon Sep 17 00:00:00 2001 From: Joe Robinson Date: Fri, 18 Jul 2014 17:29:32 +0100 Subject: Added functions to add new polls --- blavote.go | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) (limited to 'blavote.go') diff --git a/blavote.go b/blavote.go index 4d621b9..85e9c5e 100644 --- a/blavote.go +++ b/blavote.go @@ -19,8 +19,10 @@ func main() { Version bool `short:"v" long:"version" description:"Show program version"` Add bool `short:"a" long:"add" description:"Add a new poll"` New bool `short:"n" long:"new" description:"Add a new poll"` + Username string `short:"u" long:"username" description:"Username of user adding poll"` Args struct { Title string + PollId int Rest []string } `positional-args:"yes"` } @@ -53,11 +55,14 @@ func main() { if (opts.Version) { fmt.Println("v" + version) } else if (opts.Add || opts.New) { - addPoll(db, opts.Args.Title, opts.Args.Rest) + err := addPoll(db, opts.Args.Title, opts.Args.Rest, opts.Username) + if (err == nil) { + fmt.Println("Poll added") + } else { + fmt.Println(err) + } } - createUser(db, "l_bratch", true) - getUserForId(db, 1) } @@ -88,14 +93,9 @@ func initTables(db *sqlite3.Conn) { db.Exec("insert into info (key, value) values('version', '0.01')") db.Exec("create table users(id integer primary key autoincrement, name text, admin boolean)") - db.Exec("create table polls(id integer primary key autoincrement, text text, user_id int)") + db.Exec("create table polls(id integer primary key autoincrement, title text, user_id int)") db.Exec("create table options(id integer primary key autoincrement, text text, poll_id int)") db.Exec("create table votes(id integer primary key autoincrement, user_id int, poll_id int, option_id int)") } - -func addPoll(db *sqlite3.Conn, title string, options []string) { - - -} \ No newline at end of file -- cgit v1.2.3