diff --git a/app/bot.py b/app/bot.py index d3db539a..cc23b6eb 100755 --- a/app/bot.py +++ b/app/bot.py @@ -5,6 +5,11 @@ import discord import os import random import requests +import requests_cache + +requests_cache.install_cache( + "request-cache", backend="sqlite", expire_after=300, allowable_methods=("GET",) +) intents = discord.Intents.default() intents.message_content = True diff --git a/app/cogs/tarkov.py b/app/cogs/tarkov.py index b33454c9..492433e1 100755 --- a/app/cogs/tarkov.py +++ b/app/cogs/tarkov.py @@ -241,7 +241,19 @@ class Tarkov(commands.Cog): response = requests.get(wiki_url + boss_name).text soup = BeautifulSoup(response, "html.parser") - embed.set_thumbnail(url=soup.find("a", class_="image").get("href")) + portraits = tarkov.query_tarkov_api( + """{ + bosses(lang: en, gameMode:pve) { + name + imagePortraitLink + } + }""" + )["bosses"] + + for boss in portraits: + if boss["name"] == boss_name: + embed.set_thumbnail(url=boss["imagePortraitLink"]) + break health = soup.find("table", class_="wikitable").find_next("td").text.rstrip() embed.add_field(name="Health", value=health, inline=False) diff --git a/app/requirements.txt b/app/requirements.txt index 7237709a..33fee047 100755 --- a/app/requirements.txt +++ b/app/requirements.txt @@ -10,6 +10,7 @@ markovify openai owotext requests +requests-cache wifi-qrcode-generator wolframalpha yfinance \ No newline at end of file