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)
