class HasEquipment(object):
"""Creatures with this mixin class have an equipment and a bag."""
def __init__(self, *args, **kwargs):
self.equipment = Equipment()
super().__init__(*args, **kwargs)
def get_damage_info(self):
damage_info = self.equipment.get_damage_info()
if damage_info is not None:
dices, highest_side, addition = damage_info
addition += self.damage
else:
dices = self.unarmed_dices
highest_side = self.unarmed_sides
addition = self.damage
return Dice(dices, highest_side, addition)
class HasEquipment(object):
"""Creatures with this mixin class have an equipment and a bag."""
def __init__(self, *args, **kwargs):
self.equipment = Equipment()
super().__init__(*args, **kwargs)
def get_damage_info(self):
damage_info = self.equipment.get_damage_info()
if damage_info is not None:
dices, highest_side, addition = damage_info
addition += self.damage
else:
dices = self.unarmed_dices
highest_side = self.unarmed_sides
addition = self.damage
return Dice(dices, highest_side, addition)
def __init__(self, *args, **kwargs):
self.equipment = Equipment()
super().__init__(*args, **kwargs)
def test_creature_equipment():
equipment = Equipment()
items = {
Slot.Right_Hand: Weapon("short sword +1", 0, (1, 6, 1)),
Slot.Left_Hand: Weapon("short sword", 0, (1, 6, 0)),
Slot.Head: Armor("helmet", 0, 1, [Slot.Head]),
Slot.Body: Armor("armor", 0, 4, [Slot.Body], stats=[(Stat.strength, 2)]),
Slot.Feet: Armor("boots", 0, 1, [Slot.Feet]),
}
# bag items
equipment.bag_items(items.values())
for item in items.values():
assert item in equipment._bag
assert len(equipment._bag) == len(items)
# equip items
for slot, item in items.items():
equipment.equip_from_bag(equipment._bag.index(item), slot)
assert equipment.get_item(slot) is item
assert len(equipment._bag) == 0
assert equipment.applied_stats[Stat.armor] == 6
assert equipment.applied_stats[Stat.strength] == 2
# unequip items
for slot in items:
equipment.unequip(slot)
assert len(equipment._bag) == len(items)
two_hander = Weapon("two-handed sword", 0, (1, 6, 0), two_handed=True)
equipment.equip(two_hander, two_hander.compatible_slots[0])
equipment.unequip(two_hander.compatible_slots[1])
for worn_item in equipment._worn_items.values():
assert worn_item is None
for stat_value in equipment.applied_stats.values():
assert stat_value == 0
def __init__(self, *args, **kwargs):
self.equipment = Equipment()
super().__init__(*args, **kwargs)
