filifa
/
mlblive
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use sport abbrevs instead of ids

This commit is contained in:
filifa 2024-07-28 17:35:56 -05:00
parent 957077b581
commit c20a289c06
1 changed files with 36 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
cmd/schedule.go
View File

@ -25,9 +25,9 @@ import (
"scm.dairydemon.net/filifa/mlblive/cmd/internal/statsapi" "scm.dairydemon.net/filifa/mlblive/cmd/internal/statsapi"
) )
var team teamFlag
var date string var date string
var sportId int
var team teamFlag
var teamIDs = map[string]statsapi.TeamID{ var teamIDs = map[string]statsapi.TeamID{
"laa": statsapi.LAA, "laa": statsapi.LAA,
"az": statsapi.AZ, "az": statsapi.AZ,
@ -61,15 +61,44 @@ var teamIDs = map[string]statsapi.TeamID{
"mil": statsapi.MIL, "mil": statsapi.MIL,
} }
var sport string
var sportIDs = map[string]statsapi.SportID{
"mlb": statsapi.MLB,
"aaa": statsapi.AAA,
"aa": statsapi.AA,
"higha": statsapi.HighA,
"a": statsapi.A,
"rookie": statsapi.Rookie,
"winter": statsapi.Winter,
"milb": statsapi.MILB,
"indie": statsapi.Independent,
"negro": statsapi.Negro,
"kbo": statsapi.KBO,
"npb": statsapi.NPB,
"int": statsapi.International,
"int18u": statsapi.International18U,
"int16u": statsapi.International16U,
"intamateur": statsapi.InternationalAmateur,
"college": statsapi.College,
"hs": statsapi.HighSchool,
}
func schedule(cmd *cobra.Command, args []string) { func schedule(cmd *cobra.Command, args []string) {
var id string var teamId string
if team == "" { if team == "" {
id = string(team) teamId = string(team)
} else { } else {
id = strconv.Itoa(int(teamIDs[string(team)])) teamId = strconv.Itoa(int(teamIDs[string(team)]))
} }
sched, err := statsapi.RequestSchedule(strconv.Itoa(sportId), id, date) var sportId string
if sport == "" {
sportId = string(sport)
} else {
sportId = strconv.Itoa(int(sportIDs[sport]))
}
sched, err := statsapi.RequestSchedule(sportId, teamId, date)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
@ -99,5 +128,5 @@ func init() {
// is called directly, e.g.: // 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().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().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))") scheduleCmd.Flags().StringVarP(&sport, "sport", "s", "mlb", "sport to get schedule for (default: MLB)")
} }