Relpace deck gear with egg if level below 5

zjosua · Nov 19, 2023 · a2796e8 · a2796e8
1 parent 0a05c69
commit a2796e8
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/src/pokemanki/_version.py b/src/pokemanki/_version.py
@@ -37,4 +37,4 @@
 Version information
 """
 
-__version__ = "1.5.0"
+__version__ = "1.5.1"
diff --git a/src/pokemanki/main.py b/src/pokemanki/main.py
@@ -166,10 +166,19 @@ def replace_gears(
     soup = BeautifulSoup(content.tree, "html.parser")
     for tr in soup.select("tr[id]"):
         deck_id = int(tr["id"])
-        name = next((pokemon[0] for pokemon in pokemons if pokemon[1] == deck_id), None)
-        if name:
-            tr.select("img.gears")[0]["src"] = f"{pkmnimgfolder}/{name}.png"
-            tr.select("img.gears")[0]["class"] = "gears pokemon"
+        name_and_level = next(
+            ((pokemon[0], pokemon[2]) for pokemon in pokemons if pokemon[1] == deck_id),
+            None,
+        )
+        if not name_and_level:
+            continue
+        if name_and_level[1] < 5:
+            tr.select("img.gears")[0]["src"] = f"{pkmnimgfolder}/Egg.png"
+        else:
+            tr.select("img.gears")[0][
+                "src"
+            ] = f"{pkmnimgfolder}/{name_and_level[0]}.png"
+        tr.select("img.gears")[0]["class"] = "gears pokemon"
     style = soup.new_tag("style")
     style.string = ".gears.pokemon{filter:none;opacity:1}"
     soup.append(style)