Undoing removal of drones now restores their proper state

This commit is contained in:
DarkPhoenix
2019-04-11 18:20:56 +03:00
parent e90e9a5ca1
commit 5579929f83
6 changed files with 65 additions and 42 deletions

View File

@@ -336,6 +336,7 @@ class HandledProjectedModList(HandledList):
class HandledProjectedDroneList(HandledDroneCargoList):
def append(self, proj):
proj.projected = True
HandledList.append(self, proj)
@@ -343,6 +344,19 @@ class HandledProjectedDroneList(HandledDroneCargoList):
# Remove invalid or non-projectable drones
if proj.isInvalid or not proj.item.isType("projected"):
self.remove(proj)
return False
return True
def insert(self, idx, proj):
proj.projected = True
HandledList.insert(self, idx, proj)
# Remove invalid or non-projectable drones
if proj.isInvalid or not proj.item.isType("projected"):
self.remove(proj)
return False
return True
class HandledItem(object):