Poll system

.gitignore

@@ -4,3 +4,6 @@
 
 *.pyc
 /logs/
+
+
+chart.png

EmojiCollection.py

@@ -0,0 +1,50 @@
+import emoji
+import selenium
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.support.ui import WebDriverWait
+
+
+class EmojiCollection():
+  def __init__(self):
+
+    options = Options()
+    # TODO this can't be OS specific
+    # options.binary_location = "/Applications/Google Chrome Beta.app/Contents/MacOS/Google Chrome Beta"
+    options.binary_location = "C:\Program Files\Google\Chrome\Application\chrome.exe"
+
+    self.driver = webdriver.Chrome(options=options)
+    self.driver.get("https://emojipedia.org")
+    # accepts cookies
+    acceptCookies = WebDriverWait(
+      self.driver,
+      30).until(EC.presence_of_element_located((By.ID, "onetrust-accept-btn-handler"))).click()
+
+  def searchEngineReturned(self):
+    #click onto the link of the first emoji
+    clickFirstEmoji = WebDriverWait(self.driver, 30).until(
+    EC.element_to_be_clickable((By.XPATH, "/html/body/div[@class='container']/div[@class='content']/ol/*[1]/h2/a"))).click()
+  def getEmoji(self):
+    try:
+      nameOfEmoji = WebDriverWait(self.driver, 10).until(
+      EC.visibility_of_element_located((By.CSS_SELECTOR, "span.shortcode"))).text
+      return emoji.emojize(nameOfEmoji)
+    except selenium.common.exceptions.TimeOutException:
+      tmp = emoji.emojize(":red_heart:")
+      return tmp
+  #enter the requested emoji to the search engine
+  def search(self, emoji):
+    #type a key word to the search engine
+    element = self.driver.find_element(By.ID, "id_q")
+    element.send_keys(emoji)
+    #the search engine starts searching
+    element.send_keys(Keys.RETURN)
+    self.searchEngineReturned()
+    return self.getEmoji()
+
+  def quit(self):
+    self.driver.quit()
+

Server.py

@@ -0,0 +1,13 @@
+import http.server
+import socketserver
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+def runServer():
+  Handler = http.server.SimpleHTTPRequestHandler
+  port = os.environ.get('PORT')
+  with socketserver.TCPServer(("", int(port)), Handler) as httpd:
+    print(f'serving at port {port}')
+    httpd.serve_forever()
+

chart.py

@@ -0,0 +1,24 @@
+import matplotlib.pyplot as plt
+import numpy as np
+
+# Fixing random state for reproducibility
+np.random.seed(19680801)
+
+
+plt.rcdefaults()
+fig, ax = plt.subplots()
+# Example data
+people = ['Tom', 'Dick', 'Harry', 'Slim', 'Jim']
+y_pos = np.arange(len(people))
+performance = 3 + 10 * np.random.rand(len(people))
+error = np.random.rand(len(people))
+
+ax.barh(y_pos, performance, xerr=error, align='center')
+ax.set_yticks(y_pos)
+ax.set_yticklabels(people)
+ax.invert_yaxis()  # labels read top-to-bottom
+ax.set_xlabel('Performance')
+ax.set_title('How fast do you want to go today?')
+
+
+plt.savefig('chart.png')

cogs/poll.py

@@ -0,0 +1,81 @@
+from ast import alias
+import discord
+import asyncio
+import matplotlib.pyplot as plt
+import numpy as np
+from discord.ext import commands
+import emoji
+
+
+class Poll(commands.Cog):
+  def __init__(self, bot):
+    self.bot = bot
+
+  async def create_bar_chart(self, options, reactions, question):
+    totalVotes = 0
+    votes = []
+
+    #calculate the number of votes for each option
+    for reaction in reactions:
+      # Start at -1 to exclude the vote of the bot
+      v = -1
+
+      async for _ in reaction.users():
+        v += 1
+
+      totalVotes += v
+      votes.append(v)
+
+    # Calculates percentage of votes for each option
+    percentages = [round((vote/totalVotes)* 100, 2) for vote in votes]
+
+    # Initialises the figure
+    fig, ax = plt.subplots()
+    y_pos = np.arange(len(options))
+    ax.barh(y_pos, percentages, align='center')
+    ax.set_yticks(y_pos, labels=options)
+    ax.invert_yaxis()  # labels read top-to-bottom
+    ax.set_xlabel('Votes (%)')
+    ax.set_title(question)
+    plt.savefig('chart.png')
+
+
+  @commands.command(aliases=["p"])
+  async def init_poll(self, ctx, question, time, *options):
+    # Prevents user from inputting more than 3 options
+    if len(options) > 3:
+      await ctx.send("The number of options cannot exceed the allowed limit")
+      return
+    if not time.isdigit():
+      await ctx.send("Time has to be a digit")
+      return
+
+    embed = discord.Embed(title = question,
+                          description= f'Poll will end in {time} seconds :alarm_clock:. There are {len(options)} options:')
+    tmp = [':one:', ':two:', ':three:'][:len(options)]
+    emojis = [emoji.emojize(e, language='alias') for e in tmp]
+    for i in range(len(options)):
+        emo = emojis[i]
+        embed.add_field(name = emo, value = options[i], inline = True)
+    embed.add_field(name="Instructions", value="React to cast a vote", inline=False)
+    embed.set_footer(text = f'This poll is created by {ctx.author.name}')
+    poll = await ctx.send(embed = embed)
+
+    # add emojis as reactions to the poll
+    for emo in emojis:
+      await poll.add_reaction(emo)
+    await asyncio.sleep(int(time))
+
+    #calculate the result of the poll
+    message = await ctx.fetch_message(poll.id)
+
+    # create a bar chart as a result
+    await self.create_bar_chart(options, message.reactions, question)
+    result = discord.Embed(title='Result', description = "Here are the vote results")
+    file = discord.File("chart.png")
+    result.set_image(url="attachment://chart.png")
+    await ctx.send(file=file, embed = result)
+
+
+async def setup(bot):
+    await bot.add_cog(Poll(bot))