variable fix

Author: SamRemis

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

@@ -14,7 +14,6 @@
 import java.util.function.Function;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
-
 import software.amazon.smithy.aws.traits.ServiceTrait;
 import software.amazon.smithy.codegen.core.CodegenException;
 import software.amazon.smithy.codegen.core.SymbolDependency;
@@ -278,10 +277,16 @@ private static void writeDocsSkeleton(
             writer.addDependency(SmithyPythonDependency.SPHINX);
             writer.addDependency(SmithyPythonDependency.SPHINX_PYDATA_THEME);
         });
-        writeConf(settings, context);
+        var service = context.model().expectShape(settings.service());
+        String projectName = service.getTrait(TitleTrait.class)
+                .map(StringTrait::getValue)
+                .orElseGet(() -> service.getTrait(ServiceTrait.class)
+                        .map(ServiceTrait::getSdkId)
+                        .orElse(context.settings().service().getName()));
+        writeConf(settings, context, projectName);
+        writeIndexes(context, projectName);
         writeMakeBat(context);
         writeMakeFile(context);
-        writeIndexes(context);
     }
 
     /**
@@ -293,13 +298,10 @@ private static void writeDocsSkeleton(
      */
     private static void writeConf(
             PythonSettings settings,
-            GenerationContext context
+            GenerationContext context,
+            String projectName
     ) {
-        var service = context.model().expectShape(settings.service());
         String version = settings.moduleVersion();
-        String project = service.getTrait(TitleTrait.class)
-                .map(StringTrait::getValue)
-                .orElse(service.getTrait(ServiceTrait.class).get().getSdkId());
         context.writerDelegator().useFileWriter("docs/conf.py", "", writer -> {
             writer.write("""
                     import os
@@ -325,12 +327,12 @@ private static void writeConf(
                     html_theme = 'pydata_sphinx_theme'
                     html_theme_options = {
                         "logo": {
-                            "text": "AWS SDK for Python",
+                            "text": "$L",
                         }
                     }
 
                     autodoc_typehints = 'both'
-                                """, project, version);
+                                """, projectName, version, projectName);
         });
     }
 
@@ -419,20 +421,20 @@ private static void writeMakeFile(
      *
      * @param context The generation context containing the writer delegator.
      */
-    private static void writeIndexes(GenerationContext context) {
+    private static void writeIndexes(GenerationContext context, String projectName) {
         // Write the main index file for the documentation
         context.writerDelegator().useFileWriter("docs/index.rst", "", writer -> {
             writer.write("""
-                    AWS SDK For Python
-                    ====================================================
+                    $L
+                    $L
 
                     ..  toctree::
                         :maxdepth: 2
                         :titlesonly:
                         :glob:
 
                         */index
-                                """);
+                                """, projectName, "=".repeat(projectName.length()));
         });
 
         // Write the index file for the client section

codegen/core/src/main/java/software/amazon/smithy/python/codegen/writer/MarkdownToRstDocConverter.java

@@ -4,14 +4,8 @@
  */
 package software.amazon.smithy.python.codegen.writer;
 
-import org.jsoup.Jsoup;
-import org.jsoup.nodes.Document;
-import org.jsoup.nodes.Element;
-import org.jsoup.nodes.Node;
-import org.jsoup.nodes.TextNode;
-import org.jsoup.select.NodeVisitor;
-import software.amazon.smithy.utils.SetUtils;
-import software.amazon.smithy.utils.SmithyInternalApi;
+import static org.jsoup.nodes.Document.OutputSettings.Syntax.html;
+
 import org.commonmark.node.BlockQuote;
 import org.commonmark.node.FencedCodeBlock;
 import org.commonmark.node.Heading;
@@ -20,8 +14,14 @@
 import org.commonmark.node.ThematicBreak;
 import org.commonmark.parser.Parser;
 import org.commonmark.renderer.html.HtmlRenderer;
-
-import static org.jsoup.nodes.Document.OutputSettings.Syntax.html;
+import org.jsoup.Jsoup;
+import org.jsoup.nodes.Document;
+import org.jsoup.nodes.Element;
+import org.jsoup.nodes.Node;
+import org.jsoup.nodes.TextNode;
+import org.jsoup.select.NodeVisitor;
+import software.amazon.smithy.utils.SetUtils;
+import software.amazon.smithy.utils.SmithyInternalApi;
 
 /**
  * Add a runtime plugin to convert the HTML docs that are provided by services into RST
@@ -30,8 +30,12 @@
 public class MarkdownToRstDocConverter {
     private static final Parser MARKDOWN_PARSER = Parser.builder()
             .enabledBlockTypes(SetUtils.of(
-                    Heading.class, HtmlBlock.class, ThematicBreak.class, FencedCodeBlock.class,
-                    BlockQuote.class, ListBlock.class))
+                    Heading.class,
+                    HtmlBlock.class,
+                    ThematicBreak.class,
+                    FencedCodeBlock.class,
+                    BlockQuote.class,
+                    ListBlock.class))
             .build();
 
     // Singleton instance
@@ -46,11 +50,10 @@ public static MarkdownToRstDocConverter getInstance() {
         return DOC_CONVERTER;
     }
 
-
     public String convertCommonmarkToRst(String commonmark) {
         String html =
                 HtmlRenderer.builder().escapeHtml(false).build().render(MARKDOWN_PARSER.parse(commonmark));
-        Document document = Jsoup.parse(commonmark);
+        Document document = Jsoup.parse(html);
         RstNodeVisitor visitor = new RstNodeVisitor();
         document.body().traverse(visitor);
         return "\n" + visitor;
@@ -169,4 +172,4 @@ public String toString() {
             return sb.toString();
         }
     }
-}
+}