Fixing code if a ship is buyable at multiple locations, it was returning all locations you could rent at instead

2023-07-05 18:24:15 -07:00 · 2023-07-05 18:24:15 -07:00 · 2812c7bebb
commit 2812c7bebb
parent f733906a23
2 changed files with 5 additions and 1 deletions
--- a/app/bot.py
+++ b/app/bot.py
--- a/app/star_citizen.py
+++ b/app/star_citizen.py
@ -105,7 +105,11 @@ async def get_ship(ship_name):
                inline=True,
            )

-        buy_location = soup.find_all("td", {"class": "Location smwtype_wpg"})
+        buy_location = (
+            soup.find("span", {"id": "Universe_availability"})
+            .findNext("tbody")
+            .find_all("td", {"class", "Location smwtype_wpg"})
+        )

        if len(buy_location):
            embed.add_field(