def with_plugs(self, **subplugs):
"""Substitute plugs for placeholders for this phase."""
plugs_by_name = {plug.name: plug for plug in self.plugs}
new_plugs = dict(plugs_by_name)
for name, sub_class in subplugs.items():
original_plug = plugs_by_name.get(name)
accept_substitute = True
if original_plug is None:
accept_substitute = False
elif isinstance(original_plug.cls, plugs.PlugPlaceholder):
accept_substitute = issubclass(sub_class,
original_plug.cls.base_class)
else:
# Check __dict__ to see if the attribute is explicitly defined in the
# class, rather than being defined in a parent class.
accept_substitute = (
'auto_placeholder' in original_plug.cls.__dict__
and original_plug.cls.auto_placeholder
and issubclass(sub_class, original_plug.cls))
if not accept_substitute:
raise plugs.InvalidPlugError(
'Could not find valid placeholder for substitute plug %s '
'required for phase %s' % (name, self.name))
new_plugs[name] = mutablerecords.CopyRecord(original_plug,
cls=sub_class)
return mutablerecords.CopyRecord(
self,
plugs=list(new_plugs.values()),
options=self.options.format_strings(**subplugs),
measurements=[m.with_args(**subplugs) for m in self.measurements])
def with_plugs(self, **subplugs):
"""Substitute plugs for placeholders for this phase."""
plugs_by_name = {plug.name: plug for plug in self.plugs}
new_plugs = dict(plugs_by_name)
for name, sub_class in subplugs.iteritems():
original_plug = plugs_by_name.get(name)
if (original_plug is None
or not isinstance(original_plug.cls, plugs.PlugPlaceholder)
or
not issubclass(sub_class, original_plug.cls.base_class)):
raise plugs.InvalidPlugError(
'Could not find valid placeholder for substitute plug %s '
'required for phase %s' % (name, self.name))
new_plugs[name] = mutablerecords.CopyRecord(original_plug,
cls=sub_class)
return mutablerecords.CopyRecord(
self,
plugs=new_plugs.values(),
options=self.options.format_strings(**subplugs),
measurements=[m.with_args(**subplugs) for m in self.measurements])
