CodeMirror/Bonfire
1
0
Fork 0
mirror of synced 2024-06-29 03:30:57 +12:00
Code Issues Releases Wiki Activity

Corrected issue where post headers were being sent on get requests

Browse source
This commit is contained in:
phxntxm 2016-08-14 00:10:51 -05:00
parent d29bad3ea0
commit 51438fdde5
1 changed files with 25 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
cogs/strawpoll.py
View file

@ -8,16 +8,18 @@ import re
import json
import pendulum
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'
@ -25,10 +27,9 @@ class Strawpoll:
'Content-Type': 'application/json'}
self.session = aiohttp.ClientSession()
@commands.group(aliases=['strawpoll','poll','polls'], pass_context=True, invoke_without_command=True)
@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):
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) or {}
@ -36,22 +37,25 @@ class Strawpoll:
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(data['title'], id) for id, data in server_polls.items())
fmt = "\n".join(
"{}: https://strawpoll.me/{}".format(data['title'], id) for id, data in server_polls.items())
await self.bot.say("```\n{}```".format(fmt))
elif str(poll_id) in server_polls.keys():
poll = server_polls[str(poll_id)]
async with self.session.get("{}/{}".format(self.url, str(poll_id)), headers=self.headers) as response:
async with self.session.get("{}/{}".format(self.url, str(poll_id))) 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_options = "\n\t".join(
"{}: {}".format(data['options'][i], data['votes'][i]) for i in range(data['options']))
author = self.bot.get_member(poll['author'])
created_ago = (pendulum.parse(poll['date'])-pendulum.utcnow()).in_words()
created_ago = (pendulum.parse(poll['date']) - pendulum.utcnow()).in_words()
link = "https://strawpoll.me{}".format(str(poll_id))
fmt = "Link: {}\nTitle: {}\nAuthor: {}\nCreated: {}\nOptions:\n\t{}".format(link, data['title'], author.display_name, created_ago, fmt_options)
fmt = "Link: {}\nTitle: {}\nAuthor: {}\nCreated: {}\nOptions:\n\t{}".format(link, data['title'],
author.display_name,
created_ago, 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):
@ -68,18 +72,19 @@ class Strawpoll:
options = match_multi[0].splitlines()
options = [option for option in options if option]
else:
await self.bot.say("Please provide options for a new strawpoll! Use {}help if you do not know the format".format(ctx.prefix))
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}
'options': options}
async with self.session.post(self.url, data=json.dumps(payload), headers=self.headers) as response:
data = await response.json()
all_polls = config.getContent('strawpolls') or {}
server_polls = all_polls.get(ctx.message.server.id) or {}
server_polls[data['id']] = {'author': ctx.message.author.id,'date': str(pendulum.utcnow()), 'title': title}
server_polls[data['id']] = {'author': ctx.message.author.id, 'date': str(pendulum.utcnow()), 'title': title}
all_polls[ctx.message.server.id] = server_polls
config.saveContent('strawpolls',all_polls)
await self.bot.say("Link for your new strawpoll: https://strawpoll.me/{}".format(data['id']))
config.saveContent('strawpolls', all_polls)
await self.bot.say("Link for your new strawpoll: https://strawpoll.me/{}".format(data['id']))