From ae6fe8b43c08abe215dde355e04e561ad2dd1808 Mon Sep 17 00:00:00 2001
From: Milan Rossa
Date: Mon, 19 Aug 2019 14:16:33 +0200
Subject: py/compile: Improve the line numbering precision for comprehensions.

The line number for comprehensions is now always reported as the correct
global location in the script, instead of just "line 1".
---
 py/compile.c | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'py')

diff --git a/py/compile.c b/py/compile.c
index 27b706c8f..c0ae3de11 100644
--- a/py/compile.c
+++ b/py/compile.c
@@ -3112,6 +3112,9 @@ STATIC void compile_scope(compiler_t *comp, scope_t *scope, pass_kind_t pass) {
             scope->num_pos_args = 1;
         }
 
+        // Set the source line number for the start of the comprehension
+        EMIT_ARG(set_source_line, pns->source_line);
+
         if (scope->kind == SCOPE_LIST_COMP) {
             EMIT_ARG(build, 0, MP_EMIT_BUILD_LIST);
         } else if (scope->kind == SCOPE_DICT_COMP) {
-- 
cgit v1.2.3