From 080eff9b62dd4ca7190f3f6ec652668a91fe1c6b Mon Sep 17 00:00:00 2001
From: Luke Robles <lukelrobles@gmail.com>
Date: Sat, 8 Feb 2025 15:24:51 -0800
Subject: [PATCH] Move portraits ot a new method and then use that picture if
 you select the spawns for a single boss

---
 app/cogs/tarkov.py | 16 +++++++---------
 app/tarkov.py      | 11 +++++++++++
 2 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/app/cogs/tarkov.py b/app/cogs/tarkov.py
index cd46edc4..33ac9cf3 100755
--- a/app/cogs/tarkov.py
+++ b/app/cogs/tarkov.py
@@ -147,10 +147,15 @@ class Tarkov(commands.Cog):
             type="rich",
         )
         embed.set_author(name="🎲 Boss Spawn chances by map 🎲")
+        embed.set_thumbnail(url="https://i.ytimg.com/vi/Yis5rmgo_bM/maxresdefault.jpg")
 
         if boss_name:
+            for item in tarkov.get_boss_pictures():
+                if item["name"] == boss_name:
+                    embed.set_thumbnail(url=item["imagePortraitLink"])
+                    break
+
             embed.set_author(name=f"🎲 Spawn chance for {boss_name} 🎲")
-        embed.set_thumbnail(url="https://i.ytimg.com/vi/Yis5rmgo_bM/maxresdefault.jpg")
 
         levels = tarkov.get_tarkov_boss_info()
 
@@ -217,14 +222,7 @@ class Tarkov(commands.Cog):
         #         )
         #         break
 
-        portraits = tarkov.query_tarkov_api(
-            """{
-                bosses(lang: en, gameMode:pve) {
-                name
-                imagePortraitLink
-                }
-            }"""
-        )["bosses"]
+        portraits = tarkov.get_boss_pictures()
 
         for boss in portraits:
             if boss["name"] == boss_name:
diff --git a/app/tarkov.py b/app/tarkov.py
index 8c51fe4d..0a253160 100755
--- a/app/tarkov.py
+++ b/app/tarkov.py
@@ -18,6 +18,17 @@ def query_tarkov_api(query):
     ).json()["data"]
 
 
+def get_boss_pictures():
+    return query_tarkov_api(
+        """{
+                bosses(lang: en, gameMode:pve) {
+                name
+                imagePortraitLink
+                }
+            }"""
+    )["bosses"]
+
+
 def get_tarkov_boss_info():
     """
     Returns a dict of boss spawn chances per map, and their escorts