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

move teamids map to cmd

This commit is contained in:
filifa 2024-07-14 19:12:55 -05:00
parent 8be2e5de9e
commit a0ae93a2b2
3 changed files with 69 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
cmd/internal/statsapi/common.go → cmd/internal/statsapi/teams.go
View File

@ -15,35 +15,37 @@ this program. If not, see <http://www.gnu.org/licenses/>.
*/ */
package statsapi package statsapi
var TeamIds = map[string]int{ type TeamID int
"laa": 108,
"az": 109, const (
"bal": 110, LAA TeamID = 108
"bos": 111, AZ TeamID = 109
"chc": 112, BAL TeamID = 110
"cin": 113, BOS TeamID = 111
"cle": 114, CHC TeamID = 112
"col": 115, CIN TeamID = 113
"det": 116, CLE TeamID = 114
"hou": 117, COL TeamID = 115
"kc": 118, DET TeamID = 116
"lad": 119, HOU TeamID = 117
"wsh": 120, KC TeamID = 118
"nym": 121, LAD TeamID = 119
"oak": 133, WSH TeamID = 120
"pit": 134, NYM TeamID = 121
"sd": 135, OAK TeamID = 133
"sea": 136, PIT TeamID = 134
"sf": 137, SD TeamID = 135
"stl": 138, SEA TeamID = 136
"tb": 139, SF TeamID = 137
"tex": 140, STL TeamID = 138
"tor": 141, TB TeamID = 139
"min": 142, TEX TeamID = 140
"phi": 143, TOR TeamID = 141
"atl": 144, MIN TeamID = 142
"cws": 145, PHI TeamID = 143
"mia": 146, ATL TeamID = 144
"nyy": 147, CWS TeamID = 145
"mil": 158, MIA TeamID = 146
} NYY TeamID = 147
MIL TeamID = 158
)

36
cmd/root.go
View File

@ -29,18 +29,50 @@ import (
) )
var Team TeamFlag var Team TeamFlag
var teamIDs = map[string]statsapi.TeamID{
"laa": statsapi.LAA,
"az": statsapi.AZ,
"bal": statsapi.BAL,
"bos": statsapi.BOS,
"chc": statsapi.CHC,
"cin": statsapi.CIN,
"cle": statsapi.CLE,
"col": statsapi.COL,
"det": statsapi.DET,
"hou": statsapi.HOU,
"kc": statsapi.KC,
"lad": statsapi.LAD,
"wsh": statsapi.WSH,
"nym": statsapi.NYM,
"oak": statsapi.OAK,
"pit": statsapi.PIT,
"sd": statsapi.SD,
"sea": statsapi.SEA,
"sf": statsapi.SF,
"stl": statsapi.STL,
"tb": statsapi.TB,
"tex": statsapi.TEX,
"tor": statsapi.TOR,
"min": statsapi.MIN,
"phi": statsapi.PHI,
"atl": statsapi.ATL,
"cws": statsapi.CWS,
"mia": statsapi.MIA,
"nyy": statsapi.NYY,
"mil": statsapi.MIL,
}
func getGamePk() (string, error) { func getGamePk() (string, error) {
if Team == "" { if Team == "" {
return "", errors.New("need team ID") return "", errors.New("need team ID")
} }
id, ok := statsapi.TeamIds[string(Team)] id, ok := teamIDs[string(Team)]
if !ok { if !ok {
return "", errors.New("invalid team ID") return "", errors.New("invalid team ID")
} }
sched, err := statsapi.RequestSchedule("1", strconv.Itoa(id)) sched, err := statsapi.RequestSchedule("1", strconv.Itoa(int(id)))
if err != nil { if err != nil {
return "", err return "", err
} }

4
cmd/teamflag.go
View File

@ -2,8 +2,6 @@ package cmd
import ( import (
"errors" "errors"
"scm.dairydemon.net/filifa/mlblive/cmd/internal/statsapi"
) )
type TeamFlag string type TeamFlag string
@ -14,7 +12,7 @@ func (t *TeamFlag) String() string {
func (t *TeamFlag) Set(v string) error { func (t *TeamFlag) Set(v string) error {
var err error var err error
_, ok := statsapi.TeamIds[v] _, ok := teamIDs[v]
if !ok { if !ok {
err = errors.New("invalid team ID") err = errors.New("invalid team ID")
} else { } else {