move teamids map to cmd
This commit is contained in:
		
							parent
							
								
									8be2e5de9e
								
							
						
					
					
						commit
						a0ae93a2b2
					
				| 
						 | 
				
			
			@ -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
 | 
			
		||||
)
 | 
			
		||||
							
								
								
									
										36
									
								
								cmd/root.go
								
								
								
								
							
							
						
						
									
										36
									
								
								cmd/root.go
								
								
								
								
							| 
						 | 
				
			
			@ -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
 | 
			
		||||
	}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -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 {
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue