[pipeline] optimization for packaging logic (#36169)

* optimization

* fix

Author: msyyc

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

@@ -1,21 +1,14 @@
-import glob
-import os
+from pathlib import Path
 
 from subprocess import check_call
 
 from .change_log import main as change_log_main
 
 DEFAULT_DEST_FOLDER = "./dist"
 
-
-def create_package(name, dest_folder=DEFAULT_DEST_FOLDER):
-    # a package will exist in either one, or the other folder. this is why we can resolve both at the same time.
-    absdirs = [
-        os.path.dirname(package)
-        for package in (glob.glob("{}/setup.py".format(name)) + glob.glob("sdk/*/{}/setup.py".format(name)))
-    ]
-
-    absdirpath = os.path.abspath(absdirs[0])
+# prefolder: "sdk/compute"; name: "azure-mgmt-compute"
+def create_package(prefolder, name, dest_folder=DEFAULT_DEST_FOLDER):
+    absdirpath = Path(prefolder, name).absolute()
     check_call(["python", "setup.py", "bdist_wheel", "-d", dest_folder], cwd=absdirpath)
     check_call(
         ["python", "setup.py", "sdist", "--format", "zip", "-d", dest_folder],

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

@@ -19,6 +19,7 @@ def main(generate_input, generate_output):
     result = {"packages": []}
     for package in data.values():
         package_name = package["packageName"]
+        prefolder = package["path"][0]
         # Changelog
         last_version = ["first release"]
         if "azure-mgmt-" in package_name:
@@ -37,7 +38,7 @@ def main(generate_input, generate_output):
 
         _LOGGER.info(f"[PACKAGE]({package_name})[CHANGELOG]:{md_output}")
         # Built package
-        create_package(package_name)
+        create_package(prefolder, package_name)
         folder_name = package["path"][0]
         dist_path = Path(sdk_folder, folder_name, package_name, "dist")
         package["artifacts"] = [str(dist_path / package_file) for package_file in os.listdir(dist_path)]