Update PyYAML dependency handling across publish scripts and docs

Author: Taniya Mathur

docs/deployment.md

@@ -26,7 +26,7 @@ You need to have the following packages installed on your computer:
 3. [sam (AWS SAM)](https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/install-sam-cli.html)
 4. python 3.11 or later
 5. A local Docker daemon
-6. Python packages for publish.py: `pip install boto3 typer rich botocore setuptools`
+6. Python packages for publish.py: `pip install boto3 rich PyYAML botocore setuptools`
 
 For guidance on setting up a development environment, see: [Development Environment Setup Guide on Linux](./setup-development-env-linux.md) or [Development Environment Setup Guide on MacOS](./setup-development-env-macos.md)
 

publish.py

@@ -22,6 +22,7 @@
 from urllib.parse import quote
 
 import boto3
+import yaml
 from botocore.exceptions import ClientError
 from rich.console import Console
 from rich.progress import (
@@ -933,15 +934,6 @@ def _check_requirements_has_idp_common_pkg(self, func_dir):
     def _extract_function_name(self, dir_name, template_path):
         """Extract CloudFormation function name from template by matching CodeUri."""
         try:
-            try:
-                import yaml
-            except ImportError:
-                self.console.print("[yellow]PyYAML not found, installing...[/yellow]")
-                subprocess.run(
-                    [sys.executable, "-m", "pip", "install", "PyYAML"], check=True
-                )
-                import yaml
-
             # Create a custom loader that ignores CloudFormation intrinsic functions
             class CFLoader(yaml.SafeLoader):
                 pass

publish.sh

@@ -88,14 +88,19 @@ check_python_version() {
 check_and_install_packages() {
     print_info "Checking required Python packages..."
 
-    # List of required packages
-    required_packages=("typer" "rich" "boto3")
+    # List of required packages (import_name:package_name)
+    declare -A required_packages=(
+        ["typer"]="typer"
+        ["rich"]="rich" 
+        ["boto3"]="boto3"
+        ["yaml"]="PyYAML"
+    )
     missing_packages=()
 
     # Check each package
-    for package in "${required_packages[@]}"; do
-        if ! $PYTHON_CMD -c "import $package" >/dev/null 2>&1; then
-            missing_packages+=("$package")
+    for import_name in "${!required_packages[@]}"; do
+        if ! $PYTHON_CMD -c "import $import_name" >/dev/null 2>&1; then
+            missing_packages+=("${required_packages[$import_name]}")
         fi
     done