Source code for docarray.document.mixins.convert

from typing import Optional

import numpy as np

from .helper import _uri_to_buffer, _to_datauri, _is_datauri
from ...helper import T

[docs]class ConvertMixin: """Provide helper functions for :class:`Document` to support conversion between :attr:`.blob`, :attr:`.text` and :attr:`.buffer`."""
[docs] def convert_buffer_to_blob( self: T, dtype: Optional[str] = None, count: int = -1, offset: int = 0 ) -> T: """Assuming the :attr:`buffer` is a _valid_ buffer of Numpy ndarray, set :attr:`blob` accordingly. :param dtype: Data-type of the returned array; default: float. :param count: Number of items to read. ``-1`` means all data in the buffer. :param offset: Start reading the buffer from this offset (in bytes); default: 0. :return: itself after processed """ self.blob = np.frombuffer(self.buffer, dtype=dtype, count=count, offset=offset) return self
[docs] def convert_blob_to_buffer(self: T) -> T: """Convert :attr:`.blob` to :attr:`.buffer` inplace. :return: itself after processed """ self.buffer = self.blob.tobytes() return self
[docs] def convert_uri_to_datauri( self: T, charset: str = 'utf-8', base64: bool = False ) -> T: """Convert :attr:`.uri` to dataURI and store it in :attr:`.uri` inplace. :param charset: charset may be any character set registered with IANA :param base64: used to encode arbitrary octet sequences into a form that satisfies the rules of 7bit. Designed to be efficient for non-text 8 bit and binary data. Sometimes used for text data that frequently uses non-US-ASCII characters. :return: itself after processed """ if not _is_datauri(self.uri): buffer = _uri_to_buffer(self.uri) self.uri = _to_datauri(self.mime_type, buffer, charset, base64, binary=True) return self