mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 05:30:07 +08:00
121 lines
2.9 KiB
Go
121 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/floatbox/web"
|
|
ctrl "github.com/FloatTech/zbpctrl"
|
|
"github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
)
|
|
|
|
const txurl = "https://api.inews.qq.com/newsqa/v1/query/inner/publish/modules/list?modules=statisGradeCityDetail,diseaseh5Shelf"
|
|
|
|
// result 疫情查询结果
|
|
type result struct {
|
|
Data struct {
|
|
Epidemic epidemic `json:"diseaseh5Shelf"`
|
|
} `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"`
|
|
Wzzadd any `json:"wzz_add"`
|
|
} `json:"today"`
|
|
Total struct {
|
|
NowConfirm int `json:"nowConfirm"`
|
|
Confirm int `json:"confirm"`
|
|
Dead int `json:"dead"`
|
|
Heal int `json:"heal"`
|
|
Grade string `json:"grade"`
|
|
Wzz int `json:"wzz"`
|
|
} `json:"total"`
|
|
Children []*area `json:"children"`
|
|
}
|
|
|
|
func init() {
|
|
engine := control.Register("epidemic", &ctrl.Options[*zero.Ctx]{
|
|
DisableOnDefault: false,
|
|
Brief: "城市疫情查询",
|
|
Help: "- 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, "\n",
|
|
"现有确诊：", data.Total.NowConfirm, "\n",
|
|
"累计确诊：", data.Total.Confirm, "\n",
|
|
"治愈人数：", data.Total.Heal, "\n",
|
|
"死亡人数：", data.Total.Dead, "\n",
|
|
"无症状人数：", data.Total.Wzz, "\n",
|
|
"新增无症状：", data.Today.Wzzadd, "\n",
|
|
"更新时间：\n『", time, "』",
|
|
),
|
|
)
|
|
})
|
|
}
|
|
|
|
// 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
|
|
}
|
|
citydata = rcity(r.Data.Epidemic.AreaTree[0], findCityName)
|
|
return citydata, r.Data.Epidemic.LastUpdateTime, nil
|
|
}
|