From ed73d02203e0096a7fcec28506a23cc860c506f5 Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Sat, 28 Nov 2015 14:45:27 -0500 Subject: Pragmas on decorators apply to the entire function or class. #131 --- lab/parser.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'lab') diff --git a/lab/parser.py b/lab/parser.py index bb593f8..70c2b6b 100644 --- a/lab/parser.py +++ b/lab/parser.py @@ -108,7 +108,9 @@ class ParserMain(object): for lineno, ltext in enumerate(cp.lines, start=1): m0 = m1 = m2 = m3 = a = ' ' - if lineno in cp.raw_statements: + if lineno in cp.statements: + m0 = '=' + elif lineno in cp.raw_statements: m0 = '-' exits = exit_counts.get(lineno, 0) if exits > 1: -- cgit v1.2.1