def _get_all_team_info():
teams = []
for team in yahoo_api.get_all_teams():
team_id = int(team.find('ns:team_id', {'ns': 'http://fantasysports.yahooapis.com/fantasy/v2/base.rng'}).text)
teams.append(Team.from_xml_api_data(yahoo_api.get_team_with_matchups(team_id)))
return teams
def test_xml_constructor(self):
ns = {"ns": "http://fantasysports.yahooapis.com/fantasy/v2/base.rng"}
xml = et.parse('./sample-api-responses/get_team.xml').find(
'ns:team', ns)
result = Team.from_xml_api_data(xml)
expected = Team(
**{
"name": "K.E.V.I.N",
"id": 5,
"owner": "Kevin",
"is_my_team": True,
"waiver_priority": 5,
"move_count": 26,
"trade_count": 0,
"matchups": [],
"average_stats": None,
})
self.assertEqual(expected.name, result.name)
self.assertEqual(expected.id, result.id)
self.assertEqual(expected.owner, result.owner)
self.assertEqual(expected.is_my_team, result.is_my_team)
self.assertEqual(expected.waiver_priority, result.waiver_priority)
self.assertEqual(expected.move_count, result.move_count)
self.assertEqual(expected.trade_count, result.trade_count)
self.assertEqual(expected.matchups, result.matchups)
self.assertEqual(expected.average_stats, result.average_stats)
def team(id):
raw_team_data = yahoo_api.get_team(id)
if not raw_team_data:
return Response(response="Team does not exist", status=400)
team = Team.from_xml_api_data(yahoo_api.get_team(id))
return Response(
response=json.dumps(team, default=Team.serialize),
status=200,
mimetype='application/json'
)
def _get_basic_team_info():
return [Team.from_xml_api_data(x) for x in yahoo_api.get_all_teams()]
