diff --git a/main.go b/main.go index f41fd2d..062689b 100644 --- a/main.go +++ b/main.go @@ -3,6 +3,7 @@ package main import ( "log" "os" + "strconv" "time" "git.l9labs.ru/anufriev.g.a/l9_stud_bot/modules/database" @@ -31,8 +32,10 @@ func main() { if err != nil { log.Fatal(err) } - // TODO: загнать первую неделю в переменную среды - bot.Week = 5 + bot.Week, err = strconv.Atoi(os.Getenv("START_WEEK")) + if err != nil { + log.Fatal(err) + } bot.WkPath = os.Getenv("WK_PATH") now, _ := time.Parse("2006-01-02 15:04 -07", "2023-02-06 11:20 +04") for update := range *bot.Updates { diff --git a/modules/tg/bot.go b/modules/tg/bot.go index 8931bbb..ecff372 100644 --- a/modules/tg/bot.go +++ b/modules/tg/bot.go @@ -27,6 +27,11 @@ type Bot struct { var env_keys = []string{ "TELEGRAM_APITOKEN", "TELEGRAM_TEST_USER", + "WK_PATH", + "MYSQL_USER", + "MYSQL_PASS", + "MYSQL_DB", + "START_WEEK", } func CheckEnv() error {