Fix #87 - enforce proper Item in Ship.__init__

2014-04-30 11:52:56 -04:00
parent 38bd33dce4
commit fb1323ba17
1 changed files with 4 additions and 0 deletions
--- a/eos/saveddata/ship.py
+++ b/eos/saveddata/ship.py
@@ -22,6 +22,10 @@ from eos.effectHandlerHelpers import HandledItem

 class Ship(ItemAttrShortcut, HandledItem):
    def __init__(self, item):
+
+        if item.category.name != "Ship":
+            raise ValueError('Passed item "%s" (category: (%s)) is not under Ship category'%(item.name, item.category.name))
+
        self.__item = item
        self.__itemModifiedAttributes = ModifiedAttributeDict()
        if not isinstance(item, int):