Add OperationSection to event stream operation generation (#487)

SamRemis · web-flow · commit 4bcdb752773d · 2025-04-11T12:15:45.000-04:00
diff --git a/codegen/core/src/main/java/software/amazon/smithy/python/codegen/ClientGenerator.java b/codegen/core/src/main/java/software/amazon/smithy/python/codegen/ClientGenerator.java
@@ -905,7 +905,7 @@ private void writeSharedOperationInit(PythonWriter writer, OperationShape operat
     }
 
     private void generateEventStreamOperation(PythonWriter writer, OperationShape operation) {
-        writer.pushState();
+        writer.pushState(new OperationSection(service, operation));
         writer.addDependency(SmithyPythonDependency.SMITHY_CORE);
         var operationSymbol = symbolProvider.toSymbol(operation);
         writer.putContext("operation", operationSymbol);

Original file line number	Diff line number	Diff line change
`@@ -905,7 +905,7 @@ private void writeSharedOperationInit(PythonWriter writer, OperationShape operat`
`905`	`905`	`}`
`906`	`906`
`907`	`907`	`private void generateEventStreamOperation(PythonWriter writer, OperationShape operation) {`
`908`		`- writer.pushState();`
	`908`	`+ writer.pushState(new OperationSection(service, operation));`
`909`	`909`	`writer.addDependency(SmithyPythonDependency.SMITHY_CORE);`
`910`	`910`	`var operationSymbol = symbolProvider.toSymbol(operation);`
`911`	`911`	`writer.putContext("operation", operationSymbol);`