Get information about the pickling process

pickle_info.py

# Inspired from: https://airflow.incubator.apache.org/_modules/airflow/models.html#BaseOperator

import pickle
import logging
from datetime import datetime
import traceback

def pickle_info(obj, session=None):
  d = {}
  d['is_picklable'] = True
  try:
      dttm = datetime.now()
      pickled = pickle.dumps(obj)
      d['pickle_len'] = len(pickled)
      d['pickling_duration'] = "{}".format(datetime.now() - dttm)
  except Exception as e:
      logging.debug(e)
      d['is_picklable'] = False
      d['stacktrace'] = traceback.format_exc()
  return d