"""JSON Serializer for Neo4j objects."""

import json

from neo4j.time import DateTime as Neo4jDateTime


class JSONNeo4jEncoder(json.JSONEncoder):
    """JSON Serializer for Neo4j objects."""

    def default(self, o):
        """Handle if o is serializable."""
        if isinstance(o, Neo4jDateTime):
            return o.iso_format()
        return super().default(o)