shoopea/gottdad
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update graphs

Browse Source
This commit is contained in:
shoopea 2021-12-06 21:55:59 +08:00
parent c3225a79aa
commit 7c1c0670db
2 changed files with 183 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

181
bot.go
View File

@ -108,6 +108,8 @@ func (b *Bot) BotHandlers() {
b.bot.Handle("/transfer", botTransfer)
b.bot.Handle("/value", botGraphValue)
b.bot.Handle("/value_delta", botGraphValueDelta)
b.bot.Handle("/planes", botGraphPlanes)
b.bot.Handle(tb.OnPhoto, botPhoto)
b.bot.Handle(tb.OnChannelPost, botChannelPost)
@ -948,7 +950,7 @@ func botGraphValue(m *tb.Message) {
cc := cfg.GetCompanyClient(coID)
l, s, err := plotter.NewLinePoints(xys)
logErrorDebug(err, "Stats.ValueGraph : plotter.NewLinePoints")
logErrorDebug(err, "botGraphValue : plotter.NewLinePoints")
if err != nil {
bot.SendChat(m.Chat.ID, fmt.Sprintf("plotter.NewLinePoints : %s", err))
return
@ -967,7 +969,182 @@ func botGraphValue(m *tb.Message) {
}
err := p.Save(6*vg.Inch, 4*vg.Inch, "/app/data/points.png")
logErrorDebug(err, "Stats.ValueGraph : plot.Save")
logErrorDebug(err, "botGraphValue : plot.Save")
if err != nil {
bot.SendChat(m.Chat.ID, fmt.Sprintf("plot.Save : %s", err))
return
}
bot.SendChatImage(m.Chat.ID, "/app/data/points.png")
return
}
func botGraphValueDelta(m *tb.Message) {
var (
maxVal float64
unitFactor float64
unitName string
)
var vals map[uint8]plotter.XYs
vals = make(map[uint8]plotter.XYs)
for coID, dStats := range cfg.Stats {
if cfg.CompanyIsRegistered(coID) {
vals[coID] = make(plotter.XYs, 0)
for dStr, stat := range dStats {
d, err := time.Parse("20060102", dStr)
logErrorDebug(err, "botGraphValueDelta : time.Parse")
if err != nil {
bot.SendChat(m.Chat.ID, fmt.Sprintf("time.Parse : %s", err))
return
}
dateFloat := float64(d.Year()) + float64(d.Month()-1)/12
valueFloat := float64(stat.CompanyValueLastQuarter)
pt := plotter.XY{
X: dateFloat,
Y: valueFloat / unitFactor,
}
vals[coID] = append(vals[coID], pt)
}
sort.Slice(vals[coID], func(i, j int) bool { return vals[coID][i].X < vals[coID][j].X })
}
}
for coID, v := range vals {
for i := 1; i < len(vals); i++ {
v[i].Y = v[i].Y - v[i-1].Y
if math.Abs(v[i].Y) > maxVal {
maxVal = math.Abs(v[i].Y)
}
}
vals[coID] = v[1:]
}
if maxVal > 1000000000 {
unitFactor = 1000000000
unitName = "billion"
} else if maxVal > 1000000 {
unitFactor = 1000000
unitName = "million"
} else {
unitFactor = 1
unitName = ""
}
for coID, v := range vals {
for i := 1; i < len(vals)-1; i++ {
v[i].Y = v[i].Y / unitFactor
}
vals[coID] = v
}
p := plot.New()
p.Title.Text = "Company Values (delta)"
p.X.Label.Text = "Year"
if unitName != "" {
p.Y.Label.Text = fmt.Sprintf("Variation (%s)", unitName)
} else {
p.Y.Label.Text = "Variation"
}
i := 0
for coID, xys := range vals {
cc := cfg.GetCompanyClient(coID)
l, s, err := plotter.NewLinePoints(xys)
logErrorDebug(err, "botGraphValueDelta : plotter.NewLinePoints")
if err != nil {
bot.SendChat(m.Chat.ID, fmt.Sprintf("plotter.NewLinePoints : %s", err))
return
}
l.Color = plotutil.Color(i)
l.Dashes = plotutil.Dashes(2)
s.Color = plotutil.Color(i)
s.Shape = plotutil.Shape(0)
p.Add(l)
p.Add(s)
p.Legend.Add(cc.Username, l, s)
i++
}
err := p.Save(6*vg.Inch, 4*vg.Inch, "/app/data/points.png")
logErrorDebug(err, "botGraphValueDelta : plot.Save")
if err != nil {
bot.SendChat(m.Chat.ID, fmt.Sprintf("plot.Save : %s", err))
return
}
bot.SendChatImage(m.Chat.ID, "/app/data/points.png")
return
}
func botGraphPlanes(m *tb.Message) {
var planes, airports map[uint8]plotter.XYs
planes = make(map[uint8]plotter.XYs)
airports = make(map[uint8]plotter.XYs)
for coID, dStats := range cfg.Stats {
if cfg.CompanyIsRegistered(coID) {
planes[coID] = make(plotter.XYs, 0)
for dStr, stat := range dStats {
d, err := time.Parse("20060102", dStr)
logErrorDebug(err, "botGraphPlanes : time.Parse")
if err != nil {
bot.SendChat(m.Chat.ID, fmt.Sprintf("time.Parse : %s", err))
return
}
dateFloat := float64(d.Year()) + float64(d.Month()-1)/12
pt := plotter.XY{
X: dateFloat,
Y: float64(stat.Planes),
}
planes[coID] = append(planes[coID], pt)
pt = plotter.XY{
X: dateFloat,
Y: float64(stat.Airports),
}
airports[coID] = append(airports[coID], pt)
}
sort.Slice(planes[coID], func(i, j int) bool { return planes[coID][i].X < planes[coID][j].X })
sort.Slice(airports[coID], func(i, j int) bool { return airports[coID][i].X < airports[coID][j].X })
}
}
p := plot.New()
p.Title.Text = "Planes summary"
p.X.Label.Text = "Year"
p.Y.Label.Text = "Planes"
i := 0
for coID, xys := range planes {
cc := cfg.GetCompanyClient(coID)
l, s, err := plotter.NewLinePoints(xys)
logErrorDebug(err, "botGraphPlanes : plotter.NewLinePoints")
if err != nil {
bot.SendChat(m.Chat.ID, fmt.Sprintf("plotter.NewLinePoints : %s", err))
return
}
l.Color = plotutil.Color(i)
l.Dashes = plotutil.Dashes(2)
s.Color = plotutil.Color(i)
s.Shape = plotutil.Shape(0)
p.Add(l)
p.Add(s)
p.Legend.Add(cc.Username, l, s)
i++
}
err := p.Save(6*vg.Inch, 4*vg.Inch, "/app/data/points.png")
logErrorDebug(err, "botGraphPlanes : plot.Save")
if err != nil {
bot.SendChat(m.Chat.ID, fmt.Sprintf("plot.Save : %s", err))
return

8
version.go
View File

@ -1,6 +1,6 @@
// Code generated by version.sh (@generated) DO NOT EDIT.
package main
var githash = "365c8a9"
var buildstamp = "2021-12-05_14:39:05"
var commits = "232"
var version = "365c8a9-b232 - 2021-12-05_14:39:05"
var githash = "c3225a7"
var buildstamp = "2021-12-06_13:54:50"
var commits = "233"
var version = "c3225a7-b233 - 2021-12-06_13:54:50"