73 lines
3.1 KiB
Python
73 lines
3.1 KiB
Python
from discord.ext import commands
|
|
|
|
from .utils import config
|
|
from .utils import checks
|
|
|
|
import aiohttp
|
|
import re
|
|
import json
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Strawpoll(bot))
|
|
|
|
getter = re.compile(r'`(?!`)(.*?)`')
|
|
multi = re.compile(r'```(.*?)```', re.DOTALL)
|
|
|
|
|
|
class Strawpoll:
|
|
"""This class is used to create new strawpoll """
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.url = 'https://strawpoll.me/api/v2/polls'
|
|
self.headers = {'User-Agent': 'Bonfire/1.0.0',
|
|
'Content-Type': 'application/json'}
|
|
self.session = aiohttp.ClientSession(headers=headers)
|
|
|
|
def __unload(self):
|
|
self.bot.loop.create_task(self.session.close())
|
|
|
|
|
|
@commands.group(aliases=['strawpoll','poll','polls'], pass_context=True, invoke_without_command=True)
|
|
@checks.customPermsOrRole(send_messages=True)
|
|
async def strawpolls(self, ctx, poll_id: int=None):
|
|
"""This command can be used to show a strawpoll setup on this server"""
|
|
all_polls = config.getContent('strawpolls') or {}
|
|
server_polls = all_polls.get(ctx.message.server.id)
|
|
if not server_polls:
|
|
await self.bot.say("There are currently no strawpolls running on this server!")
|
|
return
|
|
if not poll_id:
|
|
fmt = "\n".join("{}: https://strawpoll.me/{}".format(title, id) for id, title in server_polls.items())
|
|
await self.bot.say("```\n{}```".format(fmt))
|
|
elif poll_id in server_polls.keys():
|
|
async with self.session.get("{}/{}".format(self.url, str(poll_id)), headers=self.headers) as response:
|
|
data = await response.json()
|
|
fmt_options = "\n\t".join("{}: {}".format(data['options'][i], data['votes'][i]) for i in range(data['options']))
|
|
fmt = "Link: {}\nTitle: {}\nOptions:\n\t{}".format("https://strawpoll.me{}".format(str(poll_id)), data['title'], fmt_options)
|
|
await self.bot.say("```\n{}```".format(fmt))
|
|
|
|
|
|
@strawpolls.command(name='create', aliases=['setup', 'add'], pass_context=True)
|
|
@checks.customPermsOrRole(kick_members=True)
|
|
async def create_strawpoll(self, ctx, title, *, options):
|
|
"""This command is used to setup a new strawpoll
|
|
The format needs to be: poll create "title here" all options here
|
|
Options need to be separated by using either one ` around each option
|
|
Or use a code block (3 ` around the options), each option on it's own line"""
|
|
match_single = getter.findall(options)
|
|
match_multi = multi.findall(options)
|
|
if match_single:
|
|
options = match_single
|
|
elif match_multi:
|
|
options = match_multi[0].splitlines()
|
|
else:
|
|
await self.bot.say("Please provide options for a new strawpoll! Use {}help if you do not know the format".format(ctx.prefix))
|
|
return
|
|
payload = {'title': title,
|
|
'options': options}
|
|
async with self.session.post(self.url, data=json.dumps(payload), headers=self.headers) as response:
|
|
data = await response.json()
|
|
await self.bot.say("Link for your new strawpoll: https://strawpoll.me/{}".format(data['id']))
|
|
|