migrate some functions from auto_release to sdk generation pipeline (Azure#40177)

* update pipeline

* format

* update for target release date

* update

* update

* update

* Update tools/azure-sdk-tools/packaging_tools/sdk_generator.py

Co-authored-by: Copilot <[email protected]>

---------

Co-authored-by: Copilot <[email protected]>

Author: msyyc

scripts/auto_release/main.py

@@ -12,7 +12,6 @@
 from ghapi.all import GhApi
 from github import Github
 from datetime import datetime, timedelta
-import importlib
 
 _LOG = logging.getLogger()
 
@@ -47,40 +46,6 @@ def print_check(cmd):
     subprocess.check_call(cmd, shell=True)
 
 
-def preview_version_plus(preview_label: str, last_version: str) -> str:
-    num = last_version.split(preview_label)
-    num[1] = str(int(num[1]) + 1)
-    return f"{num[0]}{preview_label}{num[1]}"
-
-
-def stable_version_plus(changelog: str, last_version: str):
-    flag = [False, False, False]  # breaking, feature, bugfix
-    flag[0] = "### Breaking Changes" in changelog
-    flag[1] = "### Features Added" in changelog
-    flag[2] = "### Bugs Fixed" in changelog
-
-    num = last_version.split(".")
-    if flag[0]:
-        return f"{int(num[0]) + 1}.0.0"
-    elif flag[1]:
-        return f"{num[0]}.{int(num[1]) + 1}.0"
-    elif flag[2]:
-        return f"{num[0]}.{num[1]}.{int(num[2]) + 1}"
-    else:
-        return "0.0.0"
-
-
-# find all the files of one folder, including files in subdirectory
-def all_files(path: str, files: List[str]):
-    all_folder = os.listdir(path)
-    for item in all_folder:
-        folder = str(Path(f"{path}/{item}"))
-        if os.path.isdir(folder):
-            all_files(folder, files)
-        else:
-            files.append(folder)
-
-
 def modify_file(file_path: str, func: Any):
     with open(file_path, "r") as file_in:
         content = file_in.readlines()
@@ -182,6 +147,8 @@ def generate_code(self):
             "repoHttpsUrl": "https://github.com/Azure/azure-rest-api-specs",
             "specFolder": self.spec_repo,
             file_name: [self.readme_local_folder()],
+            "targetReleaseDate": self.target_release_date,
+            "allowInvalidNextVersion": True,
         }
         log(str(input_data))
 
@@ -218,16 +185,6 @@ def create_new_branch(self):
         self.new_branch = f"t2-{self.package_name}-{current_time()}-{str(time.time())[-5:]}"
         print_check(f"git checkout -b {self.new_branch}")
 
-    def check_sdk_readme(self):
-        sdk_readme = str(Path(f"sdk/{self.sdk_folder}/{self.whole_package_name}/README.md"))
-
-        def edit_sdk_readme(content: List[str]):
-            for i in range(0, len(content)):
-                if content[i].find("MyService") > 0:
-                    content[i] = content[i].replace("MyService", self.package_name.capitalize())
-
-        modify_file(sdk_readme, edit_sdk_readme)
-
     @property
     def readme_md_path(self) -> Path:
         return Path(self.spec_repo) / "specification" / self.spec_readme.split("specification/")[-1]
@@ -236,89 +193,6 @@ def readme_md_path(self) -> Path:
     def readme_python_md_path(self) -> Path:
         return Path(str(self.readme_md_path).replace("readme.md", "readme.python.md"))
 
