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
-1
View File
@@ -113,7 +113,6 @@ def create_random_tables_content(
) -> str:
template = jinja_env.get_template(f"random_tables_template.{suffix}")
return template.render(
conjure_animals=True,
tables=tables,
use_dnd_decorations=use_dnd_decorations,
)