@@ -220,13 +220,25 @@ def main(generate_input, generate_output):
220
220
result = {}
221
221
python_tag = data .get ("python_tag" )
222
222
package_total = set ()
223
- readme_and_tsp = data .get ("relatedReadmeMdFiles" , []) + data .get ("relatedTypeSpecProjectFolder" , [])
223
+ readme_and_tsp = [("relatedReadmeMdFiles" , item ) for item in data .get ("relatedReadmeMdFiles" , [])] + [
224
+ ("relatedTypeSpecProjectFolder" , item ) for item in data .get ("relatedTypeSpecProjectFolder" , [])
225
+ ]
224
226
run_in_pipeline = data .get ("runMode" ) is not None
225
- for readme_or_tsp in readme_and_tsp :
227
+ for input_type , readme_or_tsp in readme_and_tsp :
226
228
_LOGGER .info (f"[CODEGEN]({ readme_or_tsp } )codegen begin" )
227
229
try :
228
230
code_generation_start_time = time .time ()
229
- if "resource-manager" in readme_or_tsp :
231
+ if input_type == "relatedTypeSpecProjectFolder" :
232
+ del_outdated_generated_files (str (Path (spec_folder , readme_or_tsp )))
233
+ config = gen_typespec (
234
+ readme_or_tsp ,
235
+ spec_folder ,
236
+ data ["headSha" ],
237
+ data ["repoHttpsUrl" ],
238
+ run_in_pipeline ,
239
+ data .get ("apiVersion" ),
240
+ )
241
+ elif "resource-manager" in readme_or_tsp :
230
242
relative_path_readme = str (Path (spec_folder , readme_or_tsp ))
231
243
del_outdated_files (relative_path_readme )
232
244
generate_mgmt = partial (
@@ -242,18 +254,8 @@ def main(generate_input, generate_output):
242
254
config = generate_mgmt ()
243
255
if need_regen_for_multiapi_package (spec_folder , readme_or_tsp ):
244
256
generate_mgmt ()
245
- elif "data-plane" in readme_or_tsp :
246
- config = gen_dpg (readme_or_tsp , data .get ("autorestConfig" , "" ), dpg_relative_folder (spec_folder ))
247
257
else :
248
- del_outdated_generated_files (str (Path (spec_folder , readme_or_tsp )))
249
- config = gen_typespec (
250
- readme_or_tsp ,
251
- spec_folder ,
252
- data ["headSha" ],
253
- data ["repoHttpsUrl" ],
254
- run_in_pipeline ,
255
- data .get ("apiVersion" ),
256
- )
258
+ config = gen_dpg (readme_or_tsp , data .get ("autorestConfig" , "" ), dpg_relative_folder (spec_folder ))
257
259
_LOGGER .info (f"code generation cost time: { int (time .time () - code_generation_start_time )} seconds" )
258
260
except Exception as e :
259
261
_LOGGER .error (f"fail to generate sdk for { readme_or_tsp } : { str (e )} " )
0 commit comments