Refactoring

Author: tisnik

scripts/gen_doc.py

@@ -9,20 +9,18 @@
 DIRECTORIES = ["src", "tests/unit", "tests/integration", "tests/e2e"]
 
 # README.md files won't be generated in following directories
-DIRS_TO_SKIP = [
+DIRS_TO_SKIP = {
     "tests/integration",
     "tests/e2e",
     "tests/e2e/configuration",
-    "tests/e2e",
-    "tests/e2e/configuration",
     "tests/e2e/rag",
     "tests/e2e/secrets",
     "tests/e2e/skills",
     "tests/e2e/skills/echo",
     "tests/e2e/skills/echo/references",
     "tests/e2e/skills/summarize",
     "tests/e2e/skills/summarize/references",
-]
+}
 
 
 def generate_docfile(directory: Path) -> None:
@@ -70,15 +68,18 @@ def generate_documentation_on_path(path: Path) -> None:
         path (str or os.PathLike): Directory in which to generate the README.md file.
     """
     directory = path
+
+    # directory can be skipped if it's part of DIRS_TO_SKIP global list.
+    if path.as_posix() in DIRS_TO_SKIP:
+        print(f"[gendoc] Skipping {path}")
+        return
+
     cwd = os.getcwd()
-    os.chdir(directory)
+    os.chdir(path)
 
     try:
-        if str(directory) in DIRS_TO_SKIP:
-            print(f"[gendoc] Skipping {directory}")
-        else:
-            print(f"[gendoc] Generating README.md in: {directory}")
-            generate_docfile(directory)
+        print(f"[gendoc] Generating README.md in: {path}")
+        generate_docfile(path)
     finally:
         os.chdir(cwd)