ee696643cd
* Don't treat typing as an "in-between" module for import order That was a < 3.5 era thing. * Tighten scope of some pylint unused-import disables To avoid isort moving a top level one around, undesirably broadening its scope.
26 lines
710 B
Python
26 lines
710 B
Python
"""Helpers to help with encoding Home Assistant objects in JSON."""
|
|
from datetime import datetime
|
|
import json
|
|
import logging
|
|
from typing import Any
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class JSONEncoder(json.JSONEncoder):
|
|
"""JSONEncoder that supports Home Assistant objects."""
|
|
|
|
# pylint: disable=method-hidden
|
|
def default(self, o: Any) -> Any:
|
|
"""Convert Home Assistant objects.
|
|
|
|
Hand other objects to the original method.
|
|
"""
|
|
if isinstance(o, datetime):
|
|
return o.isoformat()
|
|
if isinstance(o, set):
|
|
return list(o)
|
|
if hasattr(o, 'as_dict'):
|
|
return o.as_dict()
|
|
|
|
return json.JSONEncoder.default(self, o)
|