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

@@ -934,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

@@ -90,21 +90,19 @@ check_python_version() {
 check_and_install_packages() {
     print_info "Checking required Python packages..."
 
-    # List of required packages (import_name:package_name pairs)
-    required_packages=(
-        "typer:typer"
-        "rich:rich"
-        "boto3:boto3"
-        "yaml:PyYAML"
+    # 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_pair in "${required_packages[@]}"; do
-        import_name="${package_pair%%:*}"
-        package_name="${package_pair##*:}"
+    for import_name in "${!required_packages[@]}"; do
         if ! $PYTHON_CMD -c "import $import_name" >/dev/null 2>&1; then
-            missing_packages+=("$package_name")
+            missing_packages+=("${required_packages[$import_name]}")
         fi
     done