mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-24 10:40:09 +08:00
127 lines
2.9 KiB
Go
127 lines
2.9 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/process"
|
|
"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: true,
|
|
Help: "本插件用于查询城市疫情状况\n" +
|
|
"使用方法列如： 北京疫情 \n",
|
|
})
|
|
engine.OnSuffix("疫情").SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
text := ctx.State["args"].(string)
|
|
if text == "" {
|
|
ctx.SendChain(message.Text("你还没有输入城市名字呢！"))
|
|
return
|
|
}
|
|
data, times, err := queryEpidemic(text)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
if data == nil {
|
|
ctx.SendChain(message.Text("没有找到【", text, "】城市的疫情数据."))
|
|
return
|
|
}
|
|
process.SleepAbout1sTo2s()
|
|
ctx.SendChain(
|
|
message.Text(
|
|
"【", data.Name, "】疫情数据:\n",
|
|
"新增：", data.Today.Confirm,
|
|
" ,现有确诊：", data.Total.NowConfirm,
|
|
" ,治愈：", data.Total.Heal,
|
|
" ,死亡：", data.Total.Dead, " ", data.Total.Grade,
|
|
),
|
|
message.Text("\n"),
|
|
message.Text("更新时间："+times),
|
|
message.Text("\n"),
|
|
message.Text("温馨提示：请大家做好防疫工作，出门带好口罩！"),
|
|
)
|
|
})
|
|
}
|
|
|
|
// 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
|
|
}
|