Fix bug in broadcast

This commit is contained in:
Dustin Pianalto 2020-08-18 01:44:12 -08:00
parent 863ad75bfc
commit b9fc97d18f

View File

@ -560,10 +560,10 @@ class Rcon(commands.Cog):
# else: # else:
# await ctx.send(f'You are not authorized to run this command.') # await ctx.send(f'You are not authorized to run this command.')
async def _broadcast(self, *, ctx: discord.ext.commands.Context, message: str, server_name: str, async def _broadcast(self, *, message: str, server_name: str,
msg: discord.Message, message_lock: asyncio.Lock): msg: discord.Message, message_lock: asyncio.Lock):
suc = await self.bot.aio_session.get( suc = await self.bot.aio_session.post(
f'{self.bot.api_base}/rcon/{ctx.guild.id}/{server_name}/broadcast/', f'{self.bot.api_base}/rcon/{msg.guild.id}/{server_name}/broadcast/',
headers=self.bot.auth_header, headers=self.bot.auth_header,
json={'message': message} json={'message': message}
) )
@ -608,14 +608,14 @@ class Rcon(commands.Cog):
msg = await ctx.send(f'Broadcasting "{message}" to all servers.') msg = await ctx.send(f'Broadcasting "{message}" to all servers.')
lock = asyncio.Lock() lock = asyncio.Lock()
for server in guild_servers: for server in guild_servers:
futures.append(self._broadcast(ctx=ctx, message=message, server_name=server["name"], futures.append(self._broadcast(message=message, server_name=server["name"],
msg=msg, message_lock=lock)) msg=msg, message_lock=lock))
else: else:
for server in guild_servers: for server in guild_servers:
if server["name"].lower().replace(" ", "_") == server_name.lower(): if server["name"].lower().replace(" ", "_") == server_name.lower():
msg = await ctx.send(f'Broadcasting "{message}" to {server["name"]}.') msg = await ctx.send(f'Broadcasting "{message}" to {server["name"]}.')
lock = asyncio.Lock() lock = asyncio.Lock()
futures.append(self._broadcast(ctx=ctx, message=message, server_name=server["name"], futures.append(self._broadcast(message=message, server_name=server["name"],
msg=msg, message_lock=lock)) msg=msg, message_lock=lock))
break break
else: else: