@@ -627,7 +627,7 @@ def run(
627
627
return migration_run_data ["migrate_return_value" ], pr_lazy_json
628
628
629
629
630
- def _compute_time_per_migrator (migrators ):
630
+ def _compute_time_per_migrator (mctx , migrators ):
631
631
# we weight each migrator by the number of available nodes to migrate
632
632
num_nodes = []
633
633
for migrator in tqdm .tqdm (migrators , ncols = 80 , desc = "computing time per migrator" ):
@@ -885,7 +885,9 @@ def _is_migrator_done(_mg_start, good_prs, time_per, pr_limit):
885
885
return False
886
886
887
887
888
- def _run_migrator (migrator , mctx , temp , time_per , git_backend : GitPlatformBackend ):
888
+ def _run_migrator (
889
+ migrator , mctx , temp , time_per , dry_run , git_backend : GitPlatformBackend
890
+ ):
889
891
_mg_start = time .time ()
890
892
891
893
migrator_name = get_migrator_name (migrator )
@@ -1026,7 +1028,8 @@ def _run_migrator(migrator, mctx, temp, time_per, git_backend: GitPlatformBacken
1026
1028
os .chdir (BOT_HOME_DIR )
1027
1029
1028
1030
# Write graph partially through
1029
- dump_graph (mctx .graph )
1031
+ if not dry_run :
1032
+ dump_graph (mctx .graph )
1030
1033
1031
1034
with filter_reprinted_lines ("rm-tmp" ):
1032
1035
for f in glob .glob ("/tmp/*" ):
@@ -1226,6 +1229,7 @@ def main(ctx: CliContext) -> None:
1226
1229
graph = gx ,
1227
1230
smithy_version = smithy_version ,
1228
1231
pinning_version = pinning_version ,
1232
+ dry_run = ctx .dry_run ,
1229
1233
)
1230
1234
migrators = load_migrators ()
1231
1235
@@ -1237,6 +1241,7 @@ def main(ctx: CliContext) -> None:
1237
1241
time_per_migrator ,
1238
1242
tot_time_per_migrator ,
1239
1243
) = _compute_time_per_migrator (
1244
+ mctx ,
1240
1245
migrators ,
1241
1246
)
1242
1247
for i , migrator in enumerate (migrators ):
@@ -1260,7 +1265,7 @@ def main(ctx: CliContext) -> None:
1260
1265
1261
1266
for mg_ind , migrator in enumerate (migrators ):
1262
1267
good_prs = _run_migrator (
1263
- migrator , mctx , temp , time_per_migrator [mg_ind ], git_backend
1268
+ migrator , mctx , temp , time_per_migrator [mg_ind ], ctx . dry_run , git_backend
1264
1269
)
1265
1270
if good_prs > 0 :
1266
1271
pass
0 commit comments