john/kwaylon
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ddos protection

Browse Source
This commit is contained in:
jsl12
2021-08-02 15:10:59 -05:00
parent 117e9930e4
commit a0aa557425
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
robopage.py
View File

@@ -1,6 +1,7 @@
import logging import logging
import os import os
import re import re
from threading import Lock
import discord import discord
import nltk import nltk
@@ -15,6 +16,8 @@ LIL_STINKY_ID = 704043422276780072
class RoboPage(discord.Client): class RoboPage(discord.Client):
db_path: str = 'messages.db'
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super(RoboPage, self).__init__(*args, **kwargs) super(RoboPage, self).__init__(*args, **kwargs)
self.jokes = [ self.jokes = [
@@ -23,7 +26,7 @@ class RoboPage(discord.Client):
AssJoke(), AssJoke(),
DominosJoke() DominosJoke()
] ]
self.db_path = 'messages.db' self.lock = Lock()
def run(self): def run(self):
return super().run(os.getenv('DISCORD_TOKEN')) return super().run(os.getenv('DISCORD_TOKEN'))
@@ -52,8 +55,12 @@ class RoboPage(discord.Client):
await message.reply(await self.get_cancelled_totals(limit=1000, days=14)) await message.reply(await self.get_cancelled_totals(limit=1000, days=14))
elif (m := re.search('top cancelled (?P<name>\w+)', message.content)) is not None: elif (m := re.search('top cancelled (?P<name>\w+)', message.content)) is not None:
if self.lock.acquire(blocking=False):
msg: discord.Message = await message.reply('Hold please...') msg: discord.Message = await message.reply('Hold please...')
await message.reply(await self.top_cancellations(user=m.group('name'), limit=1000, days=14)) await message.reply(await self.top_cancellations(user=m.group('name'), limit=1000, days=14))
self.lock.release()
else:
await message.reply("I'm busy!")
for joke in self.jokes: for joke in self.jokes:
if (scan_res := joke.scan(message)): if (scan_res := joke.scan(message)):