Added daily waifu

Simplifyed more code
pull/8/head
sgoudham 4 years ago
parent dd01f45c5a
commit b52975f329

@ -248,6 +248,7 @@ class Anime(Cog):
def __init__(self, bot): def __init__(self, bot):
self.bot = bot self.bot = bot
self.headers = {'apikey': my_waifu_list_auth}
@Cog.listener() @Cog.listener()
async def on_ready(self): async def on_ready(self):
@ -272,11 +273,10 @@ class Anime(Cog):
url = "https://mywaifulist.moe/api/v1/search/" url = "https://mywaifulist.moe/api/v1/search/"
data = {"term": waifu2, data = {"term": waifu2,
'content-type': "application/json"} 'content-type': "application/json"}
headers = {'apikey': my_waifu_list_auth}
# Searching API for waifu(s) # Searching API for waifu(s)
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.post(url, data=data, headers=headers) as resp: async with session.post(url, data=data, headers=self.headers) as resp:
# Store waifu's in dict when request is successful, else send an error # Store waifu's in dict when request is successful, else send an error
if resp.status == 200: if resp.status == 200:
waifu_dict = await resp.json() waifu_dict = await resp.json()
@ -315,11 +315,10 @@ class Anime(Cog):
# Set variables to retrieve data from the API # Set variables to retrieve data from the API
url = "https://mywaifulist.moe/api/v1/meta/random" url = "https://mywaifulist.moe/api/v1/meta/random"
headers = {'apikey': my_waifu_list_auth}
# Retrieve a random waifu from the API # Retrieve a random waifu from the API
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.get(url, headers=headers) as resp: async with session.get(url, headers=self.headers) as resp:
# Store waifu's in dict when request is successful, else send an error # Store waifu's in dict when request is successful, else send an error
if resp.status == 200: if resp.status == 200:
waifu_dict = await resp.json() waifu_dict = await resp.json()
@ -336,11 +335,10 @@ class Anime(Cog):
"""Returns the daily Waifu from MyWaifuList""" """Returns the daily Waifu from MyWaifuList"""
url = "https://mywaifulist.moe/api/v1/meta/daily" url = "https://mywaifulist.moe/api/v1/meta/daily"
headers = {'apikey': my_waifu_list_auth}
# Retrieve a random waifu from the API # Retrieve a random waifu from the API
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.get(url, headers=headers) as resp: async with session.get(url, headers=self.headers) as resp:
# Store waifu's in dict when request is successful, else send an error # Store waifu's in dict when request is successful, else send an error
if resp.status == 200: if resp.status == 200:
waifu_dict = await resp.json() waifu_dict = await resp.json()

Loading…
Cancel
Save