Add added_anims arg to PiCreatureScene.look_at

This commit is contained in:
Grant Sanderson 2020-08-16 09:44:54 -07:00
parent caca6d0c0e
commit a822c93019

View file

@ -226,13 +226,15 @@ class PiCreatureScene(Scene):
def change_mode(self, mode):
self.play(self.get_primary_pi_creature().change_mode, mode)
def look_at(self, thing_to_look_at, pi_creatures=None, **kwargs):
def look_at(self, thing_to_look_at, pi_creatures=None, added_anims=None, **kwargs):
if pi_creatures is None:
pi_creatures = self.get_pi_creatures()
args = list(it.chain(*[
[pi.look_at, thing_to_look_at]
for pi in pi_creatures
]))
if added_anims is not None:
args.extend(added_anims)
self.play(*args, **kwargs)