mirror of
https://github.com/nprasad2077/NBA_Go.git
synced 2026-09-22 14:05:13 +00:00
179 lines
6.1 KiB
Go
179 lines
6.1 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"time"
|
|
"fmt"
|
|
|
|
"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 := 2026; season <= 2026; 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(1300 * time.Millisecond)
|
|
utils.SleepWithJitter(1000 * time.Millisecond)
|
|
}
|
|
}
|
|
|
|
// importPlayerAdvancedPlayoffs fetches and stores advanced stats for playoffs seasons
|
|
func importPlayerAdvancedPlayoffs(db *gorm.DB) {
|
|
for season := 2026; season <= 2026; 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(1300 * time.Millisecond)
|
|
utils.SleepWithJitter(1500 * time.Millisecond)
|
|
}
|
|
}
|
|
|
|
// importPlayerTotalsScrape fetches & stores scraped regular-season total stats
|
|
func importPlayerTotalsScrape(db *gorm.DB) {
|
|
for season := 2026; season <= 2026; 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(1300 * time.Millisecond)
|
|
utils.SleepWithJitter(1250 * time.Millisecond)
|
|
}
|
|
}
|
|
|
|
// importPlayerPlayoffsScrape fetches & stores scraped playoff total stats
|
|
func importPlayerTotalsPlayoffsScrape(db *gorm.DB) {
|
|
for season := 2026; season <= 2026; 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(1300 * time.Millisecond)
|
|
utils.SleepWithJitter(1700 * time.Millisecond)
|
|
}
|
|
}
|
|
|
|
// importGameSchedules fetches and stores game schedules
|
|
func importGameSchedules(db *gorm.DB) {
|
|
// An NBA season typically runs from October to June
|
|
months := []string{
|
|
// "september", "october", "november", "december", "january",
|
|
// "february", "march", "april", "may", "june",
|
|
"january", "february",
|
|
// "february", "march", "april", "may", "june",
|
|
// "october", "november", "december",
|
|
}
|
|
|
|
for season := 2025; season <= 2026; season++ {
|
|
log.Printf("--- Starting Game Schedule Import for Season: %d ---", season)
|
|
for _, month := range months {
|
|
// The service will print a warning and skip if a month has no data (e.g. May/June for a season not yet finished)
|
|
if err := services.FetchAndStoreGameSchedule(db, season, month); err != nil {
|
|
// Log the error but continue to the next month/season
|
|
log.Printf("Game schedule import failed for %s %d: %v", month, season, err)
|
|
}
|
|
log.Printf("Game schedule import for %s, %d complete.", month, season)
|
|
// Respectful delay between requests
|
|
time.Sleep(1400 * 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)
|
|
// for games within a recent date range.
|
|
func importBoxScores(db *gorm.DB) {
|
|
// now := time.Now()
|
|
|
|
from := time.Date(2026, time.January, 27, 0, 0, 0, 0, time.UTC)
|
|
// from := time.Date(now.Year(), now.Month(), now.Day()-1, 5, 30, 0, 0, time.UTC)
|
|
to := time.Date(2026, time.February, 28, 5, 30, 0, 0, time.UTC)
|
|
|
|
dateRangeComment := fmt.Sprintf("--- Starting Box Score Data Import for games between %s and %s ---",
|
|
from.Format("January 2, 2006"),
|
|
to.Format("January 2, 2006"))
|
|
|
|
log.Println(dateRangeComment)
|
|
|
|
if err := services.FetchAndStoreBoxScoreDataForDateRange(db, from, to); err != nil {
|
|
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 ---")
|
|
|
|
// 1. Define the season range. Your service fetches from newest to oldest.
|
|
startSeason := 2025
|
|
endSeason := 2017 // Basketball-Reference has data going back this far.
|
|
|
|
// 2. Define the static list of player IDs to import.
|
|
// You can add or remove any Basketball-Reference player ID here.
|
|
// Examples:
|
|
// LeBron James: "jamesle01"
|
|
// Stephen Curry: "curryst01"
|
|
// Nikola Jokić: "jokicni01"
|
|
// Luka Dončić: "doncilu01"
|
|
targetPlayerIDs := []string{
|
|
"jamesle01",
|
|
"curryst01",
|
|
"jokicni01",
|
|
"doncilu01",
|
|
"hardeja01",
|
|
"irvinky01",
|
|
"duranke01",
|
|
"youngtr01",
|
|
// "lillada01",
|
|
"gilgesh01",
|
|
"brunsja01",
|
|
"edwaran01",
|
|
// "mitchdo01",
|
|
// "bookede01",
|
|
"derozde01",
|
|
// "aldrila01",
|
|
"westbru01",
|
|
// "paulch01",
|
|
"butleji01",
|
|
"davisan02",
|
|
"leonaka01",
|
|
// "youngtr01",
|
|
"tatumja01",
|
|
|
|
// Add more player IDs here as needed
|
|
}
|
|
|
|
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)
|
|
|
|
// 3. Loop through each player ID and fetch their shot chart data.
|
|
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 the error but continue with the next player.
|
|
log.Printf("❌ Shot chart import failed for player %s: %v", playerID, err)
|
|
}
|
|
|
|
// 4. Be a good internet citizen. Pause between scraping each player.
|
|
utils.SleepWithJitter(10000 * time.Millisecond)
|
|
}
|
|
|
|
log.Println("--- Finished Player Shot Chart Import from Predefined List ---")
|
|
}
|