Support context management on imports

diff --git a/grammar/JinjaGrammar.g4 b/grammar/JinjaGrammar.g4
index 9669ecbfc186374ad8d98aebe015bdc32635f83b..f9dda842e6a9e0f3c5d008373c882a64583703ac 100644 (file)
--- a/grammar/JinjaGrammar.g4
+++ b/grammar/JinjaGrammar.g4
@@ -57,8 +57,8 @@ statement_import_variable_list
     ;
 
 statement_import
-    : STATEMENT_ID_IMPORT SP statement_import_file SP STATEMENT_ID_IMPORT_AS SP variable_name
-    | STATEMENT_ID_FROM SP statement_import_file SP STATEMENT_ID_IMPORT SP statement_import_variable_list
+    : STATEMENT_ID_IMPORT SP statement_import_file SP STATEMENT_ID_IMPORT_AS SP variable_name (SP statement_include_context)?
+    | STATEMENT_ID_FROM SP statement_import_file SP STATEMENT_ID_IMPORT SP statement_import_variable_list (SP statement_include_context)?
     ;
 
 block_statement_id