import pandas as pd
from dbengine.handlers import BaseRefHandler


class DataFrameHandler(BaseRefHandler):
  def is_eligible_for(self, obj):
    return isinstance(obj, pd.DataFrame)
  
  def tag(self):
    return "DataFrame"
  
  def serialize_to_bytes(self, df) -> bytes:
    from io import BytesIO
    import pickle
    return pickle.dumps(df)
  
  def deserialize_from_bytes(self, data: bytes):
    import pickle
    return pickle.loads(data)