mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-20 22:40:23 +08:00
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package qingyunke
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"math/rand"
|
|
"net/http"
|
|
"regexp"
|
|
)
|
|
|
|
var (
|
|
reImg = `https?://[^"]+?(\.((jpg)|(png)|(jpeg)|(gif)|(bmp)))`
|
|
)
|
|
|
|
//取图片
|
|
func getPicture() string {
|
|
prefix := "https://fabiaoqing.com/tag/detail/id/5682/page"
|
|
url := fmt.Sprintf("%d.html", rand.Intn(11)+1)
|
|
url = prefix + url
|
|
log.Println("正在" + url + "寻找图片")
|
|
urls := getImgs(url)
|
|
fmt.Println(urls)
|
|
imageUrl := urls[rand.Intn(len(urls))]
|
|
log.Println("取到" + imageUrl)
|
|
return imageUrl
|
|
}
|
|
|
|
func HandleError(err error, why string) {
|
|
if err != nil {
|
|
fmt.Println(why, err)
|
|
}
|
|
}
|
|
|
|
func getImgs(url string) (urls []string) {
|
|
pageStr := GetPageStr(url)
|
|
re := regexp.MustCompile(reImg)
|
|
results := re.FindAllStringSubmatch(pageStr, -1)
|
|
fmt.Printf("共找到%d条结果\n", len(results))
|
|
for _, result := range results {
|
|
url := result[0]
|
|
urls = append(urls, url)
|
|
}
|
|
return
|
|
}
|
|
|
|
func GetPageStr(url string) (pageStr string) {
|
|
resp, err := http.Get(url)
|
|
HandleError(err, "http.Get url")
|
|
defer resp.Body.Close()
|
|
// 2.读取页面内容
|
|
pageBytes, err := ioutil.ReadAll(resp.Body)
|
|
HandleError(err, "ioutil.ReadAll")
|
|
// 字节转字符串
|
|
pageStr = string(pageBytes)
|
|
return pageStr
|
|
}
|