diff --git a/mkdocs_monorepo_plugin/edit_uri.py b/mkdocs_monorepo_plugin/edit_uri.py index 8845ec3..24adc5a 100644 --- a/mkdocs_monorepo_plugin/edit_uri.py +++ b/mkdocs_monorepo_plugin/edit_uri.py @@ -32,11 +32,12 @@ def __get_root_docs_dir(self): def __get_page_dir_alias(self): parts = self.page.url.split("/") - while True: + while parts: parts.pop() - alias = path.join(*parts) + alias = path.join(*parts) if parts else "" if alias in self.plugin.aliases: return alias + return None def __get_page_docs_dir(self): alias = self.__get_page_dir_alias() @@ -102,6 +103,8 @@ def __is_root(self): def build(self): if self.__is_root(): return self.page.edit_url + if self.__get_page_dir_alias() is None: + return self.page.edit_url if self.__has_repo(): config = self.__get_page_config_file_yaml() return config["repo_url"] + config["edit_uri"] + self.__get_page_src_path()