def roster_management(): if "gm_email" in session: team = Team() team_data = team.get_team_data_by_gm_email(session["gm_email"]) # Add team_id to session for td in team_data: app.logger.debug("team_id: %s" % td["team_id"]) session["team_id"] = td["team_id"] roster_form = RosterManagementForm() if roster_form.validate_on_submit(): if roster_form.add_button.data: team.add_player(roster_form.first_name.data, roster_form.last_name.data) for t in team_data: team.assign_player_to_team(roster_form.first_name.data, roster_form.last_name.data, t["team_name"]) app.logger.debug( "%s %s is added to the roster." % (roster_form.first_name.data, roster_form.last_name.data)) return redirect(url_for('roster_management')) else: app.logger.debug( "NOT ADDED: %s %s" % (roster_form.first_name.data, roster_form.last_name.data)) return render_template("gm/roster_management.html", team_data=team_data, form=roster_form)
def __init__(self, games, schoolA, schoolB, playersA, playersB): self.games = games teamA = Team(schoolA) teamB = Team(schoolB) for player in playersA.split(","): #splitting the names by the comma (in the input-Kevin:1,) and adding it to the player list teamA.add_player(player) for player in playersB.split(","): #same thing with the opponent teamB.add_player(player) self.teamA = teamA self.teamB = teamB self.playersA = playersA self.playersB = playersB
class TeamTest(unittest.TestCase): def setUp(self): self.player1 = Player(None, None) self.team = Team("test", bullet.btVector3(0, 0, 0), [self.player1], "ignore", 0) def test_add_player(self): p2 = Player(None, None) self.team.add_player(p2) self.assertIn(p2, self.team.players) def test_remove_player(self): self.team.remove_player(self.player1) self.assertNotIn(self.player1, self.team.players) def test_add_score(self): self.team.add_score(5) self.assertEqual(self.team.score, 5)
def parse_teams(teamcards, season_title, player_dict): team_dict_maker = TeamDictionaryMaker({}) for teamcard in teamcards: new_team = Team() new_team.active_season_title = season_title links = teamcard.find_elements_by_tag_name("a") new_team.name = links[0].wrapped_element.text new_team.players = [] player_name = links[2].get_attribute("href").split('/')[-1] player_name = player_name.replace(player_name[0], player_name[0].lower(), 1) if player_dict.get(player_name, Player("Bad name", -1)).name == "Bad name": player_name = player_name.replace(player_name[0], player_name[0].upper(), 1) new_team.add_player(player_dict.get(player_name, player_dict.get("Unavailable"))) player_name = links[4].get_attribute("href").split('/')[-1] player_name = player_name.replace(player_name[0], player_name[0].lower(), 1) if player_dict.get(player_name, Player("Bad name", -1)).name == "Bad name": player_name = player_name.replace(player_name[0], player_name[0].upper(), 1) new_team.add_player(player_dict.get(player_name, player_dict.get("Unavailable"))) player_name = links[6].get_attribute("href").split('/')[-1] player_name = player_name.replace(player_name[0], player_name[0].lower(), 1) if player_dict.get(player_name, Player("Bad name", -1)).name == "Bad name": player_name = player_name.replace(player_name[0], player_name[0].upper(), 1) new_team.add_player(player_dict.get(player_name, player_dict.get("Unavailable"))) team_dict_maker.fix_name(new_team) return team_dict_maker.team_dict
def team_registration(): if "gm_email" in session: return "You already have a team registered!" form = TeamRegistration() team = Team() if form.validate_on_submit(): team.add_user(form.gm_email.data, form.password.data, form.gm_name.data) team.add_team(form.team_name.data, form.gm_name.data) for player in form.players.data: team.add_player(player['first_name'], player['last_name']) team.assign_player_to_team(player['first_name'], player['last_name'], form.team_name.data) flash("Account created for %s" % form.gm_email.data, 'success') return redirect(url_for('team_registration_confirmation')) else: # return form.errors return render_template('team_registration.html', show_home=0, title='Team Registration', form=form)
def test_add_player(self): teamA = Team("Cate") playerA = Player("Kevin") teamA.add_player(playerA) self.assertEqual([playerA],teamA.players)