Fixed 2h detection (#287)

d4lfteam · Jun 13, 2024 · bd154a9 · bd154a9
1 parent e03589e
commit bd154a9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/__init__.py b/src/__init__.py
@@ -1 +1 @@
-__version__ = "5.5.5"
+__version__ = "5.5.6"
diff --git a/src/item/descr/item_type.py b/src/item/descr/item_type.py
@@ -88,10 +88,9 @@ def _find_item_power_and_type(item: Item, concatenated_str: str) -> Item:
                 last_char_idx = tmp_idx
                 max_length = len(item_type.value)
     # common mistake is that "Armor" is on a seperate line and can not be detected properly
-    # TODO: Language specific
     if item.item_type is None and ("chest" in concatenated_str or "armor" in concatenated_str):
         item.item_type = ItemType.ChestArmor
-    if "two-handed" in concatenated_str or "two- handed" in concatenated_str:
+    if any(substring in concatenated_str for substring in ["two -handed", "two handed", "two- handed", "two-handed"]):
         if item.item_type == ItemType.Sword:
             item.item_type = ItemType.Sword2H
         elif item.item_type == ItemType.Mace: