def handle(self, *args, **options): # clear database Team.objects.all().delete() User.objects.all().delete() Player.objects.all().delete() with open('../../data/teams.csv') as csvfile: teams = csv.DictReader(csvfile) # import ipdb; ipdb.set_trace() for idx, (team_data, color) in enumerate(zip(teams, color_palette)): if team_data["Team Name"] == "": continue team = Team() team.name = team_data["Team Name"] team.university = team_data["University"] team.token = team_data["Token"] team.color = color team.flag = "flag%s" % idx team.score = 0 team.save() teams = Team.objects.all() print("Loaded {} teams:".format(teams.count())) for team in teams: print(team)
def handle(self, *args, **options): # clear database Team.objects.all().delete() User.objects.all().delete() Player.objects.all().delete() with open('../../data/teams.json') as json_data: json_teams = json.load(json_data) for json_team in json_teams: team = Team() team.name = json_team["name"] team.university = json_team["university"] team.token = json_team["token"] team.color = json_team["color"] team.flag = json_team["flag"] team.score = json_team["score"] team.save() univ_mail = json_team["university"].replace(" ", "") + ".com" user = User() user.email = "player@" + univ_mail user.username = "******" + univ_mail[:-4].lower() user.first_name = "First-Name" user.last_name = "Last Name" user.set_password("12qwaszx") user.save() player = Player() user.player = player player.last_name = "Last Name" player.first_name = "First-Name" player.display_name = "Rebel_" + univ_mail player.team = team player.save() teams = Team.objects.all() print("Loaded {} teams:".format(teams.count())) for team in Team.objects.all(): player = team.player_set.all().first() print(" * " + team.name, "Player login:"******"12qwaszx")