コード例 #1
0
ファイル: parsehtml.go プロジェクト: sbalev/gobananas
func getClues(n *html.Node) (uint8, uint8, error) {
	if c := htmlutil.GetElChildren(n, "table"); len(c) == 0 {
		return 0, 0, gridErr("table not found")
	} else {
		n = c[0]
	}
	if cl, _ := htmlutil.GetAttr(n, "class"); cl != "infotablegrey" {
		return 0, 0, gridErr("Wrong table class")
	}
	tab, err := htmlutil.ParseTable(n)
	if err != nil {
		return 0, 0, err
	}
	if len(tab) != 2 || len(tab[0]) != 2 || len(tab[1]) != 2 {
		return 0, 0, gridErr("Wrong table size")
	}
	v, err := getClue(tab[1][0])
	if err != nil {
		return 0, 0, err
	}
	h, err := getClue(tab[0][1])
	if err != nil {
		return 0, 0, nil
	}
	return v, h, nil
}
コード例 #2
0
ファイル: parsehtml.go プロジェクト: sbalev/gobananas
func FetchGrid(url string) (*Grid, error) {
	resp, err := http.Get(url)
	if err != nil {
		return nil, err
	}
	defer resp.Body.Close()
	n, err := html.Parse(resp.Body)
	if err != nil {
		return nil, err
	}
	for _, tag := range []string{"html", "body", "table"} {
		if c := htmlutil.GetElChildren(n, tag); len(c) == 0 {
			return nil, gridErr(tag + " not found")
		} else {
			n = c[0]
		}
	}
	if cl, _ := htmlutil.GetAttr(n, "class"); cl != "kakurotablegrey" {
		return nil, gridErr("Wrong table class")
	}
	tab, err := htmlutil.ParseTable(n)
	if len(tab) < 2 {
		return nil, gridErr("Grid too short")
	}
	colCount := len(tab[0])
	if colCount < 2 {
		return nil, gridErr("Grid too narrow")
	}
	for _, r := range tab {
		if len(r) != colCount {
			return nil, gridErr("Rows of different lengths")
		}
	}
	g := &Grid{h: len(tab), w: colCount, cells: make([][]Cell, len(tab))}
	for i, r := range tab {
		g.cells[i] = make([]Cell, g.w)
		for j, c := range r {
			if x, err := parseHTMLCell(c); err != nil {
				return nil, err
			} else {
				g.cells[i][j] = x
			}
		}
	}
	return g, nil
}