def revert_overrides_agents(self): if not hasattr(self, 'overriden_agents'): return else: for agent_type, factory in self.overriden_agents.iteritems(): if factory: registry.override(agent_type, factory)
def override_agent(self, agent_type, factory): if not hasattr(self, 'overriden_agents'): self.overriden_agents = dict() old = registry.registry_lookup(agent_type) self.overriden_agents[agent_type] = old registry.override(agent_type, factory)