update accession of contextForm from metadata for migrator

microbiomedata · Oct 8, 2024 · ab7eb47 · ab7eb47
1 parent 7f2ec78
commit ab7eb47
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nmdc_server/migrations/versions/988105f6581d_update_datasetDoi_to_awardDoi.py b/nmdc_server/migrations/versions/988105f6581d_update_datasetDoi_to_awardDoi.py
@@ -52,7 +52,7 @@ def upgrade():
     mappings = []
     for submission_metadata in session.query(SubmissionMetadata):
         metadata_submission = submission_metadata.metadata_submission
-        context_form = metadata_submission["contextForm"]
+        context_form = metadata_submission.get("contextForm")
 
         if not context_form.get("awardDoi", None) and context_form.get("datasetDoi", None):
             holder = context_form["datasetDoi"]
@@ -70,7 +70,7 @@ def downgrade():
     mappings = []
     for submission_metadata in session.query(SubmissionMetadata):
         metadata_submission = submission_metadata.metadata_submission
-        context_form = metadata_submission["contextForm"]
+        context_form = metadata_submission.get("contextForm")
 
         if not context_form.get("datasetDoi", None) and context_form.get("awardDoi", None):
             holder = context_form["awardDoi"]