Source code for deephaven.pandasplugin.pandas_as_table

#
# Copyright (c) 2016-2022 Deephaven Data Labs and Patent Pending
#

from deephaven.pandas import to_table
from deephaven.plugin.object_type import Exporter, FetchOnlyObjectType
from pandas import DataFrame

NAME = "pandas.DataFrame"


class PandasDataFrameSerializer(FetchOnlyObjectType):
    @property
    def name(self) -> str:
        return NAME

    def is_type(self, object) -> bool:
        return isinstance(object, DataFrame)

    def to_bytes(self, exporter: Exporter, data_frame: DataFrame):
        exporter.reference(to_table(data_frame))
        return b''