update data && fix api text

sundy-li · Jan 8, 2018 · d57be69 · d57be69
1 parent 8e9b944
commit d57be69
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/api.go b/api.go
@@ -1,10 +1,11 @@
 package wechat_brain
 
 import (
-	"io/ioutil"
 	"net/http"
 	"net/url"
 	"strings"
+
+	"github.com/PuerkitoBio/goquery"
 )
 
 var (
@@ -27,9 +28,9 @@ func GetFromApi(req *http.Request, quiz string, options []string) (res map[strin
 	if resp == nil {
 		return
 	}
-	bs, _ := ioutil.ReadAll(resp.Body)
+	doc, _ := goquery.NewDocumentFromReader(resp.Body)
 	defer resp.Body.Close()
-	str := string(bs)
+	str := doc.Find("#content_left .result").Text()
 	for _, option := range options {
 		res[option] = strings.Count(str, option)
 	}

diff --git a/questions.data b/questions.data