Source code for unistream.records.dataclass

# -*- coding: utf-8 -*-

"""
Implements :class:`DataClassRecord`, a frozen record built on :mod:`dataclasses`.
"""

import typing as T
import uuid
import json
import dataclasses

from func_args.api import BaseFrozenModel

from ..utils import get_utc_now
from ..record import BaseRecord


def id_factory() -> str:
    return str(uuid.uuid4())


def create_at_factory() -> str:
    return get_utc_now().isoformat()




@dataclasses.dataclass(frozen=True)
class DataClassRecord(BaseRecord, BaseFrozenModel):
    """
    Record built on top of `dataclasses <https://docs.python.org/3/library/dataclasses.html>`_.
    """

    id: str = dataclasses.field(default_factory=id_factory)
    create_at: str = dataclasses.field(default_factory=create_at_factory)



    def serialize(self) -> str:
        return json.dumps(dataclasses.asdict(self))




    @classmethod
    def deserialize(
        cls,
        data: str,
    ) -> "DataClassRecord":
        return cls(**json.loads(data))




T_DATA_CLASS_RECORD = T.TypeVar("T_DATA_CLASS_RECORD", bound=DataClassRecord)