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

Merge pull request #321 from velour/fixcountdown 

countdown: maybe use actually January
This commit is contained in:
Chris Sexton 2020-11-30 15:01:36 -05:00 committed by GitHub
parent d7839c96d8 aa9dc12c9b
commit c8e1956099
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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")