diff --git a/leagues.py b/leagues.py index c08be59..893943d 100644 --- a/leagues.py +++ b/leagues.py @@ -102,6 +102,14 @@ class league_structure(object): teams += teams_list return teams + def team_names_in_league(self): + teams = [] + for division in self.league.values(): + for teams_list in division.values(): + for team in teams_list: + teams.append(team.name) + return teams + def teams_in_subleague(self, subleague_name): teams = [] if subleague_name in self.league.keys(): diff --git a/the_prestige.py b/the_prestige.py index 515b1d9..ee6960e 100644 --- a/the_prestige.py +++ b/the_prestige.py @@ -994,13 +994,20 @@ class LeagueScheduleCommand(Command): league = leagues.load_league_file(league_name) current_series = league.day_to_series_num(league.day) if str(current_series+1) in league.schedule.keys(): - sched_embed = discord.Embed(title=f"{league.name}'s Schedule:") + sched_embed = discord.Embed(title=f"{league.name}'s Schedule:", color=discord.Color.magenta()) days = [0,1,2,3] for day in days: if str(current_series+day) in league.schedule.keys(): schedule_text = "" + teams = league.team_names_in_league() for game in league.schedule[str(current_series+day)]: schedule_text += f"**{game[0]}** @ **{game[1]}**\n" + teams.pop(teams.index(game[0])) + teams.pop(teams.index(game[1])) + if len(teams) > 0: + schedule_text += "Resting:\n" + for team in teams: + schedule_text += f"**{team}**\n" sched_embed.add_field(name=f"Days {((current_series+day-1)*league.series_length) + 1} - {(current_series+day)*(league.series_length)}", value=schedule_text, inline = False) await msg.channel.send(embed=sched_embed) else: