myned/modufur
1
0
Fork 0
mirror of https://github.com/myned/modufur.git synced 2024-11-01 21:02:38 +00:00
Code Issues Activity

Merge branch 'master' into dev

Browse source
This commit is contained in:
Myned 2017-10-15 02:18:10 -04:00
commit be1d861d1a
3 changed files with 19 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/main/cogs/management.py
View file

@ -16,12 +16,15 @@ class Administration:
self.bot = bot
self.RATE_LIMIT = 2.1
self.queue = asyncio.Queue()
self.deleting = False
for channel in u.tasks.get('management', {}).get('auto_delete', []):
if u.tasks['auto_del']:
for channel in u.tasks['auto_del']:
temp = self.bot.get_channel(channel)
self.bot.loop.create_task(self.queue_on_message(temp))
self.bot.loop.create_task(self.delete())
print('Looping #{}'.format(temp.name))
self.bot.loop.create_task(self.delete())
self.deleting = True
# @commands.group(aliases=['pr', 'clear', 'cl'])
# @commands.is_owner()
@ -136,7 +139,7 @@ class Administration:
await ctx.send('❌ **Deletion timed out.**', delete_after=10)
async def delete(self):
while not self.bot.is_closed():
while self.deleting:
message = await self.queue.get()
await asyncio.sleep(self.RATE_LIMIT)
try:
@ -164,8 +167,10 @@ class Administration:
await self.queue.put(message)
except exc.Abort:
u.tasks['management']['auto_delete'].remove(channel.id)
u.tasks['auto_del'].remove(channel.id)
u.dump(u.tasks, 'cogs/tasks.pkl')
if not u.tasks['auto_del']:
self.deleting = False
print('Stopped looping {}'.format(channel.id))
await channel.send('✅ **Stopped deleting messages in** {}**.**'.format(channel.mention), delete_after=5)
@ -179,11 +184,13 @@ class Administration:
channel = ctx.channel
try:
if channel.id not in u.tasks.setdefault('management', {}).setdefault('auto_delete', []):
u.tasks['management']['auto_delete'].append(channel.id)
if channel.id not in u.tasks['auto_del']:
u.tasks['auto_del'].append(channel.id)
u.dump(u.tasks, 'cogs/tasks.pkl')
self.bot.loop.create_task(self.queue_on_message(channel))
if not self.deleting:
self.bot.loop.create_task(self.delete())
self.deleting = True
print('Looping #{}'.format(channel.name))
await ctx.send('✅ **Auto-deleting all messages in this channel.**', delete_after=5)
else:

2
src/main/utils/utils.py
View file

@ -51,7 +51,7 @@ def dump(obj, filename):
settings = setdefault('settings.pkl', {'del_ctx': []})
tasks = setdefault('cogs/tasks.pkl', {})
tasks = setdefault('cogs/tasks.pkl', {'auto_del': []})
async def clear(obj, interval=10 * 60, replace=None):