ceph/src/boost/libs/python/src/object/pickle_support.cpp

   1 //  (C) Copyright R.W. Grosse-Kunstleve 2002.
   2 //  Distributed under the Boost Software License, Version 1.0. (See
   3 //  accompanying file LICENSE_1_0.txt or copy at
   4 //  http://www.boost.org/LICENSE_1_0.txt)
   5
   6 #include <boost/python/make_function.hpp>
   7 #include <boost/python/object/class.hpp>
   8 #include <boost/python/tuple.hpp>
   9 #include <boost/python/list.hpp>
  10 #include <boost/python/dict.hpp>
  11 #include <boost/python/str.hpp>
  12
  13 namespace boost { namespace python {
  14
  15 namespace {
  16
  17   tuple instance_reduce(object instance_obj)
  18   {
  19       list result;
  20       object instance_class(instance_obj.attr("__class__"));
  21       result.append(instance_class);
  22       object none;
  23       if (!getattr(instance_obj, "__safe_for_unpickling__", none))
  24       {
  25           str type_name(getattr(instance_class, "__name__"));
  26           str module_name(getattr(instance_class, "__module__", object("")));
  27           if (module_name)
  28               module_name += ".";
  29
  30           PyErr_SetObject(
  31                PyExc_RuntimeError,
  32                ( "Pickling of \"%s\" instances is not enabled"
  33                  " (http://www.boost.org/libs/python/doc/v2/pickle.html)"
  34                   % (module_name+type_name)).ptr()
  35           );
  36
  37           throw_error_already_set();
  38       }
  39       object getinitargs = getattr(instance_obj, "__getinitargs__", none);
  40       tuple initargs;
  41       if (!getinitargs.is_none()) {
  42           initargs = tuple(getinitargs());
  43       }
  44       result.append(initargs);
  45       object getstate = getattr(instance_obj, "__getstate__", none);
  46       object instance_dict = getattr(instance_obj, "__dict__", none);
  47       long len_instance_dict = 0;
  48       if (!instance_dict.is_none()) {
  49           len_instance_dict = len(instance_dict);
  50       }
  51       if (!getstate.is_none()) {
  52           if (len_instance_dict > 0) {
  53               object getstate_manages_dict = getattr(
  54                 instance_obj, "__getstate_manages_dict__", none);
  55               if (getstate_manages_dict.is_none()) {
  56                   PyErr_SetString(PyExc_RuntimeError,
  57                     "Incomplete pickle support"
  58                     " (__getstate_manages_dict__ not set)");
  59                   throw_error_already_set();
  60               }
  61           }
  62           result.append(getstate());
  63       }
  64       else if (len_instance_dict > 0) {
  65           result.append(instance_dict);
  66       }
  67       return tuple(result);
  68   }
  69
  70 } // namespace
  71
  72 object const& make_instance_reduce_function()
  73 {
  74     static object result(&instance_reduce);
  75     return result;
  76 }
  77
  78 }} // namespace boost::python