class LobbyTest(unittest.TestCase):
def setUp(self):
self.lobby = Lobby(4)
def test_add(self):
self.lobby.add("Player")
self.assertIn("Player", self.lobby.player_list)
def test_multiple_add(self):
self.lobby.add("Player1", "Player2")
self.assertTrue("Player1" in self.lobby.player_list
and "Player2" in self.lobby.player_list,
msg=str(self.lobby.player_list))
def test_duplicate_add(self):
self.lobby.add("Player")
self.assertRaises(AssertionError, self.lobby.add, "Player")
def test_lobby_max(self):
self.lobby.add("1")
self.lobby.add("2")
self.lobby.add("3")
self.lobby.add("4")
self.assertRaises(AssertionError, self.lobby.add, "5")
def test_remove(self):
self.lobby.add("Player")
self.lobby.remove("Player")
self.assertNotIn("Player", self.lobby.player_list)
def test_remove_nonexisting(self):
self.assertRaises(ValueError, self.lobby.remove, "Player")
def test_shuffle(self):
self.lobby.add("1", "2", "3", "4")
self.lobby.shuffle()
self.assertTrue(self.lobby.get_team_one and self.lobby.get_team_two)
def test_swap(self):
self.lobby.add("Player1", "Player2")
self.lobby.team_one.add("Player1")
self.lobby.team_two.add("Player2")
self.lobby.swap("Player1", "Player2")
self.assertTrue("Player1" in self.lobby.team_two
and "Player2" in self.lobby.team_one)
def test_spots_left(self):
self.lobby.add("Player1", "Player2")
self.assertEqual(self.lobby.spots_left(), 2)
class Scrim(commands.Cog):
pattern_object = re.compile(r"<@!(\d*)>")
CREATE_LOBBY_MESSAGE = "Be sure to create a lobby first using the `?create NUMBER` command!"
def __init__(self, bot):
self.bot = bot
self.lobby = None
#--- helper functions ---#
##########################
def stringify_teams(self, team1: list, team2: list):
table = PrettyTable()
table.add_column("Team 1", team1)
table.add_column("Team 2", team2)
table.align = "c"
return table.get_string()
def extract_id(self, players):
new_list = []
for person in players:
matched = re.match(Scrim.pattern_object, person)
if matched:
print("Searching for USER with this ID:", matched.group(1))
user = self.bot.get_user(int(matched.group(1)))
new_list.append(user.name)
else:
new_list.append(person)
return new_list
#--- commands ---#
##################
@commands.command()
async def create(self, ctx, num_players: int):
'''Starts up the lobby, use number of TOTAL players.'''
self.lobby = Lobby(num_players)
await ctx.send(
"Lobby created for " + str(num_players) +
" total players. Use ?join to enter now or ?add NAME_HERE to add someone!"
)
@create.error
async def create_error(self, ctx, error):
if isinstance(error, commands.MissingRequiredArgument):
await ctx.send(
"You need to input the number of players on each team. For example, `?create 10`."
)
elif isinstance(error.original, AssertionError):
await ctx.send(
"Uneven amount of players. This is the TOTAL amount of players, so it should be even!"
)
else:
await ctx.send("Unforeseen error.")
print(str(type(error)))
print(error)
@commands.command()
async def join(self, ctx):
'''Join an existing lobby.'''
self.lobby.add(str(ctx.author.name))
if self.lobby.spots_left() == 0:
spots_message = "Lobby is now at max capacity (" + str(
self.lobby.max) + ")."
else:
spots_message = str(
self.lobby.spots_left()) + " player(s) can join."
await ctx.send(
str(ctx.author.name) + " has joined!" + " " + spots_message)
@join.error
async def join_error(self, ctx, error):
if isinstance(error.original, AttributeError):
await ctx.send(Scrim.CREATE_LOBBY_MESSAGE)
elif isinstance(error.original, AssertionError):
await ctx.send(error.original.args)
else:
print(error)
await ctx.send("Unexpected error. Try again maybe?")
@commands.command()
async def add(self, ctx, player, *args):
'''Add another player by @ing them or with a custom name.'''
total_players = (
player, ) + args #to consider player a part of a tuple
total_players = self.extract_id(total_players)
self.lobby.add(*total_players)
await ctx.send("Added " + str(total_players) + ".")
@commands.command()
async def clear(self, ctx):
'''Reset the lobby.'''
self.lobby = None
await ctx.send("Lobby cleared.")
@commands.command()
async def remove(self, ctx, player):
'''Removes another player. This command is case sensitive!'''
self.lobby.remove(player)
await ctx.send("Removed " + str(player))
@remove.error
async def remove_error(self, ctx, error):
if isinstance(error, commands.MissingRequiredArgument):
await ctx.send("Make sure you include what player you're removing!"
)
if isinstance(error.original, AttributeError):
await ctx.send(Scrim.CREATE_LOBBY_MESSAGE)
if isinstance(error.original, AssertionError):
await ctx.send(error.original.args)
elif isinstance(error.original, ValueError):
await ctx.send(
"Player can't be found! Usernames are case senstive. If you want to remove a player with spaces in their name, put quotation marks around it. `?remove \"Name With Spaces\"`"
)
else:
print(error)
await ctx.send("Unexpected error. Try again maybe?")
@commands.command()
async def showlist(self, ctx):
'''Outputs a list of the players in the lobby.'''
row_numbers = [i for i in range(1, len(self.lobby.player_list) + 1)]
await ctx.send("```\n" + tabulate({"Players": self.lobby.player_list},
headers="keys",
showindex=row_numbers,
tablefmt="psql") + "```")
@showlist.error
async def showlist_error(self, ctx, error):
if isinstance(error.original, AttributeError):
await ctx.send(Scrim.CREATE_LOBBY_MESSAGE)
elif isinstance(error.original, AssertionError):
await ctx.send(error.original.args)
else:
print(error)
await ctx.send("Unexpected error. Try again maybe?")
@commands.command()
async def showteams(self, ctx):
'''Outputs a table of the teams.'''
table = {
"Team 1": self.lobby.get_team_one(),
"Team 2": self.lobby.get_team_two()
}
roll_string = "Roll #" + str(self.lobby.num_rolls())
await ctx.send(roll_string + "```\n" +
tabulate(table, headers="keys", tablefmt="psql") +
"```")
@showteams.error
async def showteam_error(self, ctx, error):
if isinstance(error.original, AttributeError):
await ctx.send(Scrim.CREATE_LOBBY_MESSAGE)
elif isinstance(error.original, AssertionError):
await ctx.send(error.original.args)
else:
print(error)
await ctx.send("Unexpected error. Try again maybe?")
@commands.command()
async def swap(self, ctx, player1, player2):
'''Swaps two players on opposite teams.'''
self.lobby.swap(player1, player2)
await ctx.invoke(self.showteams)
@swap.error
async def swap_error(self, ctx, error):
if isinstance(error.original, AttributeError):
await ctx.send(Scrim.CREATE_LOBBY_MESSAGE)
if isinstance(error.original, AssertionError):
await ctx.send(error.original.args)
elif isinstance(error.original, ValueError):
await ctx.send(
"Player is not in the team(s)! Usernames are case senstive!")
else:
print(error)
await ctx.send("Unexpected error. Try again maybe?")
@commands.command()
async def shuffle(self, ctx):
'''Shuffles the lobby.'''
self.lobby.shuffle()
await ctx.invoke(self.showteams)
@commands.command()
async def tenmans(self, ctx):
'''Teaches you how to use the bot.'''
message = '''
Hey I heard you wanted to tenman? Here's how use the commands.
1. Create the lobby
- `?create 10`
2. Add players / join the lobby
- `?join` or `?add @username` or `?add random_name`
- you can view who's in the lobby with `?showlist`
- want to add a name with spaces in it? use quotation marks! `?add \"Name With Spaces\"`
3. Shuffle the players into teams, and I will provide the teams.
- `?shuffle`
'''
await ctx.send(message)
