mgmt, minor improvement on automation script for local run (Azure#40475)

Author: weidongxu-microsoft

eng/mgmt/automation/generate_utils.py

@@ -100,13 +100,13 @@ def remove_generated_source_code(sdk_folder: str, namespace: str):
     test_folder = os.path.join(sdk_folder, "src/test/java", namespace.replace(".", "/"), "generated")
     sample_folder = os.path.join(sdk_folder, "src/samples/java", namespace.replace(".", "/"), "generated")
 
-    logging.info(f'Removing main source folder: {main_folder}')
+    logging.info(f"Removing main source folder: {main_folder}")
     shutil.rmtree(main_folder, ignore_errors=True)
 
-    logging.info(f'Removing generated test folder: {test_folder}')
+    logging.info(f"Removing generated test folder: {test_folder}")
     shutil.rmtree(test_folder, ignore_errors=True)
 
-    logging.info(f'Removing generated samples folder: {sample_folder}')
+    logging.info(f"Removing generated samples folder: {sample_folder}")
     shutil.rmtree(sample_folder, ignore_errors=True)
 
 
@@ -367,7 +367,7 @@ def generate_typespec_project(
             ]
 
         if tspconfig_valid:
-            check_call(tsp_cmd, sdk_root)
+            check_call(tsp_cmd, sdk_root, shell=True)
 
             sdk_folder = find_sdk_folder(sdk_root)
             logging.info("SDK folder: " + sdk_folder)
@@ -393,7 +393,7 @@ def generate_typespec_project(
                     drop_changes(sdk_root)
                     remove_generated_source_code(sdk_folder, f"{group_id}.{service}")
                     # regenerate
-                    check_call(tsp_cmd, sdk_root)
+                    check_call(tsp_cmd, sdk_root, shell=True)
                 succeeded = True
     except subprocess.CalledProcessError as error:
         error_message = (
@@ -406,9 +406,9 @@ def generate_typespec_project(
     return succeeded, require_sdk_integration, sdk_folder, service, module
 
 
-def check_call(cmd: List[str], work_dir: str):
+def check_call(cmd: List[str], work_dir: str, shell: bool = False):
     logging.info("Command line: " + " ".join(cmd))
-    subprocess.check_call(cmd, cwd=work_dir)
+    subprocess.check_call(cmd, cwd=work_dir, shell=shell)
 
 
 def drop_changes(work_dir: str):

eng/mgmt/automation/utils.py

@@ -177,19 +177,30 @@ def update_service_ci_and_pom(sdk_root: str, service: str, group: str, module: s
 
 
 def update_version(sdk_root: str, output_folder: str):
+    # find the python command
+    python_cmd = "python"
+    try:
+        subprocess.check_call([python_cmd, "--version"], shell=True)
+    except subprocess.CalledProcessError:
+        python_cmd = "python3"
+        try:
+            subprocess.check_call([python_cmd, "--version"], shell=True)
+        except subprocess.CalledProcessError:
+            raise Exception("python or python3 not found")
+
     pwd = os.getcwd()
     try:
         os.chdir(sdk_root)
         print(os.getcwd())
         subprocess.run(
-            "python3 eng/versioning/update_versions.py --ut library --bt client --sr",
+            "{0} eng/versioning/update_versions.py --ut library --bt client --sr".format(python_cmd),
             stdout=subprocess.DEVNULL,
             stderr=sys.stderr,
             shell=True,
         )
         subprocess.run(
-            "python3 eng/versioning/update_versions.py --ut library --bt client --tf {0}/README.md".format(
-                output_folder
+            "{0} eng/versioning/update_versions.py --ut library --bt client --tf {1}/README.md".format(
+                python_cmd, output_folder
             ),
             stdout=subprocess.DEVNULL,
             stderr=sys.stderr,