From 760d76ce69b28e4e02ab66145aa3cf20b1935303 Mon Sep 17 00:00:00 2001 From: Quildra Date: Sat, 16 Nov 2024 09:55:40 +0000 Subject: [PATCH] =?UTF-8?q?=20-=20Fix=20for=20Flab=C3=A9b=C3=A9=20and=20it?= =?UTF-8?q?s=20flower=20colours?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ui/workers/gather_evolutions_worker.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/ui/workers/gather_evolutions_worker.py b/ui/workers/gather_evolutions_worker.py index 46431f4..91550ab 100644 --- a/ui/workers/gather_evolutions_worker.py +++ b/ui/workers/gather_evolutions_worker.py @@ -115,6 +115,19 @@ class GatherEvolutions(QRunnable): } evolution_tree["evolves_to"].append(node) + if evolution_tree["pokemon"] == "Flabébé": + def fix_form(node, new_form): + node["form"] = new_form + for next in node["evolves_to"]: + fix_form(next, new_form) + + flower_form = get_form_name(pokemon_form) + if "female" in flower_form.lower(): + flower_form = flower_form.replace("Female", "").replace("female", "").strip() + else: + flower_form = flower_form.replace("Male", "").replace("male", "").strip() + fix_form(evolution_tree, flower_form) + cacheable_container = {} if evolution_tree: self.traverse_and_store(evolution_tree, cacheable_container, gender)