Catherine Arnott-Smith

import json


class BinaryNinjaEncoder(json.JSONEncoder):
    """Custom JSON encoder for BinaryNinja types"""

    def default(self, obj):
        # Handle Binary Ninja specific types

        # for values, we try to obtain the json for the value property
        if hasattr(obj, "value") and hasattr(obj, "confidence"):
            return obj.value

        # for strings, we simply compute the native string
        if hasattr(obj, "__str__"):
            return str(obj)

        # for all the others values we use the normal json encoder
        return json.JSONEncoder.default(self, obj)