From bb2ceb99683f58d88e96ad427283691261955c28 Mon Sep 17 00:00:00 2001 From: Dan Albert Date: Sat, 10 Sep 2022 15:59:34 -0700 Subject: [PATCH] Log planning times per-task. --- game/commander/tasks/packageplanningtask.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/game/commander/tasks/packageplanningtask.py b/game/commander/tasks/packageplanningtask.py index e515f013..e2ab8c00 100644 --- a/game/commander/tasks/packageplanningtask.py +++ b/game/commander/tasks/packageplanningtask.py @@ -92,6 +92,7 @@ class PackagePlanningTask(TheaterCommanderTask, Generic[MissionTargetT]): return 1 def fulfill_mission(self, state: TheaterState) -> bool: + color = "blue" if state.context.coalition.player else "red" self.propose_flights() fulfiller = PackageFulfiller( state.context.coalition, @@ -99,11 +100,12 @@ class PackagePlanningTask(TheaterCommanderTask, Generic[MissionTargetT]): state.context.game_db.flights, state.context.settings, ) - self.package = fulfiller.plan_mission( - ProposedMission(self.target, self.flights), - self.purchase_multiplier, - state.context.tracer, - ) + with state.context.tracer.trace(f"{color} {self.flights[0].task} planning"): + self.package = fulfiller.plan_mission( + ProposedMission(self.target, self.flights), + self.purchase_multiplier, + state.context.tracer, + ) return self.package is not None def propose_common_escorts(self) -> None: