source4/param/pyparam_util.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007-2008
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include <Python.h>
  21 #include "includes.h"
  22 #include "param/param.h"
  23 #include "param/loadparm.h"
  24 #include "lib/talloc/pytalloc.h"
  25
  26 #define PyLoadparmContext_AsLoadparmContext(obj) py_talloc_get_type(obj, struct loadparm_context)
  27
  28 _PUBLIC_ struct loadparm_context *lp_from_py_object(TALLOC_CTX *mem_ctx, PyObject *py_obj)
  29 {
  30     struct loadparm_context *lp_ctx;
  31
  32     if (PyString_Check(py_obj)) {
  33         lp_ctx = loadparm_init(mem_ctx);
  34         if (!lp_load(lp_ctx, PyString_AsString(py_obj))) {
  35             talloc_free(lp_ctx);
  36                         PyErr_Format(PyExc_RuntimeError, "Unable to load %s",
  37                                                  PyString_AsString(py_obj));
  38             return NULL;
  39         }
  40         return lp_ctx;
  41     }
  42
  43     if (py_obj == Py_None) {
  44         lp_ctx = loadparm_init(mem_ctx);
  45                 /* We're not checking that loading the file succeeded *on purpose */
  46         lp_load_default(lp_ctx);
  47         return lp_ctx;
  48     }
  49
  50     return PyLoadparmContext_AsLoadparmContext(py_obj);
  51 }
  52
  53 struct loadparm_context *py_default_loadparm_context(TALLOC_CTX *mem_ctx)
  54 {
  55     struct loadparm_context *ret;
  56     ret = loadparm_init(mem_ctx);
  57     if (!lp_load_default(ret))
  58         return NULL;
  59     return ret;
  60 }
  61
  62