Skip to content

Commit ef60033

Browse files
committed
try to insert testdata before data
1 parent 5625320 commit ef60033

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

bin/upgrade.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -253,10 +253,16 @@ def add_args(new_data: dict[str, Any]) -> bool:
253253
if generators_path.exists():
254254
generators_data = read_yaml(generators_path)
255255
assert generators_data is not None
256-
assert isinstance(generators_data, dict)
256+
assert isinstance(generators_data, CommentedMap)
257257

258258
if "testdata.yaml" not in generators_data:
259-
generators_data["testdata.yaml"] = CommentedMap()
259+
if "data" in generators_data:
260+
# insert before data
261+
pos = [*generators_data.keys()].index("data")
262+
generators_data.insert(pos, "testdata.yaml", CommentedMap())
263+
else:
264+
# insert at end
265+
generators_data["testdata.yaml"] = CommentedMap()
260266
if add_args(generators_data["testdata.yaml"]):
261267
write_yaml(generators_data, generators_path)
262268
else:

0 commit comments

Comments
 (0)