Source code for idmtools.utils.json

JSON utilities for idmtools such as encoders and decoders.

Copyright 2021, Bill & Melinda Gates Foundation. All rights reserved.
import json
from enum import Enum
from json import JSONEncoder
from logging import getLogger
from typing import List, Any, Dict, Union
from uuid import UUID
from idmtools.assets import AssetCollection, Asset
from idmtools.core import EntityStatus
from idmtools.entities import CommandLine
from idmtools.entities.experiment import Experiment
from idmtools.entities.itask import ITask
from idmtools.entities.simulation import Simulation
from idmtools.utils.entities import as_dict
from datetime import datetime

user_logger = getLogger('user')

[docs]class IDMJSONEncoder(JSONEncoder): """ IDMJSONEncoder handles encoding IDM specific items. """
[docs] def default(self, o): """ JSON Encode item. Args: o: Object to encode Returns: JSON encoded object """ if isinstance(o, Enum): return o.value elif isinstance(o, EntityStatus): return o.value elif isinstance(o, Experiment): return o.to_dict() elif isinstance(o, Simulation): return o.to_dict() elif isinstance(o, ITask): result = o.to_dict() result["task_type"] = o.__class__.__name__ return result elif isinstance(o, bytes): return None elif isinstance(o, (CommandLine, UUID)): return str(o) elif isinstance(o, Asset): return as_dict(o, exclude=['content']) elif isinstance(o, AssetCollection): return o.assets elif isinstance(o, (dict, int, list, str)): return o elif isinstance(o, datetime): return str(o)
[docs]def load_json_file(path: str) -> Union[Dict[Any, Any], List]: """ Load a json object from a file. Args: path: Path to file Returns: Contents of file parsed by JSON """ if not path: return try: with open(path, 'r') as fp: return json.load(fp) except IOError as e: user_logger.error(f"The file at {path} could not be loaded or parsed to JSON.\n{e}")