Merge pull request #64 from openrewrite/requirePrintEqualsInput

Call `requirePrintEqualsInput` from `PythonParser#parseInputs`
openrewrite · Aug 5, 2023 · 177f72d · 177f72d
2 parents f9d1b4e + 5daaafb
commit 177f72d
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/main/java/org/openrewrite/python/PythonParser.java b/src/main/java/org/openrewrite/python/PythonParser.java
@@ -69,16 +69,16 @@ public Stream<SourceFile> parseInputs(Iterable<Input> inputs, @Nullable Path rel
         ParsingExecutionContextView pctx = ParsingExecutionContextView.view(ctx);
         ParsingEventListener parsingListener = pctx.getParsingListener();
 
-        return acceptedInputs(inputs).map(sourceFile -> {
-            Path path = sourceFile.getRelativePath(relativeTo);
-            try (EncodingDetectingInputStream is = sourceFile.getSource(ctx)) {
+        return acceptedInputs(inputs).map(input -> {
+            Path path = input.getRelativePath(relativeTo);
+            try (EncodingDetectingInputStream is = input.getSource(ctx)) {
                 Py.CompilationUnit py = new PsiPythonMapper(path, is.getCharset(), is.isCharsetBomMarked(), styles, mapLanguageLevel(languageLevel))
                         .mapSource(is.readFully());
-                parsingListener.parsed(sourceFile, py);
-                return py;
+                parsingListener.parsed(input, py);
+                return requirePrintEqualsInput(py, input, relativeTo, ctx);
             } catch (Throwable t) {
                 ctx.getOnError().accept(t);
-                return ParseError.build(this, sourceFile, relativeTo, ctx, t);
+                return ParseError.build(this, input, relativeTo, ctx, t);
             }
         });
     }