def load(self, db, worldid):
super(WeaponHolder, self).load(db, worldid)
self.__init()
weapons = db("SELECT weapon_id, number, remaining_ticks FROM weapon_storage WHERE owner_id = ?", worldid)
for weapon_id, number, ticks in weapons:
# create weapon and add to storage manually
if self.session.db.get_weapon_stackable(weapon_id):
weapon = StackableWeapon(self.session, weapon_id)
weapon.set_number_of_weapons(number)
else:
weapon = Weapon(self.session, weapon_id)
self._weapon_storage.append(weapon)
# if weapon not ready add scheduled call and remove from fireable
if ticks:
weapon.attack_ready = False
Scheduler().add_new_object(weapon.make_attack_ready, weapon, ticks)
else:
self._fireable.append(weapon)
weapon.add_weapon_fired_listener(Callback(self._remove_from_fireable, weapon))
weapon.add_attack_ready_listener(Callback(self._add_to_fireable, weapon))
weapon.add_weapon_fired_listener(self._increase_fired_weapons_number)
self.on_storage_modified()
# load target after all objects have been loaded
Scheduler().add_new_object(Callback(self.load_target, db), self, run_in = 0)
self.log.debug("%s weapon storage after load: %s", self, self._weapon_storage)
