Skip to content

Commit ee6ce70

Browse files
authored
Improve code-generated Config object docstring (#595)
1 parent f9fbb43 commit ee6ce70

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

codegen/core/src/main/java/software/amazon/smithy/python/codegen/generators/ConfigGenerator.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
import java.util.List;
1111
import java.util.Locale;
1212
import java.util.TreeSet;
13+
import software.amazon.smithy.aws.traits.ServiceTrait;
1314
import software.amazon.smithy.codegen.core.Symbol;
1415
import software.amazon.smithy.model.knowledge.EventStreamIndex;
1516
import software.amazon.smithy.model.knowledge.ServiceIndex;
@@ -331,6 +332,11 @@ private void generateConfig(GenerationContext context, PythonWriter writer) {
331332
}
332333

333334
var finalProperties = List.copyOf(properties);
335+
final String serviceId = context.settings()
336+
.service(context.model())
337+
.getTrait(ServiceTrait.class)
338+
.map(ServiceTrait::getSdkId)
339+
.orElse(context.settings().service().getName());
334340
writer.pushState(new ConfigSection(finalProperties));
335341
writer.addStdlibImport("dataclasses", "dataclass");
336342
writer.write("""
@@ -352,7 +358,7 @@ def __init__(
352358
${C|}
353359
""",
354360
configSymbol.getName(),
355-
context.settings().service().getName(),
361+
serviceId,
356362
writer.consumer(w -> writePropertyDeclarations(w, finalProperties)),
357363
writer.consumer(w -> writeInitParams(w, finalProperties)),
358364
writer.consumer(w -> documentProperties(w, finalProperties)),

0 commit comments

Comments
 (0)