[sdk generation pipeline] find client name in __all__ from __init__.py (#41687)

* find client in __all__ of __init__.py

* update

* update

* update log

Author: msyyc

tools/azure-sdk-tools/packaging_tools/package_utils.py

@@ -1,4 +1,5 @@
 import os
+import ast
 import time
 import importlib
 from typing import Optional, Tuple, Dict, Any, List
@@ -174,20 +175,47 @@ def next_version(self) -> str:
             self._next_version = self.calculate_next_version()
         return self._next_version
 
+    @property
+    def extract_client_title_from_init(self) -> str:
+        """
+        Extract the client title from a package's __init__.py file.
+
+        Args:
+            package_name (str): The package name (e.g., "azure-mgmt-compute")
+
+        Returns:
+            str: The client title if found, empty string otherwise
+        """
+        init_file = Path(self.whole_package_name) / self.whole_package_name.replace("-", "/") / "__init__.py"
+        try:
+            with open(init_file, "r") as f:
+                content = f.read()
+            tree = ast.parse(content)
+            # Look for __all__ assignment
+            for node in ast.walk(tree):
+                if isinstance(node, ast.Assign):
+                    # Check if any target is __all__
+                    for target in node.targets:
+                        if isinstance(target, ast.Name) and target.id == "__all__" and isinstance(node.value, ast.List):
+                            # Extract the value
+                            for elt in node.value.elts:
+                                if isinstance(elt, ast.Constant) and elt.value.endswith("Client"):
+                                    return elt.value
+        except Exception as e:
+            _LOGGER.info(f"Failed to extract title from {init_file}: {e}")
+
+        return ""
+
     # Use the template to update readme and setup by packaging_tools
     @return_origin_path
     def check_file_with_packaging_tool(self):
-        module = importlib.import_module(self.whole_package_name.replace("-", "."))
-        title = ""
-        for item in getattr(module, "__all__", []):
-            if item.endswith("Client"):
-                title = item
-                break
+
+        os.chdir(Path(f"sdk/{self.sdk_folder}"))
+        title = self.extract_client_title_from_init
 
         if not title:
-            _LOGGER.info(f"Can not find the title in {self.whole_package_name}")
+            _LOGGER.info(f"Can not find the title for {self.whole_package_name}")
 
-        os.chdir(Path(f"sdk/{self.sdk_folder}"))
         # add `title` and update `is_stable` in sdk_packaging.toml
         toml = Path(self.whole_package_name) / "sdk_packaging.toml"
         stable_config = f'is_stable = {"true" if self.tag_is_stable and self.next_version != "1.0.0b1" else "false"}\n'
@@ -338,14 +366,14 @@ def check_pyproject_toml(self):
         os.chdir(Path("sdk") / self.sdk_folder / self.whole_package_name)
         # Configure and ensure pyproject.toml exists with required settings
         toml_path = Path("pyproject.toml")
-        
+
         # Default configurations to enforce
         default_configs = {
             "breaking": "false",
             "pyright": "false",
             "mypy": "false",
         }
-        
+
         # Create new pyproject.toml if it doesn't exist
         if not toml_path.exists():
             with open(toml_path, "w") as file:
@@ -354,27 +382,27 @@ def check_pyproject_toml(self):
                     file.write(f"{key} = {value}\n")
                 _LOGGER.info("Created pyproject.toml with default configurations")
             return
-            
+
         # If file exists, ensure all required configurations are present
         def edit_toml(content: List[str]):
             # Track if we have the [tool.azure-sdk-build] section
             has_section = False
             config_exists = {key: False for key in default_configs}
-            
+
             # Check for existing configurations and section
             for i, line in enumerate(content):
                 if "[tool.azure-sdk-build]" in line:
                     has_section = True
-                
+
                 # Check for each configuration
                 for key in default_configs:
                     if f"{key} = " in line:
                         config_exists[key] = True
-            
+
             # Add section if it doesn't exist
             if not has_section:
                 content.append("\n[tool.azure-sdk-build]\n")
-                
+
             # Add missing configurations
             for key, value in default_configs.items():
                 if not config_exists[key]:

tools/azure-sdk-tools/packaging_tools/sdk_package.py

@@ -51,9 +51,9 @@ def main(generate_input, generate_output):
         try:
             md_output = execute_func_with_timeout(change_log_func)
         except multiprocessing.TimeoutError:
-            md_output = "change log generation was timeout!!!"
+            md_output = "change log generation was timeout!!! You need to write it manually!!!"
         except:
-            md_output = "change log generation failed!!!"
+            md_output = "change log generation failed!!! You need to write it manually!!!"
         finally:
             for file in ["stable.json", "current.json"]:
                 file_path = Path(sdk_folder, prefolder, package_name, file)