cws
/
catbase
mirror of https://github.com/velour/catbase.git
1
0
Fork 0
Code Issues Releases Wiki Activity

countdown: maybe use actually January

This commit is contained in:
Chris Sexton 2020-11-30 14:55:35 -05:00
parent d7839c96d8
commit aa9dc12c9b
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.go
View File

@ -15,6 +15,7 @@ import (
"github.com/velour/catbase/plugins/achievements"
"github.com/velour/catbase/plugins/aoc"
"github.com/velour/catbase/plugins/countdown"
"github.com/velour/catbase/plugins/goals"
"github.com/velour/catbase/plugins/meme"
"github.com/velour/catbase/plugins/sms"
@ -152,6 +153,7 @@ func main() {
b.AddPlugin(meme.New(b))
b.AddPlugin(achievements.New(b))
b.AddPlugin(sms.New(b))
b.AddPlugin(countdown.New(b))
b.AddPlugin(gpt2.New(b))
// catches anything left, will always return true
b.AddPlugin(fact.New(b))

6
plugins/countdown/countdown.go
View File

@ -12,7 +12,8 @@ import (
"github.com/velour/catbase/config"
)
var nextYear = time.Date(time.Now().Year()+1, 0, 0, 0, 0, 0, 0, time.Local)
var nextYear = time.Date(time.Now().Year()+1, time.January, 0, 0, 0, 0, 0, time.Local)
var thisYear = time.Now().Year()
type CountdownPlugin struct {
b bot.Bot
@ -52,8 +53,7 @@ func (p *CountdownPlugin) newYearRollover() {
logError(err, "error getting transaction")
return
}
year := time.Now().Year() - 1
q := fmt.Sprintf(`insert into counter_%d select * from counter`, year)
q := fmt.Sprintf(`create table counter_%d as select * from counter`, thisYear)
_, err = tx.Exec(q)
if err != nil {
logError(err, "error running insert into")