ahawker/ulid

"""
    ulid/providers/default
    ~~~~~~~~~~~~~~~~~~~~~~

    Contains data provider that creates new randomness values for the same timestamp.
"""
import os

from .. import hints
from . import base, time


class Provider(base.Provider):
    """
    Provider that creates new randomness values for the same timestamp.
    """

    def timestamp(self) -> hints.Bytes:
        """
        Create a new timestamp value.

        :return: Timestamp value in bytes.
        :rtype: :class:`~bytes`
        """
        return time.milliseconds().to_bytes(6, byteorder='big')

    def randomness(self, timestamp: hints.Bytes) -> hints.Bytes:
        """
        Create a new randomness value.

        :param timestamp: Timestamp in milliseconds
        :type timestamp: :class:`~bytes`
        :return: Randomness value in bytes.
        :rtype: :class:`~bytes`
        """
        return os.urandom(10)