Source code for czml3.base

import sys
from typing import Any

from pydantic import BaseModel, ConfigDict, model_validator

if sys.version_info[1] >= 11:
    from typing import Self
else:
    from typing_extensions import Self  # pragma: no cover

NON_DELETE_PROPERTIES = ["id", "delete"]




class BaseCZMLObject(BaseModel):
    model_config = ConfigDict(extra="forbid")



    @model_validator(mode="after")
    def check_delete(self) -> Self:
        if hasattr(self, "delete") and self.delete:
            for k in type(self).model_fields:
                if k not in NON_DELETE_PROPERTIES and getattr(self, k) is not None:
                    setattr(self, k, None)
        return self


    def __str__(self) -> str:
        return self.to_json()



    def dumps(self, **kwargs: Any) -> str:
        """Serialize the object to a JSON string.

        kwargs are passed to `BaseModel.model_dump_json()`.

        :return: JSON string representation of the object with None values excluded
        :rtype: str
        """
        return self.model_dump_json(exclude_none=True, **kwargs)




    def to_json(self, *, indent: int = 4, **kwargs: Any) -> str:
        """Return the object as a formatted JSON string.

        kwargs are passed to `BaseModel.model_dump_json()`.

        :param indent: Number of spaces for indentation, defaults to 4
        :type indent: int, optional
        :return: Formatted JSON string representation with None values excluded
        :rtype: str
        """
        return self.model_dump_json(exclude_none=True, indent=indent, **kwargs)




    def to_dict(self, **kwargs: Any) -> dict[str, Any]:
        """Return the object as a dictionary.

        kwargs are passed to `BaseModel.model_dump()`.

        :return: Dictionary representation of the object with None values excluded
        :rtype: dict
        """
        return self.model_dump(exclude_none=True, **kwargs)