Improvements to bytecode tools.

1 files changed, 5 insertions, 1 deletions

diff --git a/lab/parser.py b/lab/parser.py
index 374acea..ce6902a 100644
--- a/lab/parser.py
+++ b/lab/parser.py
@@ -118,10 +118,14 @@ class ParserMain(object):
             else:
                 srclines = None
             print("\n%s: " % bp.code)
+            upto = None
             for disline in disgen.disgen(bp.code):
                 if disline.first:
                     if srclines:
-                        print("%100s%s" % ("", srclines[disline.lineno-1]))
+                        upto = upto or disline.lineno-1
+                        while upto <= disline.lineno-1:
+                            print("%100s%s" % ("", srclines[upto]))
+                            upto += 1
                     elif disline.offset > 0:
                         print("")
                 line = disgen.format_dis_line(disline)