When resolving +1 weapons, replaced check for inheritance from Weapon

with check for *improved_version* method.

Fixes https://github.com/canismarko/dungeon-sheets/issues/112
This commit is contained in:
Mark Wolfman
2021-09-07 20:10:53 -05:00
parent 288b090ca9
commit 4f8a0e442b
5 changed files with 23 additions and 12 deletions
+2 -7
View File
@@ -136,13 +136,8 @@ class ContentRegistry:
else:
attr = found_attrs[0]
# Apply weapon/etc. bonuses
if bonus > 0:
if (
issubclass(attr, weapons.Weapon)
or issubclass(attr, armor.Shield)
or issubclass(attr, armor.Armor)
):
attr = attr.improved_version(bonus)
if bonus > 0 and hasattr(attr, 'improved_version'):
attr = attr.improved_version(bonus)
return attr