-    # Use the template to update readme and setup by packaging_tools
-    @return_origin_path
-    def check_file_with_packaging_tool(self):
-        print_check(f"pip install {self.get_whl_package} --force-reinstall")
-        module = importlib.import_module(self.whole_package_name.replace("-", "."))
-        title = ""
-        for item in getattr(module, "__all__", []):
-            if item.endswith("Client"):
-                title = item
-                break
-
-        if not title:
-            log(f"Can not find the title in {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 = "is_stable = " + ("true" if self.tag_is_stable else "false") + "\n"
-        if toml.exists():
-
-            def edit_toml(content: List[str]):
-                has_title = False
-                has_isstable = False
-                for idx in range(len(content)):
-                    if "title" in content[idx]:
-                        has_title = True
-                    if "is_stable" in content[idx]:
-                        has_isstable = True
-                        content[idx] = stable_config
-                if not has_title:
-                    content.append(f'title = "{title}"\n')
-                if not has_isstable:
-                    content.append(stable_config)
-
-            modify_file(str(toml), edit_toml)
-        else:
-            log(f"{os.getcwd()}/{toml} does not exist")
-
-        print_check(f"python -m packaging_tools --build-conf {self.whole_package_name}")
-        log("packaging_tools --build-conf successfully ")
-
-    def check_pprint_name(self):
-        pprint_name = self.package_name.capitalize()
-
-        def edit_file_for_pprint_name(content: List[str]):
-            for i in range(0, len(content)):
-                content[i] = content[i].replace("MyService", pprint_name)
-
-        for file in os.listdir(self.sdk_code_path()):
-            file_path = str(Path(self.sdk_code_path()) / file)
-            if os.path.isfile(file_path):
-                modify_file(file_path, edit_file_for_pprint_name)
-        log(f' replace "MyService" with "{pprint_name}" successfully ')
-
-    def get_all_files_under_package_folder(self) -> List[str]:
-        files = []
-        all_files(self.sdk_code_path(), files)
-        return files
-
-    def calculate_next_version_proc(self, last_version: str):
-        preview_tag = not self.tag_is_stable
-        changelog = self.get_changelog()
-        if changelog == "":
-            self.version_suggestion = "(it should be stable)" if self.tag_is_stable else "(it should be perview)"
-            return "0.0.0"
-        preview_version = "rc" in last_version or "b" in last_version
-        #                                           |   preview tag                     | stable tag
-        # preview version(1.0.0rc1/1.0.0b1)         | 1.0.0rc2(track1)/1.0.0b2(track2)  |  1.0.0
-        # stable  version (1.0.0) (breaking change) | 2.0.0rc1(track1)/2.0.0b1(track2)  |  2.0.0
-        # stable  version (1.0.0) (feature)         | 1.1.0rc1(track1)/1.1.0b1(track2)  |  1.1.0
-        # stable  version (1.0.0) (bugfix)          | 1.0.1rc1(track1)/1.0.1b1(track2)  |  1.0.1
-        preview_label = "b"
-        if preview_version and preview_tag:
-            next_version = preview_version_plus(preview_label, last_version)
-        elif preview_version and not preview_tag:
-            next_version = last_version.split(preview_label)[0]
-        elif not preview_version and preview_tag:
-            next_version = stable_version_plus(changelog, last_version) + preview_label + "1"
-        else:
-            next_version = stable_version_plus(changelog, last_version)
-
-        return next_version
-
     def get_autorest_result(self) -> Dict[Any, Any]:
         with open(self.autorest_result, "r") as file_in:
             content = json.load(file_in)
@@ -336,52 +210,6 @@ def get_last_release_version(self) -> str:
         except:
             return ""
 
-    def calculate_next_version(self):
-        last_version = self.get_last_release_version()
-        if last_version:
-            self.next_version = self.calculate_next_version_proc(last_version)
-        else:
-            self.next_version = "1.0.0b1"
-
-    def edit_all_version_file(self):
-        files = self.get_all_files_under_package_folder()
-
-        def edit_version_file(content: List[str]):
-            for i in range(0, len(content)):
-                if content[i].find("VERSION") > -1:
-                    content[i] = f'VERSION = "{self.next_version}"\n'
-                    break
-
-        for file in files:
-            if Path(file).name == "_version.py":
-                modify_file(file, edit_version_file)
-
-    def check_version(self):
-        self.calculate_next_version()
-        self.edit_all_version_file()
-
-    def check_changelog_file(self):
-        def edit_changelog_proc(content: List[str]):
-            next_version = self.next_version
-            content[1:1] = [
-                "\n",
-                f"## {next_version}{self.version_suggestion} ({self.target_release_date})\n\n",
-                self.get_changelog(),
-                "\n",
-            ]
-            if next_version == "1.0.0b1":
-                for _ in range(4):
-                    content.pop()
-
-        modify_file(str(Path(self.sdk_code_path()) / "CHANGELOG.md"), edit_changelog_proc)
-
-    def check_dev_requirement(self):
-        file = Path(f"sdk/{self.sdk_folder}/{self.whole_package_name}/dev_requirements.txt")
-        content = ["-e ../../../tools/azure-sdk-tools\n", "../../identity/azure-identity\n"]
-        if not file.exists():
-            with open(file, "w") as file_out:
-                file_out.writelines(content)
-
     def check_package_size(self):
         if self.after_multiapi_combiner:
             packages = self.get_private_package()
@@ -392,7 +220,10 @@ def check_package_size(self):
                     )
 
     def check_model_flatten(self):
-        if self.whole_package_name in ["azure-mgmt-mysqlflexibleservers", "azure-mgmt-postgresqlflexibleservers"]:
+        if self.whole_package_name in [
+            "azure-mgmt-mysqlflexibleservers",
+            "azure-mgmt-postgresqlflexibleservers",
+        ]:
             return
         if self.from_swagger:
             last_version = self.get_last_release_version()
@@ -416,38 +247,9 @@ def check_model_flatten(self):
                     issue.create_comment(f"@{assignee}, {message}")
                     raise Exception(message)
 
-    @return_origin_path
-    def check_pyproject_toml(self):
-        os.chdir(Path("sdk") / self.sdk_folder / self.whole_package_name)
-        # add `breaking = false` in pyproject.toml
-        toml = Path("pyproject.toml")
-        if not toml.exists():
-            with open(toml, "w") as file:
-                file.write("[tool.azure-sdk-build]\nbreaking = false\n")
-                _LOG.info("create pyproject.toml")
-
-        def edit_toml(content: List[str]):
-            has_breaking = False
-            for line in content:
-                if "breaking = false" in line:
-                    has_breaking = True
-                    break
-            if not has_breaking:
-                _LOG.info("add breaking = false to pyproject.toml")
-                content.append("breaking = false\n")
-
-        modify_file(str(toml), edit_toml)
-
     def check_file(self):
-        self.check_file_with_packaging_tool()
-        self.check_pprint_name()
-        self.check_sdk_readme()
-        self.check_version()
-        self.check_changelog_file()
-        self.check_dev_requirement()
         self.check_package_size()
         self.check_model_flatten()
-        self.check_pyproject_toml()
 
     def sdk_code_path(self) -> str:
         return str(Path(f"sdk/{self.sdk_folder}/{self.whole_package_name}"))
@@ -550,10 +352,6 @@ def get_private_package(self) -> List[str]:
         content = self.get_autorest_result()
         return content["packages"][0]["artifacts"]
 
-    @property
-    def get_whl_package(self) -> str:
-        return [package for package in self.get_private_package() if package.endswith(".whl")][0]
-
     def ask_check_policy(self):
         changelog = self.get_changelog()
         if changelog == "":