@@ -627,7 +627,7 @@ def run(
627
627
return migration_run_data ["migrate_return_value" ], ljpr
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" ):
@@ -886,7 +886,9 @@ def _is_migrator_done(_mg_start, good_prs, time_per, pr_limit):
886
886
return False
887
887
888
888
889
- def _run_migrator (migrator , mctx , temp , time_per , git_backend : GitPlatformBackend ):
889
+ def _run_migrator (
890
+ migrator , mctx , temp , time_per , dry_run , git_backend : GitPlatformBackend
891
+ ):
890
892
_mg_start = time .time ()
891
893
892
894
migrator_name = get_migrator_name (migrator )
@@ -1028,7 +1030,8 @@ def _run_migrator(migrator, mctx, temp, time_per, git_backend: GitPlatformBacken
1028
1030
os .chdir (BOT_HOME_DIR )
1029
1031
1030
1032
# Write graph partially through
1031
- dump_graph (mctx .graph )
1033
+ if not dry_run :
1034
+ dump_graph (mctx .graph )
1032
1035
1033
1036
with filter_reprinted_lines ("rm-tmp" ):
1034
1037
for f in glob .glob ("/tmp/*" ):
@@ -1228,6 +1231,7 @@ def main(ctx: CliContext) -> None:
1228
1231
graph = gx ,
1229
1232
smithy_version = smithy_version ,
1230
1233
pinning_version = pinning_version ,
1234
+ dry_run = ctx .dry_run ,
1231
1235
)
1232
1236
migrators = load_migrators ()
1233
1237
@@ -1239,6 +1243,7 @@ def main(ctx: CliContext) -> None:
1239
1243
time_per_migrator ,
1240
1244
tot_time_per_migrator ,
1241
1245
) = _compute_time_per_migrator (
1246
+ mctx ,
1242
1247
migrators ,
1243
1248
)
1244
1249
for i , migrator in enumerate (migrators ):
@@ -1262,7 +1267,7 @@ def main(ctx: CliContext) -> None:
1262
1267
1263
1268
for mg_ind , migrator in enumerate (migrators ):
1264
1269
good_prs = _run_migrator (
1265
- migrator , mctx , temp , time_per_migrator [mg_ind ], git_backend
1270
+ migrator , mctx , temp , time_per_migrator [mg_ind ], ctx . dry_run , git_backend
1266
1271
)
1267
1272
if good_prs > 0 :
1268
1273
pass
0 commit comments