X-Git-Url: https://git.ctpug.org.za/?p=koperkapel.git;a=blobdiff_plain;f=koperkapel%2Fvehicles%2Fwalking.py;h=8061f2c31c94e83689d286907252f2d61cbaaaca;hp=f7b73aa398cbf31ca6451f0b117bd0c980e00139;hb=94041e07b1111324d2666a1f3aa2ea5cdbdbb805;hpb=96dd67b1005fa0123f96409cecb62ae1a2342152 diff --git a/koperkapel/vehicles/walking.py b/koperkapel/vehicles/walking.py index f7b73aa..8061f2c 100644 --- a/koperkapel/vehicles/walking.py +++ b/koperkapel/vehicles/walking.py @@ -3,26 +3,30 @@ from .base import Vehicle, circle_of_seats from ..roaches import ( default_roaches, roaches_quartet, roaches_nonet, WorldRoach) +from ..weapons import default_weapons class Walking(Vehicle): vehicle_type = "walking" + weapons_taped_on = False def init_seats(self): return circle_of_seats(6, vehicle=self) def get_avatar(self, world): + weapon = default_weapons.assemble( + world.weapons.current, tape=self.weapons_taped_on) num_roaches = len(world.roaches) roach = WorldRoach() if num_roaches == 1: # Return a single large roach - avatar = default_roaches.assemble(roach) + avatar = default_roaches.assemble(roach, weapon) avatar.anchor = (0, 0) elif num_roaches < 6: - avatar = roaches_quartet.assemble(roach) + avatar = roaches_quartet.assemble(roach, weapon) avatar.anchor = (0, 0) else: - avatar = roaches_nonet.assemble(roach) + avatar = roaches_nonet.assemble(roach, weapon) avatar.anchor = (0, 0) return avatar