Skip to content

Commit c121ae3

Browse files
authored
1 parent c87bab1 commit c121ae3

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

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

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,19 +24,23 @@ def main(generate_input, generate_output):
2424
data = json.load(reader)
2525
_LOGGER.info(f"auto_package input: {data}")
2626

27-
spec_folder = re.sub("specification", "", data["specFolder"])
2827
sdk_folder = "."
2928
result = {}
3029
python_tag = data.get("python_tag")
3130
package_total = set()
3231

32+
# make sure spec_folder like: "../azure-rest-api-specs"
33+
spec_folder = data["specFolder"]
3334
input_readme = data["relatedReadmeMdFile"]
34-
relative_path_readme = str(Path(spec_folder, input_readme))
35+
if "specification" in spec_folder:
36+
spec_folder = str(Path(spec_folder.split("specification")[0]))
37+
if "specification" not in input_readme:
38+
input_readme = str(Path("specification") / input_readme)
3539
_LOGGER.info(f"[CODEGEN]({input_readme})codegen begin")
3640

3741
if "resource-manager" in input_readme:
3842
config = generate(
39-
CONFIG_FILE, sdk_folder, [], relative_path_readme, spec_folder, force_generation=True, python_tag=python_tag
43+
CONFIG_FILE, sdk_folder, [], input_readme, spec_folder, force_generation=True, python_tag=python_tag
4044
)
4145
else:
4246
config = gen_dpg(input_readme, data.get("autorestConfig", ""), spec_folder)

0 commit comments

Comments
 (0)