openSUSE-Python
diff --git a/‎src/config.h‎
Lines changed: 4 additions & 0 deletions b/‎src/config.h‎
Lines changed: 4 additions & 0 deletions
diff --git a/‎src/dmidecodemodule.c‎
Lines changed: 26 additions & 1 deletion b/‎src/dmidecodemodule.c‎
Lines changed: 26 additions & 1 deletion
diff --git a/‎src/dmihelper.h‎
Lines changed: 1 addition & 0 deletions b/‎src/dmihelper.h‎
Lines changed: 1 addition & 0 deletions
@@ -23,4 +23,8 @@
 #define ALIGNMENT_WORKAROUND
 #endif
 
+#ifndef PYTHON_XML_MAP
+#define PYTHON_XML_MAP "/usr/share/python-dmidecode/pythonmap.xmp"
+#endif
+
 #endif
@@ -21,6 +21,7 @@ static void init(void)
         opt.type = NULL;
         opt.mappingxml = NULL;
         opt.dmiversion_n = NULL;
+        opt.python_xml_map = strdup(PYTHON_XML_MAP);
 }
 
 u8 *parse_opt_type(u8 * p, const char *arg)
@@ -250,7 +251,7 @@ static PyObject *dmidecode_get(PyObject *self, const char *section)
                 // Convert the retrieved XML nodes to Python dicts
                 if( opt.mappingxml == NULL ) {
                         // Load mapping into memory
-                        opt.mappingxml = xmlReadFile("pythonmap.xml", NULL, 0);
+                        opt.mappingxml = xmlReadFile(opt.python_xml_map, NULL, 0);
                         assert( opt.mappingxml != NULL );
                 }
 
@@ -387,6 +388,27 @@ static PyObject *dmidecode_set_dev(PyObject * self, PyObject * arg)
         //PyErr_Occurred();
 }
 
+static PyObject *dmidecode_set_pythonxmlmap(PyObject * self, PyObject * arg)
+{
+        if(PyString_Check(arg)) {
+                struct stat fileinfo;
+                char *fname = PyString_AsString(arg);
+
+                memset(&fileinfo, 0, sizeof(struct stat));
+
+                if( stat(fname, &fileinfo) != 0 ) {
+                        PyErr_SetString(PyExc_IOError, "Could not access the given python map XML file");
+                        return NULL;
+                }
+
+                free(opt.python_xml_map);
+                opt.python_xml_map = strdup(fname);
+                Py_RETURN_TRUE;
+        } else {
+                Py_RETURN_FALSE;
+        }
+}
+
 
 static PyMethodDef DMIDataMethods[] = {
         {(char *)"dump", dmidecode_dump, METH_NOARGS, (char *)"Dump dmidata to set file"},
@@ -407,6 +429,9 @@ static PyMethodDef DMIDataMethods[] = {
 
         {(char *)"type", dmidecode_get_type, METH_VARARGS, (char *)"By Type"},
 
+        {(char *)"pythonmap", dmidecode_set_pythonxmlmap, METH_O,
+         (char *) "Use another python dict map definition. The default file is " PYTHON_XML_MAP},
+
         {NULL, NULL, 0, NULL}
 };
 
 
@@ -115,6 +115,7 @@ typedef struct _options {
         xmlDoc *mappingxml;
         xmlNode *dmiversion_n;
         PyObject *dumpfile;
+        char *python_xml_map;
 } options;
 extern options opt;