mirror of
https://github.com/myned/modufur.git
synced 2024-12-24 22:27:28 +00:00
Added auto-heart command
This commit is contained in:
parent
b43b863c15
commit
3aac112ecb
1 changed files with 67 additions and 4 deletions
|
@ -55,6 +55,11 @@ class MsG:
|
||||||
print('STARTED : auto-reversifying in #{}'.format(temp.name))
|
print('STARTED : auto-reversifying in #{}'.format(temp.name))
|
||||||
self.reversifying = True
|
self.reversifying = True
|
||||||
self.bot.loop.create_task(self._reversify())
|
self.bot.loop.create_task(self._reversify())
|
||||||
|
if u.tasks['auto_hrt']:
|
||||||
|
for channel in u.tasks['auto_hrt']:
|
||||||
|
temp = self.bot.get_channel(channel)
|
||||||
|
self.bot.loop.create_task(self.queue_for_hearts(channel=temp))
|
||||||
|
print(f'STARTED : auto-hearting in #{temp.name}')
|
||||||
# if not self.updating:
|
# if not self.updating:
|
||||||
# self.updating = True
|
# self.updating = True
|
||||||
# self.bot.loop.create_task(self._update_suggested())
|
# self.bot.loop.create_task(self._update_suggested())
|
||||||
|
@ -124,34 +129,92 @@ class MsG:
|
||||||
while self.hearting:
|
while self.hearting:
|
||||||
temp = await self.heartqueue.get()
|
temp = await self.heartqueue.get()
|
||||||
|
|
||||||
|
if isinstance(temp[1], d.Embed):
|
||||||
await temp[0].send(embed=temp[1])
|
await temp[0].send(embed=temp[1])
|
||||||
|
|
||||||
await asyncio.sleep(self.RATE_LIMIT)
|
await asyncio.sleep(self.RATE_LIMIT)
|
||||||
|
elif isinstance(temp[1], d.Message):
|
||||||
|
for match in re.finditer('(https?:\/\/[^ ]*\.(?:gif|png|jpg|jpeg))', temp[1].content):
|
||||||
|
await temp[0].send(match)
|
||||||
|
|
||||||
|
await asyncio.sleep(self.RATE_LIMIT)
|
||||||
|
|
||||||
|
for attachment in temp[1].attachments:
|
||||||
|
await temp[0].send(attachment.url)
|
||||||
|
await asyncio.sleep(self.RATE_LIMIT)
|
||||||
|
|
||||||
print('STOPPED : hearting')
|
print('STOPPED : hearting')
|
||||||
|
|
||||||
async def queue_for_hearts(self, *, message, send):
|
async def queue_for_hearts(self, *, message=None, send=None, channel=None, reaction=True, timeout=60 * 60):
|
||||||
def on_reaction(reaction, user):
|
def on_reaction(reaction, user):
|
||||||
if reaction.emoji == '\N{HEAVY BLACK HEART}' and reaction.message.id == message.id:
|
if reaction.emoji == '\N{HEAVY BLACK HEART}' and reaction.message.id == message.id:
|
||||||
raise exc.Save(user)
|
raise exc.Save(user)
|
||||||
return False
|
return False
|
||||||
|
def on_message(msg):
|
||||||
if 'stop h' in msg.content.lower():
|
if 'stop h' in msg.content.lower():
|
||||||
|
raise exc.Abort
|
||||||
|
return msg.channel.id == channel.id and (re.search('(https?:\/\/[^ ]*\.(?:gif|png|jpg|jpeg))', msg.content) or msg.attachments)
|
||||||
|
|
||||||
|
if message:
|
||||||
try:
|
try:
|
||||||
|
if reaction:
|
||||||
await message.add_reaction('\N{HEAVY BLACK HEART}')
|
await message.add_reaction('\N{HEAVY BLACK HEART}')
|
||||||
await asyncio.sleep(1)
|
await asyncio.sleep(1)
|
||||||
|
|
||||||
while self.hearting:
|
while self.hearting:
|
||||||
try:
|
try:
|
||||||
await asyncio.gather(*[self.bot.wait_for('reaction_add', check=on_reaction, timeout=60 * 60),
|
await self.bot.wait_for('reaction_add', check=on_reaction, timeout=timeout)
|
||||||
self.bot.wait_for('reaction_remove', check=on_reaction, timeout=60 * 60)])
|
|
||||||
|
|
||||||
except exc.Save as e:
|
except exc.Save as e:
|
||||||
await self.heartqueue.put((e.user, send))
|
await self.heartqueue.put((e.user, send if send else message))
|
||||||
|
|
||||||
except asyncio.TimeoutError:
|
except asyncio.TimeoutError:
|
||||||
await message.add_reaction('\N{WHITE HEAVY CHECK MARK}')
|
await message.add_reaction('\N{WHITE HEAVY CHECK MARK}')
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
async for message in channel.history(limit=300):
|
||||||
|
if re.search('(https?:\/\/[^ ]*\.(?:gif|png|jpg|jpeg))', message.content) or message.attachments:
|
||||||
|
self.bot.loop.create_task(self._wait_for_reaction(message))
|
||||||
|
|
||||||
|
while self.hearting:
|
||||||
|
message = await self.bot.wait_for('message', check=on_message)
|
||||||
|
self.bot.loop.create_task(self._wait_for_reaction(message))
|
||||||
|
|
||||||
|
except exc.Abort:
|
||||||
|
u.tasks['auto_hrt'].remove(channel.id)
|
||||||
|
u.dump(u.tasks, 'cogs/tasks.pkl')
|
||||||
|
print('STOPPED : auto-hearting in #{}'.format(channel.name))
|
||||||
|
await channel.send('**Stopped queueing messages for hearting in** {}'.format(channel.mention), delete_after=5)
|
||||||
|
|
||||||
|
async def _wait_for_reaction(self, message):
|
||||||
|
def on_reaction(reaction, user):
|
||||||
|
if reaction.emoji == '\N{HEAVY BLACK HEART}' and reaction.message.id == message.id:
|
||||||
|
raise exc.Save(user)
|
||||||
|
return False
|
||||||
|
|
||||||
|
while self.hearting:
|
||||||
|
try:
|
||||||
|
await self.bot.wait_for('reaction_add', check=on_reaction)
|
||||||
|
|
||||||
|
except exc.Save as e:
|
||||||
|
await self.heartqueue.put((e.user, message))
|
||||||
|
|
||||||
|
@cmds.command(name='autoheart', aliases=['autohrt'])
|
||||||
|
@cmds.has_permissions(administrator=True)
|
||||||
|
async def auto_heart(self, ctx):
|
||||||
|
try:
|
||||||
|
if ctx.channel.id not in u.tasks['auto_hrt']:
|
||||||
|
u.tasks['auto_hrt'].append(ctx.channel.id)
|
||||||
|
u.dump(u.tasks, 'cogs/tasks.pkl')
|
||||||
|
self.bot.loop.create_task(self.queue_for_hearts(channel=ctx.channel))
|
||||||
|
print('STARTED : auto-hearting in #{}'.format(ctx.channel.name))
|
||||||
|
await ctx.send('**Auto-hearting all messages in {}**'.format(ctx.channel.mention), delete_after=5)
|
||||||
|
else:
|
||||||
|
raise exc.Exists
|
||||||
|
|
||||||
|
except exc.Exists:
|
||||||
await ctx.send('**Already auto-hearting in {}.** Type `stop h(earting)` to stop.'.format(ctx.channel.mention), delete_after=7)
|
await ctx.send('**Already auto-hearting in {}.** Type `stop h(earting)` to stop.'.format(ctx.channel.mention), delete_after=7)
|
||||||
|
await ctx.message.add_reaction('\N{CROSS MARK}')
|
||||||
|
|
||||||
# @cmds.command()
|
# @cmds.command()
|
||||||
# async def auto_post(self, ctx):
|
# async def auto_post(self, ctx):
|
||||||
|
|
Loading…
Reference in a new issue