usb-tools/amaranth
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

sim._pycoro: make src_loc() more robust.

Browse source 
* Guard for finished coroutines.
  * Guard for coroutines yielding from iterators and not generators.
This commit is contained in:
whitequark 2020-08-27 07:11:14 +00:00
parent 141eaf0e40
commit c00219d9f3
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
nmigen/sim/_pycoro.py
View file

@ -32,7 +32,9 @@ class PyCoroProcess(Process):
def src_loc(self): def src_loc(self):
coroutine = self.coroutine coroutine = self.coroutine
while coroutine.gi_yieldfrom is not None: if coroutine is None:
return None
while coroutine.gi_yieldfrom is not None and inspect.isgenerator(coroutine.gi_yieldfrom):
coroutine = coroutine.gi_yieldfrom coroutine = coroutine.gi_yieldfrom
if inspect.isgenerator(coroutine): if inspect.isgenerator(coroutine):
frame = coroutine.gi_frame frame = coroutine.gi_frame