예제 #1
0
    def player_save(self, team, team_object):

        self.stdout.write('Get team {team_name}'.format(team_name=team.full_name))
        for player_object in team_object['players']:
            try:
                player = team.players.get(
                    first_name=player_object['first_name'],
                    last_name=player_object['last_name'])
                self.stdout.write('Update player {first_name} {last_name}'.format(
                    first_name=player_object['first_name'],
                    last_name=player_object['last_name']))
            except Player.DoesNotExist:
                self.stdout.write('Create player {first_name} {last_name}'.format(
                    first_name=player_object['first_name'],
                    last_name=player_object['last_name']))
                player = Player()

            for attr, value in player_object.items():
                if attr == 'birthplace':
                    birth_places = value.split(', ')
                    player.birthplace_city = birth_places[0]
                    player.birthplace_country = birth_places[1]
                    player.is_USA = len(birth_places) == 3
                else:
                    setattr(player, attr, value)
            player.save()
            team.players.add(player)