1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
from discord.ext import commands, tasks
import discord
import datetime
import dateutil.parser
import importlib
import utils
importlib.reload(utils)
class Remind(commands.Cog):
def __init__(self, bot):
self.bot = bot
global extension_name
extension_name = "[Remind] "
self.reminder_loop.start()
@commands.group(invoke_without_command=True)
async def remind(self, ctx, *args):
"""Set reminders for things through Discord."""
if len(args) == 0:
prefix = await self.bot.get_prefix(ctx.message)
emb = await utils.embed(ctx, f"Commands for {prefix[2]}remind", "With remind you can set events at specific dates and times.\nDates are currently in a UK date format, along with UTC +0 times.",)
emb = await utils.field(emb, f"✴{prefix[2]}remind setutc [number]", "✴️**Not fully added in.** Set the UTC timezone your server should follow. (0 by default)")
emb = await utils.field(emb, f"{prefix[2]}remind list", "Get a list of your Discord servers reminders")
emb = await utils.field(emb, f"{prefix[2]}remind [date - dd/mm/yyyy] [time - hh:mm] [your reminder]", "Add a new reminder at a specific date and time.")
await ctx.send(embed=emb)
# 1️⃣2️⃣3️⃣4️⃣5️⃣6️⃣7️⃣8️⃣9️⃣
elif len(args) >= 3:
guild_utc = await utils.sql_get('SELECT utc_timezone FROM "database1".synthy.settings WHERE guild_id = %s', (ctx.guild.id,))
args = list(args)
arg_date = args.pop(0)
arg_time = args.pop(0)
arg_msg = " ".join(args)
big_date_target = dateutil.parser.parse(arg_date + " " + arg_time, dayfirst=True)
big_date_current = dateutil.parser.parse(str(datetime.datetime.now()))
# print(f"big_date_target: {big_date_target}")
# print(f"big_date_current: {big_date_current}")
# Check date and time is valid
if big_date_current >= big_date_target:
await ctx.send(content="You cannot set a reminder in the past.")
else:
await utils.sql('INSERT INTO "database1".synthy.remind (message_id, member_id, channel_id, guild_id, datetime, reminder) VALUES (%s, %s, %s, %s, %s, %s)',
(ctx.message.id, ctx.author.id, ctx.channel.id, ctx.guild.id, big_date_target, arg_msg,))
await ctx.send(content=f"Added a reminder on {big_date_target} for: {arg_msg}")
@remind.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[]))
async def setutc(self, ctx, arg):
utc_tz = await utils.sql_get('SELECT utc_timezone FROM "database1".synthy.settings WHERE guild_id = %s', (ctx.guild.id,))
if utc_tz is None:
await utils.sql_set('INSERT INTO "database1".synthy.settings (guild_id, utc_timezone) VALUES (%s, %s)', (ctx.guild.id, arg,))
cur_time = datetime.datetime.now()
cur_time = cur_time + datetime.timedelta(hours=int(arg))
cur_time = str(cur_time).split(".")[0]
await ctx.send(
content=f"Created UTC difference for {ctx.guild}. The time remind uses will be: {cur_time}")
else:
await utils.sql_set('UPDATE "database1".synthy.settings SET utc_timezone = %s WHERE guild_id = %s', (arg, ctx.guild.id,))
# db.commit()
cur_time = datetime.datetime.now()
cur_time = cur_time + datetime.timedelta(hours=int(arg))
cur_time = str(cur_time).split(".")[0]
await ctx.send(content=f"Updated UTC for {ctx.guild}. The time remind uses will be: {cur_time}")
@remind.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[]))
async def list(self, ctx):
reminders = await utils.sql('SELECT reminder, member_id, date, time FROM "database1".synthy.remind WHERE guild_id = %s', (ctx.guild.id,))
if not reminders == ():
desc = ""
for item in reminders:
item_date = datetime.datetime.strftime(item[2], "%d/%m/%Y")
desc = desc + f"**Author:** <@{item[1]}> ({item_date} {item[3]})\n**Reminder:** {item[0]}\n\n"
emb = await utils.embed(ctx, '', desc)
await ctx.send(embed=emb)
else:
emb = await utils.embed(ctx, 'You have no reminders.', '')
await ctx.send(embed=emb)
@tasks.loop(seconds=10)
async def reminder_loop(self):
# try:
# await utils.log(self.bot, "Remind", "[Remind] Starting reminder_loop")
cur_datetime = dateutil.parser.parse(str(datetime.datetime.now()))
# await utils.log(self.bot, "Remind", f"[Remind] Current time: {cur_datetime}")
sql_return = await utils.sql('SELECT id, channel_id, reminder FROM "database1".synthy.remind WHERE %s >= datetime', (cur_datetime,))
# await utils.log(self.bot, "Remind", f"[Remind] SQL Reminders: {sql_return}")
for item in sql_return:
rmdr_channel = self.bot.get_channel(int(item["channel_id"]))
# await utils.log(self.bot.user.name, "Remind", f"[Remind] Channel to post in: {rmdr_channel.name}/{rmdr_channel.id}")
await rmdr_channel.send(content=item["reminder"])
# await utils.log(self.bot, "Remind", f'[Remind] Sent {item["reminder"]} to {rmdr_channel.name}')
await utils.sql('DELETE FROM "database1".synthy.remind WHERE id = %s', (item["id"],))
# await utils.log(self.bot, "Remind", f'[Remind] Deleted {item["id"]} from SQL/Remind')
# print("2", end="\r")
# await utils.log(self.bot.user.name, "Remind", "[Remind] End of loop.")
def setup(bot):
print("INFO: Loading [Remind]... ", end="")
bot.add_cog(Remind(bot))
print("Done!")
def teardown(bot):
bot.reminder_loop.cancel()
print("INFO: Unloading [Remind]")
|
