add nultie gpouring
This commit is contained in:
+40
-4
@@ -4,17 +4,53 @@ import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
"github.com/spf13/viper"
|
||||
|
||||
_ "github.com/go-sql-driver/mysql"
|
||||
|
||||
"git.kplus.net.ua/yevhen/resource-scraper/internal/releaseparser/groups/moonblood"
|
||||
)
|
||||
|
||||
func main() {
|
||||
stats, err := moonblood.AnalyzeGroup()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
if len(os.Args) < 2 {
|
||||
log.Fatal("usage: srrdb <releaser> [<releaser>...]")
|
||||
}
|
||||
releasers := os.Args[1:]
|
||||
|
||||
// Читаємо config/secret.yaml (без змісту в коді — тільки потрібні ключі)
|
||||
viper.SetConfigFile("config/secret.yaml")
|
||||
if err := viper.ReadInConfig(); err != nil {
|
||||
log.Fatal("failed to read config/secret.yaml: ", err)
|
||||
}
|
||||
|
||||
data, err := json.MarshalIndent(stats, "", " ")
|
||||
drv := viper.GetString("db.driver")
|
||||
dsn := viper.GetString("db.data_source_name")
|
||||
|
||||
dbase, err := sqlx.Connect(drv, dsn)
|
||||
if err != nil {
|
||||
panic("Failed to connect to the database: " + err.Error())
|
||||
}
|
||||
defer dbase.Close()
|
||||
|
||||
out := make(map[string]moonblood.GroupStats, len(releasers))
|
||||
|
||||
for _, releaser := range releasers {
|
||||
var titles []string
|
||||
|
||||
err = dbase.Select(&titles,
|
||||
`SELECT title FROM theamok_mdb.external_sources
|
||||
WHERE type = 'prescene' AND releaser = ?`, releaser)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
out[releaser] = moonblood.AnalyzeTitles(titles)
|
||||
}
|
||||
|
||||
data, err := json.MarshalIndent(out, "", " ")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user