69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"os"
|
|||
|
|
"path"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
rotates "github.com/lestrrat-go/file-rotatelogs"
|
|||
|
|
log "github.com/sirupsen/logrus"
|
|||
|
|
easy "github.com/t-tomalak/logrus-easy-formatter"
|
|||
|
|
|
|||
|
|
"github.com/huoxue1/study_xxqg/lib"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
config = lib.GetConfig()
|
|||
|
|
logFormatter := &easy.Formatter{
|
|||
|
|
TimestampFormat: "2006-01-02 15:04:05",
|
|||
|
|
LogFormat: "[%time%] [%lvl%]: %msg% \n",
|
|||
|
|
}
|
|||
|
|
w, err := rotates.New(path.Join("logs", "%Y-%m-%d.log"), rotates.WithRotationTime(time.Hour*24))
|
|||
|
|
if err != nil {
|
|||
|
|
log.Errorf("rotates init err: %v", err)
|
|||
|
|
panic(err)
|
|||
|
|
}
|
|||
|
|
log.SetOutput(w)
|
|||
|
|
log.SetFormatter(logFormatter)
|
|||
|
|
level, err := log.ParseLevel(config.LogLevel)
|
|||
|
|
|
|||
|
|
log.SetLevel(level)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
config lib.Config
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
_, err := os.Stat(`./config/`)
|
|||
|
|
if err != nil {
|
|||
|
|
os.Mkdir("./config/", 0666)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func main() {
|
|||
|
|
log.Infoln(`// 刷课模式，默认为1，
|
|||
|
|
1：只刷文章何视频
|
|||
|
|
2：只刷文章和视频和每日答题
|
|||
|
|
3：刷文章和视频和每日答题每周答题和专项答题`)
|
|||
|
|
log.Infoln("检测到模式", config.Model)
|
|||
|
|
core := lib.Core{}
|
|||
|
|
defer core.Quit()
|
|||
|
|
core.Init()
|
|||
|
|
login, err := core.Login()
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
core.LearnArticle(login)
|
|||
|
|
core.LearnVideo(login)
|
|||
|
|
if config.Model == 2 {
|
|||
|
|
core.RespondDaily(login, "daily")
|
|||
|
|
} else if config.Model == 3 {
|
|||
|
|
core.RespondDaily(login, "daily")
|
|||
|
|
core.RespondDaily(login, "weekly")
|
|||
|
|
core.RespondDaily(login, "special")
|
|||
|
|
}
|
|||
|
|
}
|