add iframe_id

blueswen · Mar 25, 2024 · 28434d3 · 28434d3
1 parent 92add9c
commit 28434d3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/mkdocs_swagger_ui_tag/plugin.py b/mkdocs_swagger_ui_tag/plugin.py
@@ -165,6 +165,7 @@ def render_template(openapi_spec_url, swagger_ui_ele):
                 with open(os.path.join(page_dir, iframe_filename), "w") as f:
                     f.write(template_output)
                 self.replace_with_iframe(soup, swagger_ui_ele, cur_id, iframe_filename)
+                return cur_id
 
             for swagger_ui_ele in swagger_ui_list:
                 if swagger_ui_ele.has_attr("grouped"):
@@ -174,8 +175,10 @@ def render_template(openapi_spec_url, swagger_ui_ele):
                 openapi_spec_url = self.path_to_url(
                     page.file, swagger_ui_ele.get("src", "")
                 )
-                render_template(
-                    openapi_spec_url=openapi_spec_url, swagger_ui_ele=swagger_ui_ele
+                iframe_id_list.append(
+                    render_template(
+                        openapi_spec_url=openapi_spec_url, swagger_ui_ele=swagger_ui_ele
+                    )
                 )
 
             if grouped_list: