updated meme to allow blank topics
This commit is contained in:
phixxy
parent
404244c573
commit
e4fe215928
1 changed files with 68 additions and 68 deletions
136
cogs/meme.py
136
cogs/meme.py
|
|
@ -1,6 +1,7 @@
|
||||||
#plugin for sparkytron3000
|
#plugin for sparkytron3000
|
||||||
import os
|
import os
|
||||||
import random
|
import random
|
||||||
|
import time
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
from cogs.base_cog.bot_base_cog import BotBaseCog
|
from cogs.base_cog.bot_base_cog import BotBaseCog
|
||||||
|
|
||||||
|
|
@ -30,81 +31,80 @@ class Meme(BotBaseCog):
|
||||||
return response
|
return response
|
||||||
except:
|
except:
|
||||||
return "error occurred in meme"
|
return "error occurred in meme"
|
||||||
|
|
||||||
|
async def generate_random_meme(self, topic):
|
||||||
|
async with self.http_session.get('https://api.imgflip.com/get_memes') as resp:
|
||||||
|
response_data = await resp.json()
|
||||||
|
response = response_data['data']['memes']
|
||||||
|
memepics = [{'name':image['name'],'url':image['url'],'id':image['id']} for image in response]
|
||||||
|
|
||||||
|
#Pick a meme format
|
||||||
|
memenumber = random.randint(1,99)
|
||||||
|
meme_name = response[memenumber-1]['name']
|
||||||
|
panel_count = response[memenumber-1]['box_count']
|
||||||
|
panel_text = await self.answer_question("Create text for a meme. The meme is " + meme_name + ". It has " + str(panel_count) + " panels. Only create one meme. Do not use emojis or hashtags! Use the topic: " + topic + ". Use the output format (DO NOT USE EXTRA NEWLINES AND DO NOT DESCRIBE THE PICTURE IN YOUR OUTPUT): \n1: [panel 1 text]\n2: [panel 2 text]")
|
||||||
|
|
||||||
|
id = memenumber
|
||||||
|
imgflip_username = os.getenv('imgflip_username')
|
||||||
|
imgflip_password = os.getenv('imgflip_password')
|
||||||
|
params = {
|
||||||
|
'username':imgflip_username,
|
||||||
|
'password':imgflip_password,
|
||||||
|
'template_id':memepics[id-1]['id']
|
||||||
|
}
|
||||||
|
boxes = []
|
||||||
|
text = panel_text.split('\n')
|
||||||
|
for x in range(len(text)):
|
||||||
|
if text[x].strip() != "":
|
||||||
|
item = text[x][3:]
|
||||||
|
if len(params)-3 < panel_count:
|
||||||
|
dictionary = {"text":item, "color": "#ffffff", "outline_color": "#000000"}
|
||||||
|
boxes.append(dictionary)
|
||||||
|
|
||||||
|
for i, box in enumerate(boxes):
|
||||||
|
params[f"boxes[{i}][text]"] = box["text"]
|
||||||
|
params[f"boxes[{i}][color]"] = box["color"]
|
||||||
|
params[f"boxes[{i}][outline_color]"] = box["outline_color"]
|
||||||
|
|
||||||
|
URL = 'https://api.imgflip.com/caption_image'
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with self.http_session.post(URL, params=params) as resp:
|
||||||
|
response = await resp.json()
|
||||||
|
image_link = response['data']['url']
|
||||||
|
except:
|
||||||
|
self.logger.exception("Error occurred in meme")
|
||||||
|
try:
|
||||||
|
#------------------------------------Saving Image Using Aiohttp---------------------------------#
|
||||||
|
filename = memepics[id-1]['name']
|
||||||
|
async with self.http_session.get(image_link) as response:
|
||||||
|
folder = "tmp/meme/"
|
||||||
|
filename = f"{folder}{time.time_ns()}.jpg"
|
||||||
|
|
||||||
|
with open(filename, "wb") as file:
|
||||||
|
while True:
|
||||||
|
chunk = await response.content.read(1024) # Read the response in chunks
|
||||||
|
if not chunk:
|
||||||
|
break
|
||||||
|
file.write(chunk)
|
||||||
|
except:
|
||||||
|
self.logger.exception("Something's Wrong with the aiohttp in meme So try again")
|
||||||
|
return image_link
|
||||||
|
|
||||||
@commands.command(
|
@commands.command(
|
||||||
description="Meme",
|
description="Meme",
|
||||||
help="Generates a meme based on input. Usage: !meme (topic)",
|
help="Generates a meme based on input. Usage: !meme (topic)",
|
||||||
brief="Generate a meme"
|
brief="Generate a meme"
|
||||||
)
|
)
|
||||||
async def meme(self, ctx):
|
async def meme(self, ctx, *topic):
|
||||||
self.logger.info(f"Meme command called by {ctx.author.name}")
|
self.logger.info(f"Meme command called by {ctx.author.name}")
|
||||||
async def generate_random_meme(topic):
|
if topic:
|
||||||
async with self.http_session.get('https://api.imgflip.com/get_memes') as resp:
|
topic = ' '.join(topic)
|
||||||
response_data = await resp.json()
|
else:
|
||||||
response = response_data['data']['memes']
|
topic = await self.answer_question("What topic would you like to generate a meme for?")
|
||||||
memepics = [{'name':image['name'],'url':image['url'],'id':image['id']} for image in response]
|
await ctx.send(f'Generating {topic} meme')
|
||||||
|
link = await self.generate_random_meme(topic)
|
||||||
#Pick a meme format
|
await ctx.send(link)
|
||||||
memenumber = random.randint(1,99)
|
|
||||||
meme_name = response[memenumber-1]['name']
|
|
||||||
panel_count = response[memenumber-1]['box_count']
|
|
||||||
panel_text = await self.answer_question("Create text for a meme. The meme is " + meme_name + ". It has " + str(panel_count) + " panels. Only create one meme. Do not use emojis or hashtags! Use the topic: " + topic + ". Use the output format (DO NOT USE EXTRA NEWLINES AND DO NOT DESCRIBE THE PICTURE IN YOUR OUTPUT): \n1: [panel 1 text]\n2: [panel 2 text]")
|
|
||||||
|
|
||||||
id = memenumber
|
|
||||||
imgflip_username = os.getenv('imgflip_username')
|
|
||||||
imgflip_password = os.getenv('imgflip_password')
|
|
||||||
params = {
|
|
||||||
'username':imgflip_username,
|
|
||||||
'password':imgflip_password,
|
|
||||||
'template_id':memepics[id-1]['id']
|
|
||||||
}
|
|
||||||
boxes = []
|
|
||||||
text = panel_text.split('\n')
|
|
||||||
for x in range(len(text)):
|
|
||||||
if text[x].strip() != "":
|
|
||||||
item = text[x][3:]
|
|
||||||
if len(params)-3 < panel_count:
|
|
||||||
dictionary = {"text":item, "color": "#ffffff", "outline_color": "#000000"}
|
|
||||||
boxes.append(dictionary)
|
|
||||||
|
|
||||||
for i, box in enumerate(boxes):
|
|
||||||
params[f"boxes[{i}][text]"] = box["text"]
|
|
||||||
params[f"boxes[{i}][color]"] = box["color"]
|
|
||||||
params[f"boxes[{i}][outline_color]"] = box["outline_color"]
|
|
||||||
|
|
||||||
URL = 'https://api.imgflip.com/caption_image'
|
|
||||||
|
|
||||||
try:
|
|
||||||
async with self.http_session.post(URL, params=params) as resp:
|
|
||||||
response = await resp.json()
|
|
||||||
image_link = response['data']['url']
|
|
||||||
except:
|
|
||||||
self.logger.exception("Error occurred in meme")
|
|
||||||
try:
|
|
||||||
#------------------------------------Saving Image Using Aiohttp---------------------------------#
|
|
||||||
filename = memepics[id-1]['name']
|
|
||||||
async with self.http_session.get(image_link) as response:
|
|
||||||
folder = "tmp/meme/"
|
|
||||||
filename = folder + topic + str(len(os.listdir(folder))) + ".jpg"
|
|
||||||
|
|
||||||
with open(filename, "wb") as file:
|
|
||||||
while True:
|
|
||||||
chunk = await response.content.read(1024) # Read the response in chunks
|
|
||||||
if not chunk:
|
|
||||||
break
|
|
||||||
file.write(chunk)
|
|
||||||
except:
|
|
||||||
self.logger.exception("Something's Wrong with the aiohttp in meme So try again")
|
|
||||||
return image_link, filename
|
|
||||||
|
|
||||||
try:
|
|
||||||
topic = ctx.message.content.split(" ", maxsplit=1)[1]
|
|
||||||
await ctx.send(f'Generating {topic} meme')
|
|
||||||
link, filepath = await generate_random_meme(topic)
|
|
||||||
await ctx.send(link)
|
|
||||||
except:
|
|
||||||
self.logger.exception("Error occurred in meme")
|
|
||||||
await ctx.send('Something went wrong try again. Usage: !meme (topic)')
|
|
||||||
|
|
||||||
async def setup(bot):
|
async def setup(bot):
|
||||||
await bot.add_cog(Meme(bot))
|
await bot.add_cog(Meme(bot))
|
||||||
Loading…
Add table
Add a link
Reference in a new issue