from prefect.serializers import Serializer, D
import pandas as pd
from io import BytesIO


class PandasSerializer(Serializer):
    __dispatch_key__ = "pandas_serializer"

    def dumps(self, obj: D) -> bytes:
        return obj.to_csv(index=False).encode("utf8")

    def loads(self, obj: D) -> pd.DataFrame:
        return pd.read_csv(BytesIO(obj))