From fb1323ba17f6ae1557ed077b678d995a6fd5af1d Mon Sep 17 00:00:00 2001 From: blitzmann Date: Wed, 30 Apr 2014 11:52:56 -0400 Subject: [PATCH] Fix #87 - enforce proper Item in `Ship.__init__` --- eos/saveddata/ship.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/eos/saveddata/ship.py b/eos/saveddata/ship.py index 6a0118011..05cd99c02 100755 --- 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):