Mercurial > parpg-source
view components/equip.py @ 170:0296f5f74a0f
Fixed saving and loading that broke with changing of the settings module.
author | Beliar <KarstenBock@gmx.net> |
---|---|
date | Sat, 25 Feb 2012 22:05:39 +0100 |
parents | fa5c769468f0 |
children |
line wrap: on
line source
# This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. from base import Base class Equip(Base): """ Component that stores the equipment (what is being worn/wielded). """ def __init__(self): Base.__init__(self, head=object, neck=object, body=object, belt=object, leg=object, feet=object, l_arm=object, r_arm=object) self.head = None self.neck = None self.body = None self.belt = None self.leg = None self.feet = None self.l_arm = None self.r_arm = None @property def saveable_fields(self): return [] class SlotInvalidError(Exception): """Error that gets raised when the slot is invalid.""" def __init__(self, slot): self.slot = slot def __str__(self): return "\"%s\" is not a valid slot." % self.slot class AlreadyEquippedError(Exception): """Error that gets raised when the equipable already has a wearer""" def __str__(self): return "The equipable is already weared." class CannotBeEquippedInSlot(Exception): """Error that gets raised when the equipable can't be equiped in that slot""" def __init__(self, slot, equipable): self.slot = slot self.equipable = equipable def __str__(self): return ("%s is not in the equipables slots. (%s)" % (self.slot, ', '.join(self.equipable.possible_slots)) ) def equip(wearer, equipable, slot): """Equip the wearer with the given equipable. @returns The equipable that was at the given slot, or None""" if equipable.wearer: raise AlreadyEquippedError if slot in equipable.possible_slots: try: old_item = getattr(wearer, slot) if hasattr(wearer, slot) else None setattr(wearer, slot, equipable) equipable.in_slot = slot equipable.wearer = wearer if old_item: old_item.in_slot = None old_item.wearer = None return old_item except AttributeError: raise SlotInvalidError(slot) raise CannotBeEquippedInSlot(slot, equipable) def get_equipable(wearer, slot): """Return the equipable in the given slot""" if not wearer: return None try: item = getattr(wearer, slot) return item except AttributeError: raise SlotInvalidError(slot) def take_equipable(wearer, slot): """Remove equipable from the given slot and return it""" item = get_equipable(wearer, slot) setattr(wearer, slot, None) if item: item.in_slot = None item.wearer = None return item