src/PyPtr.cxx

   1 // Copyright (C) 2019  EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19 #include "PyPtr.hxx"
  20 #include "Errors.hxx"
  21
  22 namespace py2cpp
  23 {
  24
  25 PyPtr PyPtr::getAttr(const std::string& attribute)const
  26 {
  27   PyObject* result = nullptr;
  28   PyObject* thisObj = get();
  29   if(nullptr == thisObj)
  30   {
  31     std::string message = "Cannot get attribute ";
  32     message += attribute;
  33     message += " on a NULL object.\n";
  34     throw AttributeException(message);
  35   }
  36
  37   if(!PyObject_HasAttrString(thisObj, attribute.c_str()))
  38   {
  39     std::string message = "Attribute ";
  40     message += attribute;
  41     message += " does not exist.\n";
  42     throw AttributeException(message);
  43   }
  44
  45   result = PyObject_GetAttrString(thisObj, attribute.c_str());
  46   return PyPtr(result);
  47 }
  48
  49 void PyPtr::setAttr(const std::string& attribute, const PyPtr& value)const
  50 {
  51   PyObject* thisObj = get();
  52   if(nullptr == thisObj)
  53   {
  54     std::string message = "Cannot set attribute ";
  55     message += attribute;
  56     message += " on a NULL object.\n";
  57     throw AttributeException(message);
  58   }
  59
  60   if(0 > PyObject_SetAttrString(thisObj, attribute.c_str(), value.get()))
  61   {
  62     std::string message = "Failed to set attribute ";
  63     message += attribute;
  64     message += ".\n";
  65     message += getLastPyError();
  66     throw AttributeException(message);
  67   }
  68 }
  69
  70 std::string PyPtr::repr()const
  71 {
  72   std::string result;
  73   PyObject* thisObj = get();
  74   if(thisObj)
  75   {
  76     PyObject* pyResult = PyObject_Repr(thisObj);
  77     if(pyResult && PyUnicode_Check(pyResult))
  78     {
  79       const char* charstr = PyUnicode_AsUTF8(pyResult);
  80       if(charstr != nullptr)
  81         result = charstr;
  82     }
  83     Py_XDECREF(pyResult);
  84   }
  85   return result;
  86 }
  87
  88 }