Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 19 additions & 8 deletions data_models/migrations/0055_auto_20230613_1348.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,15 +26,26 @@ def convert_cmr_data_to_json(apps, schema_editor):
d = getattr(doi, field)
if d is None or d == "":
logger.info(f"Field {field} is empty, setting to null")
json_data = "null"
json_data = None
else:
try:
json_data = json.loads(d)
except json.decoder.JSONDecodeError:
logger.info(
f"Failed to parse JSON, attempting to convert to JSON from serialized python object {d}"
)
json_data = ast.literal_eval(d)
if isinstance(d, str):
try:
json_data = json.loads(d)
except json.decoder.JSONDecodeError:
logger.info(
f"Failed to parse JSON on {doi.uuid} field {field}, attempting to convert to JSON from serialized python object {d}"
)
try:
if field == "cmr_data_formats" and "[" not in d:
d = f'["{d}"]'
decoded_string = d.encode("utf-8").decode('unicode_escape')
json_data = ast.literal_eval(decoded_string)
except Exception as err:
logger.error(f"Failed to convert {field} on {doi.uuid} to JSON: {err}")
logger.error(f"Value: {decoded_string}")
raise err
else:
json_data = d
setattr(doi, field, json.dumps(json_data))
doi.save()

Expand Down
60 changes: 60 additions & 0 deletions data_models/migrations/0057_auto_20230705_1319.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
# Generated by Django 4.1.5 on 2023-06-13 18:48

import ast
import json
import logging
from django.db import migrations

logger = logging.getLogger(__name__)

fields_to_convert = [
"cmr_projects",
"cmr_dates",
"cmr_plats_and_insts",
"cmr_science_keywords",
"cmr_data_formats",
]


def convert_cmr_data_to_json(apps, schema_editor):
DOI = apps.get_model("data_models", "DOI")

dois = DOI.objects.all()
for doi in dois:
for field in fields_to_convert:
logger.info(f"Converting {field} on {doi.uuid} to JSON")
d = getattr(doi, field)
if d is None or d == "":
logger.info(f"Field {field} is empty, setting to null")
json_data = None
else:
if isinstance(d, str):
try:
json_data = json.loads(d)
except json.decoder.JSONDecodeError:
logger.info(
f"Failed to parse JSON on {doi.uuid} field {field}, attempting to convert to JSON from serialized python object {d}"
)
try:
if field == "cmr_data_formats" and "[" not in d:
d = f'["{d}"]'
decoded_string = d.encode("utf-8").decode('unicode_escape')
json_data = ast.literal_eval(decoded_string)
except Exception as err:
logger.error(f"Failed to convert {field} on {doi.uuid} to JSON: {err}")
logger.error(f"Value: {decoded_string}")
raise err
else:
json_data = d
setattr(doi, field, json_data)
doi.save()


class Migration(migrations.Migration):
dependencies = [
('data_models', '0056_alter_doi_cmr_data_formats_alter_doi_cmr_dates_and_more'),
]

operations = [
migrations.RunPython(convert_cmr_data_to_json, reverse_code=migrations.RunPython.noop)
]