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

View File

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

View File

@ -29,18 +29,50 @@ import (
)
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) {
if Team == "" {
return "", errors.New("need team ID")
}
id, ok := statsapi.TeamIds[string(Team)]
id, ok := teamIDs[string(Team)]
if !ok {
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 {
return "", err
}

View File

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