def test_parse(self):
with open('test_data/teams.csv', 'r') as f:
teams = CSVTeamsParser.parse(f.read())
correct_teams = {
1, 2, 3, 5, 9, 10, 15, 16, 18, 22, 100, 101, 102, 103, 999
}
self.assertEqual(teams.difference(correct_teams), set())
def post(self, event_key_id):
self._require_admin()
event = Event.get_by_id(event_key_id)
teams_csv = self.request.get('teams_csv')
team_numbers = CSVTeamsParser.parse(teams_csv)
event_teams = []
for team_number in team_numbers:
event_teams.append(ndb.Key(EventTeam, '{}_frc{}'.format(event.key.id(), team_number)))
EventTeamManipulator.delete_keys(event_teams)
self.redirect("/admin/event/" + event.key_name)
def post(self, event_key_id):
self._require_admin()
event = Event.get_by_id(event_key_id)
teams_csv = self.request.get('teams_csv')
team_numbers = CSVTeamsParser.parse(teams_csv)
event_teams = []
teams = []
for team_number in team_numbers:
event_teams.append(EventTeam(id=event.key.id() + '_frc{}'.format(team_number),
event=event.key,
team=ndb.Key(Team, 'frc{}'.format(team_number)),
year=event.year))
teams.append(Team(id='frc{}'.format(team_number),
team_number=int(team_number)))
EventTeamManipulator.createOrUpdate(event_teams)
TeamManipulator.createOrUpdate(teams)
self.redirect("/admin/event/" + event.key_name)
def test_parse(self):
with open('test_data/teams.csv', 'r') as f:
teams = CSVTeamsParser.parse(f.read())
correct_teams = {1, 2, 3, 5, 9, 10, 15, 16, 18, 22, 100, 101, 102, 103, 999}
self.assertEqual(teams.difference(correct_teams), set())
