refactor hooks

Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>

Author: sirutBuasai

docs/src/hooks.py

@@ -16,35 +16,23 @@
     Add to mkdocs.yaml: hooks: [docs/src/hooks.py]
 """
 
-import logging
 import os
-import subprocess
 
 from generate import generate_all
-from utils import load_yaml
-
-logging.basicConfig(level=logging.INFO, format="%(levelname)s: %(message)s")
-LOGGER = logging.getLogger(__name__)
+from utils import clone_git_repository, load_yaml
 
 # Resolve paths relative to this file
 SRC_DIR = os.path.dirname(os.path.abspath(__file__))
 DOCS_DIR = os.path.dirname(SRC_DIR)
 DATA_FILE = os.path.join(SRC_DIR, "data", "images.yml")
-
-
-def clone_tutorials():
-    """Clone sample tutorials repository into docs/tutorials."""
-    tutorials_dir = os.path.join(DOCS_DIR, "tutorials")
-    if os.path.exists(tutorials_dir):
-        return
-
-    repo_url = "https://github.com/aws-samples/sample-aws-deep-learning-containers"
-    subprocess.run(["git", "clone", "--depth", "1", repo_url, tutorials_dir], check=True)
+TUTORIALS_DIR = os.path.join(DOCS_DIR, "tutorials")
 
 
 # MkDocs hook entry point
 def on_startup(command=["build", "gh-deploy", "serve"], dirty=False):
     """MkDocs hook - runs before build."""
     yaml_data = load_yaml(DATA_FILE)
-    clone_tutorials()
+    tutorials_repo = "https://github.com/aws-samples/sample-aws-deep-learning-containers"
+
+    clone_git_repository(tutorials_repo, TUTORIALS_DIR)
     generate_all(yaml_data, dry_run=False)

docs/src/main.py

@@ -1,15 +1,3 @@
-# Copyright 2018-2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License"). You
-# may not use this file except in compliance with the License. A copy of
-# the License is located at
-#
-#     http://aws.amazon.com/apache2.0/
-#
-# or in the "license" file accompanying this file. This file is
-# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
-# ANY KIND, either express or implied. See the License for the specific
-# language governing permissions and limitations under the License.
 """Documentation generation entry point.
 
 Usage:
@@ -19,11 +7,22 @@
 import argparse
 import logging
 import os
+import sys
 
 from generate import generate_all, generate_available_images, generate_support_policy
+from logger import ColoredFormatter
 from utils import load_yaml
 
-logging.basicConfig(level=logging.INFO, format="%(levelname)s: %(message)s")
+# Configure root logger - all child loggers inherit this
+root_logger = logging.getLogger()
+root_logger.setLevel(logging.INFO)
+
+console_handler = logging.StreamHandler(sys.stdout)
+console_handler.setFormatter(ColoredFormatter())
+console_handler.setLevel(logging.DEBUG)
+
+root_logger.addHandler(console_handler)
+
 LOGGER = logging.getLogger(__name__)
 
 # Resolve paths relative to this file
@@ -49,7 +48,7 @@ def main():
     yaml_data = load_yaml(DATA_FILE)
 
     if args.verbose:
-        logging.getLogger().setLevel(logging.DEBUG)
+        root_logger.setLevel(logging.DEBUG)
 
     LOGGER.info(f"Loaded data from {DATA_FILE}")
 

docs/src/utils.py

@@ -12,6 +12,9 @@
 # language governing permissions and limitations under the License.
 """Common utilities for documentation generation."""
 
+import os
+import subprocess
+
 import yaml
 
 
@@ -41,3 +44,11 @@ def write_output(path: str, content: str) -> None:
     """Write generated markdown to file."""
     with open(path, "w") as f:
         f.write(content)
+
+
+def clone_git_repository(git_repository: str, target_dir: str) -> None:
+    """Clone sample tutorials repository into docs/tutorials."""
+    if os.path.exists(target_dir):
+        return
+
+    subprocess.run(["git", "clone", "--depth", "1", git_repository, target_dir], check=True)