overload security

This commit is contained in:
klemek
2021-01-07 10:50:25 +01:00
parent 3a96a9041a
commit f779c699cf
2 changed files with 30 additions and 19 deletions
+5
View File
@@ -69,6 +69,11 @@ async def compute(client: discord.client, message: discord.Message, *args: str):
async with message.channel.typing():
progress = await message.channel.send("```Starting analysis...```")
total_msg, total_chan = await logs.load(progress, channels)
if total_msg == -1:
await message.channel.send(
"f{message.author} An analysis is already running on this server, please be patient."
)
else:
msg_count = 0
chan_count = 0
for id in logs.channels:
+7 -1
View File
@@ -15,6 +15,8 @@ if not os.path.exists(LOG_DIR):
CHUNK_SIZE = 1000
FORMAT = 2
current_analysis = []
class FakeMessage:
def __init__(self, id: int):
@@ -151,7 +153,10 @@ class GuildLogs:
async def load(
self, progress: discord.Message, target_channels: List[discord.TextChannel] = []
):
) -> Tuple[int, int]:
if self.log_file in current_analysis:
return -1, -1
current_analysis += [self.log_file]
await progress.edit(
content=f"```Reading history...\n(this might take a while)```"
)
@@ -220,4 +225,5 @@ class GuildLogs:
await progress.edit(
content=f"```Analysing...\n{total_msg} messages in {total_chan} channels```"
)
current_analysis.remove(self.log_file)
return total_msg, total_chan