mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-20 22:40:23 +08:00
124 lines
2.8 KiB
Go
124 lines
2.8 KiB
Go
// Package epidemic 城市疫情查询
|
|
package epidemic
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
|
|
"github.com/FloatTech/zbputils/binary"
|
|
"github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/control/order"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
"github.com/FloatTech/zbputils/web"
|
|
)
|
|
|
|
const (
|
|
servicename = "epidemic"
|
|
txurl = "https://view.inews.qq.com/g2/getOnsInfo?name=disease_h5"
|
|
)
|
|
|
|
// result 疫情查询结果
|
|
type result struct {
|
|
Data string `json:"data"`
|
|
}
|
|
|
|
// epidemic 疫情数据
|
|
type epidemic struct {
|
|
LastUpdateTime string `json:"lastUpdateTime"`
|
|
AreaTree []*area `json:"areaTree"`
|
|
}
|
|
|
|
// area 城市疫情数据
|
|
type area struct {
|
|
Name string `json:"name"`
|
|
Today struct {
|
|
Confirm int `json:"confirm"`
|
|
} `json:"today"`
|
|
Total struct {
|
|
NowConfirm int `json:"nowConfirm"`
|
|
Confirm int `json:"confirm"`
|
|
Dead int `json:"dead"`
|
|
Heal int `json:"heal"`
|
|
Grade string `json:"grade"`
|
|
} `json:"total"`
|
|
Children []*area `json:"children"`
|
|
}
|
|
|
|
func init() {
|
|
engine := control.Register(servicename, order.AcquirePrio(), &control.Options{
|
|
DisableOnDefault: false,
|
|
Help: "城市疫情查询\n" +
|
|
"- xxx疫情\n",
|
|
})
|
|
engine.OnSuffix("疫情").SetBlock(true).Limit(ctxext.LimitByUser).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
city := ctx.State["args"].(string)
|
|
if city == "" {
|
|
ctx.SendChain(message.Text("你还没有输入城市名字呢！"))
|
|
return
|
|
}
|
|
data, time, err := queryEpidemic(city)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
if data == nil {
|
|
ctx.SendChain(message.Text("没有找到【", city, "】城市的疫情数据."))
|
|
return
|
|
}
|
|
ctx.SendChain(
|
|
message.Text(
|
|
"【", data.Name, "】疫情数据\n",
|
|
"新增：", data.Today.Confirm, " ,",
|
|
"现有确诊：", data.Total.NowConfirm, " ,",
|
|
"治愈：", data.Total.Heal, " ,",
|
|
"死亡：", data.Total.Dead, " ", data.Total.Grade, "\n",
|
|
"更新时间：", time, "\n",
|
|
"温馨提示：请大家做好防疫工作，出门带好口罩！",
|
|
),
|
|
)
|
|
})
|
|
}
|
|
|
|
// rcity 查找城市
|
|
func rcity(a *area, cityName string) *area {
|
|
if a == nil {
|
|
return nil
|
|
}
|
|
if a.Name == cityName {
|
|
return a
|
|
}
|
|
for _, v := range a.Children {
|
|
if v.Name == cityName {
|
|
return v
|
|
}
|
|
c := rcity(v, cityName)
|
|
if c != nil {
|
|
return c
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// queryEpidemic 查询城市疫情
|
|
func queryEpidemic(findCityName string) (citydata *area, times string, err error) {
|
|
data, err := web.GetData(txurl)
|
|
if err != nil {
|
|
return
|
|
}
|
|
var r result
|
|
err = json.Unmarshal(data, &r)
|
|
if err != nil {
|
|
return
|
|
}
|
|
var e epidemic
|
|
err = json.Unmarshal(binary.StringToBytes(r.Data), &e)
|
|
if err != nil {
|
|
return
|
|
}
|
|
citydata = rcity(e.AreaTree[0], findCityName)
|
|
return citydata, e.LastUpdateTime, nil
|
|
}
|