Get planet id from galaxy infos

alaingilbert · Sep 1, 2018 · 83e004c · 83e004c
1 parent 5e8115b
commit 83e004c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/ogame.go b/ogame.go
@@ -1614,6 +1614,7 @@ func extractGalaxyInfos(pageHTML, lang string) ([]PlanetInfos, error) {
 			recyclersTxt := s.Find("div#debris" + position + " ul.ListLinks li").Eq(2).Text()
 
 			planetInfos := PlanetInfos{}
+			planetInfos.ID, _ = strconv.Atoi(s.Find("td.colonized").AttrOr("data-planet-id", ""))
 
 			moonID, _ := strconv.Atoi(s.Find("td.moon").AttrOr("data-moon-id", ""))
 			moonSize, _ := strconv.Atoi(strings.Split(s.Find("td.moon span#moonsize").Text(), " ")[0])

diff --git a/ogame_test.go b/ogame_test.go
@@ -49,6 +49,8 @@ func TestExtractGalaxyInfos(t *testing.T) {
 	pageHTMLBytes, _ := ioutil.ReadFile("samples/galaxy_ajax.html")
 	infos, _ := extractGalaxyInfos(string(pageHTMLBytes), "en")
 	assert.Equal(t, 5, len(infos))
+	assert.Equal(t, 33698600, infos[0].ID)
+	assert.Equal(t, 33698645, infos[1].ID)
 }
 
 func TestExtractUserInfos(t *testing.T) {

diff --git a/planetInfos.go b/planetInfos.go
@@ -7,6 +7,7 @@ type MoonInfos struct {
 
 // PlanetInfos ...
 type PlanetInfos struct {
+	ID              int
 	Activity        int
 	Name            string
 	Img             string