jan/masque
1
1
Fork 0
Code Issues Pull requests Releases Wiki Activity

Allow Pattern.ref() to take an Abstract

Browse source
This commit is contained in:
jan 2023-10-12 01:31:05 -07:00
commit 8d2d1ffd50
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
masque/pattern.py
View file

@ -681,7 +681,7 @@ class Pattern(PortList, AnnotatableImpl, Mirrorable):
""" """
return bool(self.ports) return bool(self.ports)
def ref(self, target: str | None, *args: Any, **kwargs: Any) -> Self: def ref(self, target: str | Abstract | None, *args: Any, **kwargs: Any) -> Self:
""" """
Convenience function which constructs a `Ref` object and adds it Convenience function which constructs a `Ref` object and adds it
to this pattern. to this pattern.
@ -694,6 +694,8 @@ class Pattern(PortList, AnnotatableImpl, Mirrorable):
Returns: Returns:
self self
""" """
if isinstance(target, Abstract):
target = target.name
self.refs[target].append(Ref(*args, **kwargs)) self.refs[target].append(Ref(*args, **kwargs))
return self return self