Modron/bot.py at master · rp50/Modron · GitHub

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
from dotenv import load_dotenv
from discord.ext import commands
import discord, logging, os, datetime, motor.motor_asyncio

logging.basicConfig(level=logging.INFO)
load_dotenv()
bot_token = os.getenv('DISCORD_BOT_TOKEN')
mongo_URI = os.getenv('MONGODB_URI')
mongo_collection = os.getenv('MONGODB_COLLECTION')

mongoClient = motor.motor_asyncio.AsyncIOMotorClient(mongo_URI, serverSelectionTimeoutMS=5000, ssl=True, ssl_cert_reqs='CERT_NONE') # Client for the Motor MongoDB
questDB = mongoClient[mongo_collection]['Quests'] # Goes to the specific Database then Collection for this project.

botIntents = discord.Intents.default()
botIntents.members = True

bot = commands.Bot(command_prefix='.', intents=botIntents)

checkmark = '✅' # easier to type 'checkmark' than the emoji

# command to quickly check A) that the bot is online, and B) the latency of the bot currently
@bot.command()
async def ping(ctx):
    await ctx.send(f'Latency: {bot.latency}')

# Quest command, takes in the given command arguments and then reads them into a dictionary for mongoDB to consume as a document. Also creates an embed that is posted for the embed.
@bot.command()
async def quest(ctx, title, date, time):
    if ctx.author != bot.user:
        questData = {}
        questData['title'] = title
        dateTimeOf = datetime.datetime.strptime(date+time, '%m/%d%H:%M')
        dateTimeOf = dateTimeOf.replace(year=datetime.datetime.now().year)
        if dateTimeOf < datetime.datetime.now():
            dateTimeOf = dateTimeOf.replace(year=dateTimeOf.year+1)
        questData['datetime'] = dateTimeOf.strftime('%A, %m/%d/%y, %I:%M%p')
        questData['creator'] = ctx.author.id
        questData['members'] = []

        logging.info('Created new quest "{}", requested by {}'.format(title, ctx.author))

        oldCounter = await questDB.find_one({'_id': 'counter'}) #gets the counter document from the database
        newCounter = {'_id': 'counter', 'quests': oldCounter['quests']+1}
        questData['_id'] = 'quest%s' % str(newCounter['quests'])

        await questDB.replace_one({'_id': 'counter'}, newCounter) # replaces the counter document in the database with the new one (basically just a +1 to the quest counter)
        await questDB.insert_one(questData) # inserts the current quest into the quest collection

        #creating the embed for the quest message
        questResponse = discord.Embed(title= questData['title'], description= questData['_id'])
        questResponse.add_field(name= 'Creator', value=ctx.author.mention)
        questResponse.add_field(name= 'Scheduled For', value= questData['datetime'])
        questResponse.add_field(name= chr(173), value= chr(173))
        questResponse.add_field(name= 'Party Members', value= ctx.author.mention)
        questResponse.set_footer(text= 'React with a %s to join.' % checkmark)

        msg = await ctx.send(embed = questResponse) #sending the message
        await msg.add_reaction(checkmark) #adding a reaction to the message

@bot.event
async def on_ready():
    logging.info('Logged in as {}'.format(bot.user))

@bot.event
async def on_raw_reaction_add(payload): # payload contains all of the details about the event
    channel = await bot.fetch_channel(payload.channel_id)
    message = await channel.fetch_message(payload.message_id)
    user = await bot.fetch_user(payload.user_id)
    if message.author == bot.user and user != bot.user:
        if len(message.embeds) > 0:
            if message.embeds[0].description.startswith('quest') and str(payload.emoji) == checkmark: # checks that it is a quest embed, the emoji is a checkmark, and the creator of the quest (user not bot) is not the one who reacted
                questDoc = await questDB.find_one({'_id': message.embeds[0].description})
                if questDoc['creator'] != user.id:
                    party = message.embeds[0].fields[3]
                    newEmbed = message.embeds[0].set_field_at(3, name=party.name, value=party.value+' '+user.mention)
                    await message.edit(embed = newEmbed)
                    questDoc['members'].append(user.id)
                    await questDB.replace_one({'_id': message.embeds[0].description}, questDoc)

                    logging.info('{} joined "{}"'.format(user, message.embeds[0].title))

@bot.event
async def on_raw_reaction_remove(payload):
    channel = await bot.fetch_channel(payload.channel_id)
    message = await channel.fetch_message(payload.message_id)
    user = await bot.fetch_user(payload.user_id)
    if message.author == bot.user and user != bot.user:
        if len(message.embeds) > 0:
            if message.embeds[0].description.startswith('quest') and str(payload.emoji) == checkmark: # same logic as the on_raw_reaction_add() above
                questDoc = await questDB.find_one({'_id': message.embeds[0].description})
                if questDoc['creator'] != user.id:
                    party = message.embeds[0].fields[3]
                    if user.name == user.display_name:
                        newParty = party.value.replace('<@{}>'.format(user.id), '', 1)
                    else:
                        newParty = party.value.replace('<@!{}>'.format(user.id), '', 1)
                    newEmbed = message.embeds[0].set_field_at(3, name=party.name, value=newParty)
                    await message.edit(embed = newEmbed)
                    questDoc['members'].remove(user.id)
                    await questDB.replace_one({'_id': message.embeds[0].description}, questDoc)

                    logging.info('{} left "{}"'.format(user, message.embeds[0].title))


bot.run(bot_token)