package main import ( "context" "errors" "log" "os" "os/signal" "syscall" "time" "github.com/nprasad2077/NBA_Go/services" "github.com/nprasad2077/NBA_Go/utils" "gorm.io/gorm" ) // importPlayerAdvanced fetches and stores advanced stats for seasons func importPlayerAdvanced(db *gorm.DB) { for season := 2001; season <= 2001; season++ { if err := services.FetchAndStorePlayerAdvancedScrapedStats(db, season, false); err != nil { log.Printf("advanced import failed for %d: %v", season, err) } log.Printf("Advanced import for season: %d", season) time.Sleep(2000 * time.Millisecond) utils.SleepWithJitter(1000 * time.Millisecond) } } // importPlayerAdvancedPlayoffs fetches and stores advanced stats for playoffs seasons func importPlayerAdvancedPlayoffs(db *gorm.DB) { for season := 2001; season <= 2001; season++ { if err := services.FetchAndStorePlayerAdvancedScrapedStats(db, season, true); err != nil { log.Printf("advanced import failed for %d: %v", season, err) } log.Printf("Advanced Playoffs import for season: %d", season) time.Sleep(2000 * time.Millisecond) utils.SleepWithJitter(1500 * time.Millisecond) } } // importPlayerTotalsScrape fetches & stores scraped regular-season total stats func importPlayerTotalsScrape(db *gorm.DB) { for season := 2001; season <= 2001; season++ { if err := services.FetchAndStorePlayerTotalScrapedStats(db, season, false); err != nil { log.Printf("scraped totals import failed for %d: %v", season, err) } log.Printf("Player Totals import for season: %d", season) time.Sleep(2000 * time.Millisecond) utils.SleepWithJitter(1250 * time.Millisecond) } } // importPlayerPlayoffsScrape fetches & stores scraped playoff total stats func importPlayerTotalsPlayoffsScrape(db *gorm.DB) { for season := 2001; season <= 2001; season++ { if err := services.FetchAndStorePlayerTotalScrapedStats(db, season, true); err != nil { log.Printf("scraped playoffs import failed for %d: %v", season, err) } log.Printf("Player Playoffs Totals import for season: %d", season) time.Sleep(2000 * time.Millisecond) utils.SleepWithJitter(1700 * time.Millisecond) } } // importGameSchedules fetches and stores game schedules func importGameSchedules(db *gorm.DB) { months := []string{"october", "november", "december", "january", "february", "march", "april", "may", "june"} for season := 2001; season <= 2001; season++ { log.Printf("--- Starting Game Schedule Import for Season: %d ---", season) for _, month := range months { if err := services.FetchAndStoreGameSchedule(db, season, month); err != nil { log.Printf("Game schedule import failed for %s %d: %v", month, season, err) } log.Printf("Game schedule import for %s, %d complete.", month, season) time.Sleep(2500 * time.Millisecond) utils.SleepWithJitter(1800 * time.Millisecond) } log.Printf("--- Finished Game Schedule Import for Season: %d ---", season) } } // importBoxScores fetches and stores all box score data (line scores, player/team stats) // in safe 20-day session chunks with cool-off pauses and immediate database upserts. func importBoxScores(db *gorm.DB) { ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM) defer stop() from := time.Date(2000, time.October, 31, 0, 0, 0, 0, time.UTC) to := time.Date(2001, time.June, 30, 23, 59, 59, 0, time.UTC) chunkDays := 20 // 20-day session chunks coolOff := 20 * time.Second // 20s cool-off with jitter between chunks skipExisting := true // Automatically skip games that already have line scores if err := services.FetchAndStoreBoxScoreDataChunked(ctx, db, from, to, chunkDays, coolOff, skipExisting); err != nil { if errors.Is(err, context.Canceled) { log.Println("👋 Box score import interrupted by user. Safe to resume anytime!") return } log.Fatalf("Box score import failed: %v", err) } log.Println("--- Finished Box Score Data Import ---") } // importPlayerShotCharts fetches shot charts for a PREDEFINED list of players for a given range of seasons. func importPlayerShotCharts(db *gorm.DB) { log.Println("--- Starting Player Shot Chart Import from Predefined List ---") startSeason := 2026 endSeason := 2017 targetPlayerIDs := []string{ // "jamesle01", "curryst01", // "jokicni01", // "doncilu01", // "hardeja01", // "irvinky01", // "duranke01", // "youngtr01", // "gilgesh01", // "brunsja01", // "edwaran01", // "mitchdo01", // "derozde01", // "westbru01", // "butleji01", // "davisan02", // "leonaka01", // "tatumja01", // "hardeja01", // "curryst01", // "duranke01", // "jamesle01", // "lillada01", // "westbru01", // "georgpa01", // "derozde01", // "thomais02", // "davisan02", // "townska01", // "walkeke02", // "bealbr01", "antetgi01", "bookede01", // "youngtr01", // "doncilu01", // "jokicni01", "randlju01", "embiijo01", // "tatumja01", // "gilgesh01", // "edwaran01", "mitchdo01", // "brunsja01", "herroty01", "brownja02", "maxeyty01", "murraja01", } if len(targetPlayerIDs) == 0 { log.Println("⚠️ Player ID list is empty. Skipping shot chart import.") return } log.Printf("Found %d players in the predefined list. Starting shot chart import for seasons %d down to %d.", len(targetPlayerIDs), startSeason, endSeason) for i, playerID := range targetPlayerIDs { log.Printf("(%d/%d) Importing shot charts for player: %s", i+1, len(targetPlayerIDs), playerID) err := services.FetchAndStoreShotChartScrapedForPlayer(db, playerID, startSeason, endSeason) if err != nil { log.Printf("❌ Shot chart import failed for player %s: %v", playerID, err) } utils.SleepWithJitter(10000 * time.Millisecond) } log.Println("--- Finished Player Shot Chart Import from Predefined List ---") } // importMarkPlayoffGames marks games as playoff using the dedicated Basketball Reference playoff schedule. func importMarkPlayoffGames(db *gorm.DB) { for season := 2001; season <= 2001; season++ { if err := services.FetchAndMarkPlayoffGames(db, season); err != nil { log.Printf("playoff marking failed for %d: %v", season, err) } log.Printf("Playoff games marked for season: %d", season) } }