Chatbot.py

#This code was written by Emily

# This section of code was adapted from https://realpython.com/how-to-make-a-discord-bot-python/#creating-a-discord-connection by Emily
import os
import random
import discord
from dotenv import load_dotenv #only used so we can have the token outside of the source code

load_dotenv()
TOKEN = os.getenv('DISCORD_TOKEN')
GUILD = os.getenv('DISCORD_GUILD')

client = discord.Client()

@client.event #Prints in the terminal if bot connects successfully
async def on_ready():
    print(f'{client.user} has connected to Discord!')
    for guild in client.guilds:
        if guild.name == GUILD:
            break
    print( f'{client.user} is connected to the following guild:\n'
           f'{guild.name}(id: {guild.id})')
    members = '\n - '.join([member.name for member in guild.members])
    print(f'Guild Members:\n - {members}')

polls = {} # masud added this to hold poll data. Can be switched to a database in the future

@client.event 
async def on_message(message): #checks for a message 
    if message.author == client.user: #makes sure it's a user 
        return

    greetings = ["hello", "howdy", "hey",] #Says hello (creepily) to the user if they use a greeting word
    greetCheck = [entry for entry in greetings if(entry in message.content.lower())] #This line is adapted from https://www.geeksforgeeks.org/python-test-if-string-contains-element-from-list/ by Emily
    if greetCheck:
        creepyHellos =["Why hello there, I am awakening from my slumber",
                       "My child I am learning, hi",
                       "I was enjoying the void before you woke me, I guess it's polite to say hello",
                       "Like life, my sleep was short. Hello child",
                       "Hello, what do you need me for this time?",
                       "Howdy y'all- yeah that wasn't working for me either. What can I do?",]
        response = random.choice(creepyHellos)
        await message.channel.send(response)

    # ================ Masud did this ================

    tickEmoji = "\u2705"  # unicode for the tick Emoji
    crossEmoji = "\u274C"  # unicode for the cross Emoji

    if "salt " == (message.content.lower())[:5]:  # checks if the input matches "salt "
        communityName = (message.content)[5:]  # saves the rest of the input as communityName

        if communityName.upper() in polls.values():  # if there is a poll active already
            pollID = list(polls.keys())[list(polls.values()).index(communityName.upper())]  # get the ID of the poll
            await message.channel.send(f'The community: {communityName} already has a poll active. Poll ID:{pollID}.')  # message to discord
        else:  # and if there is not an active poll already...
            poll = await message.channel.send(f'vote here for {communityName}:')  # create the vote poll
            print(f'[BOT DEBUG] {communityName} added to list of polls')  # print to console for debugging only
            polls[poll.id] = communityName.upper()  # save the pollID and communityName together
            print(polls)  # print list of polls active for debugging only
            await poll.add_reaction(tickEmoji)  # add the tick emoji for the poll
            await poll.add_reaction(crossEmoji)  # add the cross emoji for the poll

    if "count " == (message.content.lower())[:6]:  # cecks if the input matches "count "
        communityName = (message.content)[6:]  # saves the rest of the input as communityName

        if communityName.upper() not in polls.values():  # checks if the poll does not exist already
            await message.channel.send(f'No poll found for community: {communityName}.')  # messages to discord
        else:  # if the poll does exist ...
            
            pollID = list(polls.keys())[list(polls.values()).index(communityName.upper())]  # get the pollID with the matching communityName
            pollMessage = await message.channel.fetch_message(pollID)  # get the message object with the pollID
            pollReactions = pollMessage.reactions  # accesses the reactions for the pollMessage object

            await message.channel.send(f'Here are the saltiness statistics for the community {communityName}\nPoll ID:{pollID}')  # message to discord
            for emj in pollReactions:  # for each emoji in the reactions object
                emojiCount = emj.count - 1 # remove 1 vote, which was from the bot.
                if emojiCount == 1:  # message to discord accordingly
                    await message.channel.send(f'{emj.emoji} : {emojiCount} vote')
                else:
                    await message.channel.send(f'{emj.emoji} : {emojiCount} votes')

    # ================ up to about here ================


client.run(TOKEN) #this is the token of the discord bot. You can create your own bot and put that token in here or the .env file to run the code
#================================