19 lines
445 B
Python
19 lines
445 B
Python
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) |