def queue_reminders(self):
try:
next_reminder = Reminder.select().order_by(
Reminder.remind_at.asc()).limit(1).get()
except Reminder.DoesNotExist:
return
self.reminder_task.set_next_schedule(next_reminder.remind_at)
def cmd_remind_list(self, event, limit=None, mode='server'):
user = event.msg.author
count = Reminder.count_for_user(user.id, event.guild.id)
total_count = Reminder.count_for_user(user.id)
embed = MessageEmbed()
embed.title = '{} reminder{} ({} total)'.format(
count if mode == 'server' else total_count, 's' if
(count != 1 and mode == 'server') or
(total_count != 1 and mode == 'global') else '', total_count)
embed.set_author(name=u'{}#{}'.format(
user.username,
user.discriminator,
),
icon_url=user.avatar_url)
embed.color = get_dominant_colors_user(user,
user.get_avatar_url('png'))
embed.set_footer(text='You can cancel reminders with !r clear [ID]')
if (count == 0 and mode == 'server') or total_count == 0:
embed.description = 'You have no upcoming reminders{}.'.format(
' in this server. Use `!r list global` to list all your upcoming reminders'
if total_count > 0 else '')
else:
query = Reminder.select(Reminder).where(
(Reminder.message_id << Reminder.with_message_join(
(Message.id, )).where(
(Message.author_id == event.author.id) &
(Message.guild_id == event.guild.id if mode ==
'server' else True)))
& (Reminder.remind_at >
(datetime.utcnow() + timedelta(seconds=1)))).order_by(
Reminder.remind_at).limit(limit)
for reminder in query:
time = humanize_duration(reminder.remind_at -
datetime.utcnow())
channel = Message.select().where(
Message.id == reminder.message_id).get().channel_id
channel = self.state.channels.get(channel)
embed.add_field(
name=u'#{} in {}'.format(reminder.id, time),
value=u'[`#{}`](https://discordapp.com/channels/{}/{}/{}) {}'
.format(
channel.name if channel.type != ChannelType.DM else
'Jetski', channel.guild_id
if channel.type != ChannelType.DM else '@me',
channel.id, reminder.message_id, S(reminder.content)))
return event.msg.reply(embed=embed)
def cmd_remind_list(self, event, limit=None, mode=None):
user = event.msg.author
count = Reminder.count_for_user(user.id)
avatar = u'https://cdn.discordapp.com/avatars/{}/{}.png'.format(
user.id,
user.avatar,
)
embed = MessageEmbed()
embed.title = '{} reminder{}'.format(count, '' if count == 1 else 's')
embed.set_author(name=u'{}#{}'.format(
user.username,
user.discriminator,
), icon_url=avatar)
embed.color = get_dominant_colors_user(user, avatar)
embed.set_footer(text='You can cancel reminders with !r clear [ID]')
if count == 0:
embed.description = 'You have no upcoming reminders.'
else:
query = Reminder.select(Reminder).where(
(Reminder.message_id << Reminder.with_message_join((Message.id, )).where(
Message.author_id == event.author.id
)) & (Reminder.remind_at > (datetime.utcnow() + timedelta(seconds=1)))
).order_by(Reminder.remind_at).limit(limit)
for reminder in query:
time = humanize_duration(reminder.remind_at - datetime.utcnow())
channel = Message.select().where(Message.id == reminder.message_id).get().channel_id
channel = self.state.channels.get(channel)
embed.add_field(
name=u'#{} in {}'.format(
reminder.id,
time
),
value=u'[`#{}`](https://discordapp.com/channels/{}/{}/{}) {}'.format(
channel.name if channel.type != ChannelType.DM else 'Jetski',
channel.guild_id if channel.type != ChannelType.DM else '@me',
channel.id,
reminder.message_id,
S(reminder.content)
)
)
return event.msg.reply(embed=embed)
def cmd_remind_clear(self, event, reminder='all'):
if reminder == 'all':
count = Reminder.count_for_user(event.author.id)
if Reminder.count_for_user(event.author.id) == 0:
return event.msg.reply('<:{}> cannot clear reminders when you don\'t have any'.format(RED_TICK_EMOJI))
msg = event.msg.reply('Ok, clear {} reminders?'.format(count))
msg.chain(False).\
add_reaction(GREEN_TICK_EMOJI).\
add_reaction(RED_TICK_EMOJI)
try:
mra_event = self.wait_for_event(
'MessageReactionAdd',
message_id=msg.id,
conditional=lambda e: (
e.emoji.id in (GREEN_TICK_EMOJI_ID, RED_TICK_EMOJI_ID) and
e.user_id == event.author.id
)).get(timeout=10)
except gevent.Timeout:
return
finally:
msg.delete()
if mra_event.emoji.id != GREEN_TICK_EMOJI_ID:
return
count = Reminder.delete_all_for_user(event.author.id)
return event.msg.reply(':ok_hand: I cleared {} reminders for you'.format(count))
else:
try:
# stupid catch because python sucks
try:
reminder = int(reminder)
except:
return event.msg.reply('cannot convert `{}` to `int`'.format(S(reminder)))
r = Reminder.select(Reminder).where(
(Reminder.message_id << Reminder.with_message_join((Message.id, )).where(
Message.author_id == event.author.id
)) & (Reminder.id == reminder)
).get()
except Reminder.DoesNotExist:
return event.msg.reply('<:{}> cannot find reminder #{}'.format(RED_TICK_EMOJI, reminder))
msg = event.msg.reply('Ok, clear reminder #{}?'.format(reminder))
msg.chain(False).\
add_reaction(GREEN_TICK_EMOJI).\
add_reaction(RED_TICK_EMOJI)
try:
mra_event = self.wait_for_event(
'MessageReactionAdd',
message_id=msg.id,
conditional=lambda e: (
e.emoji.id in (GREEN_TICK_EMOJI_ID, RED_TICK_EMOJI_ID) and
e.user_id == event.author.id
)).get(timeout=10)
except gevent.Timeout:
return
finally:
msg.delete()
if mra_event.emoji.id != GREEN_TICK_EMOJI_ID:
return
Reminder.delete_for_user(event.author.id, r.id)
return event.msg.reply(':ok_hand: I cleared reminder #{} for you'.format(r.id))
