download_logos/main.go

74 lines
1.4 KiB
Go
Raw Permalink Normal View History

2023-06-18 05:56:59 +00:00
package main
import (
"log"
"os"
"strconv"
"strings"
"github.com/474420502/extractor"
"github.com/474420502/requests"
)
type ItemList struct {
LogoUri []string `exp:"//div[@class='result-group bg-white']//img" method:"AttributeValue,src"`
}
func main() {
ses := requests.NewSession()
tp := ses.Get("https://brandeps.com/logo/Latest?page=500")
dp, err := os.ReadFile("logos/.record")
if err != nil {
log.Panic(err)
}
startPage, err := strconv.Atoi(string(dp))
if err != nil {
log.Panic(err)
}
page := tp.QueryParam("page")
for i := startPage; i < 500; i++ {
log.Println("page", i)
// strconv.Itoa(i)
err = os.WriteFile("logos/.record", []byte(strconv.Itoa(i)), 0644)
if err != nil {
panic(err)
}
page.IntSet(int64(i))
res, err := tp.Execute()
if err != nil {
panic(err)
}
// log.Println(res.ContentString())
etor := extractor.ExtractHtml(res.Content())
itemlist := &ItemList{}
etor.GetObjectByTag(itemlist)
if len(itemlist.LogoUri) <= 0 {
break
}
for _, uri := range itemlist.LogoUri {
// uri = strings.Replace(uri, ".svg", ".eps", 1)
log.Println(uri)
res, err = ses.Get(uri).Execute()
if err != nil {
panic(err)
}
index := strings.LastIndex(uri, "/")
if index >= 0 {
log.Println(uri[index+1:])
err = os.WriteFile("logos/"+uri[index+1:], res.Content(), 0644)
if err != nil {
panic(err)
}
}
}
}
// https://brandeps.com/logo/Latest?page=4
}