OK, start with a cache argument in generate_work_orders.

pull/926/head
jMyles 2019-03-24 15:23:04 +01:00
parent 4c0d51fd7f
commit c06cf4ba3b
1 changed files with 4 additions and 4 deletions

View File

@ -630,7 +630,7 @@ class Bob(Character):
return treasure_map return treasure_map
def generate_work_orders(self, map_id, *capsules, num_ursulas=None): def generate_work_orders(self, map_id, *capsules, num_ursulas=None, cache=False):
from nucypher.policy.models import WorkOrder # Prevent circular import from nucypher.policy.models import WorkOrder # Prevent circular import
try: try:
@ -659,9 +659,9 @@ class Bob(Character):
work_order = WorkOrder.construct_by_bob( work_order = WorkOrder.construct_by_bob(
arrangement_id, capsules_to_include, ursula, self) arrangement_id, capsules_to_include, ursula, self)
generated_work_orders[node_id] = work_order generated_work_orders[node_id] = work_order
# TODO: Fix this. It's always taking the last capsule
# TODO: Fix this. It's always using the last capsule, via leaky-loop if cache:
self._saved_work_orders[node_id][capsule] = work_order self._saved_work_orders[node_id][capsule] = work_order
if num_ursulas == len(generated_work_orders): if num_ursulas == len(generated_work_orders):
break break