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())