diff --git a/models/tbot.go b/models/tbot.go index 21c4e0d..8b94ad4 100644 --- a/models/tbot.go +++ b/models/tbot.go @@ -3,6 +3,7 @@ package models import ( "fmt" "net/http" + "regexp" "time" "github.com/beego/beego/v2/core/logs" @@ -53,6 +54,9 @@ func initTgBot() { if m.Sender.ID != Config.TelegramUserID { return } + if regexp.MustCompile(`.js$`).FindString(m.Document.FileName) == "" && regexp.MustCompile(`.py$`).FindString(m.Document.FileName) == "" { + return + } b.Download(m.Document.MediaFile(), ExecPath+"/scripts/"+m.Document.FileName) m.Text = fmt.Sprintf("run " + m.Document.FileName) handle(m)