diff --git a/cmd/schedule.go b/cmd/schedule.go index f438eca..077ac50 100644 --- a/cmd/schedule.go +++ b/cmd/schedule.go @@ -27,6 +27,7 @@ import ( var team teamFlag var date string +var sportId int var teamIDs = map[string]statsapi.TeamID{ "laa": statsapi.LAA, "az": statsapi.AZ, @@ -68,7 +69,7 @@ func schedule(cmd *cobra.Command, args []string) { id = strconv.Itoa(int(teamIDs[string(team)])) } - sched, err := statsapi.RequestSchedule("1", id, date) + sched, err := statsapi.RequestSchedule(strconv.Itoa(sportId), id, date) if err != nil { log.Fatal(err) } @@ -98,4 +99,5 @@ func init() { // is called directly, e.g.: scheduleCmd.Flags().VarP(&team, "team", "t", "team to get schedule for (atl, az, bal, bos, chc, cin, cle, col, cws, det, hou, kc, laa, lad, mia, mil, min, nym, nyy, oak, phi, pit, sd, sea, sf, stl, tb, tex, tor, wsh)") scheduleCmd.Flags().StringVarP(&date, "date", "d", "", "date to get schedule for (YYYY-MM-DD)") + scheduleCmd.Flags().IntVarP(&sportId, "sport", "s", 1, "sport ID get schedule for (default: 1 (MLB))") }