Skip to content

Commit 41f9c3d

Browse files
Resolved Copilot PR suggestions
1 parent 6597ce5 commit 41f9c3d

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

Deployment/data_migration/migrate.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -432,7 +432,11 @@ def import_search(export_dir: Path, endpoint: str) -> None:
432432

433433
schema_files = sorted(search_dir.glob("*_schema.json"))
434434
for schema_file in schema_files:
435-
index_name = schema_file.stem.replace("_schema", "")
435+
index_name = (
436+
schema_file.stem[: -len("_schema")]
437+
if schema_file.stem.endswith("_schema")
438+
else schema_file.stem
439+
)
436440
logger.info("Importing search index: %s", index_name)
437441
_import_search_index_schema(index_client, schema_file, index_name)
438442
docs_file = search_dir / f"{index_name}_documents.jsonl"
@@ -562,11 +566,11 @@ def _export_collection(db, collection_name: str, cosmos_dir: Path) -> None:
562566

563567
# Write a checksum sidecar for integrity verification
564568
_write_checksum(out_path, doc_count)
565-
except Exception:
569+
except Exception as exc:
566570
logger.exception(
567571
" Failed to export collection '%s'", collection_name
568572
)
569-
573+
raise
570574

571575
def _bson_to_serializable(doc: dict) -> dict:
572576
"""Recursively convert BSON types to JSON-serializable types."""

0 commit comments

Comments
 (0)