Add provider plugins loader

Signed-off-by: Simo Sorce <simo@redhat.com>

2 files changed, 73 insertions, 0 deletions

diff --git a/ipsilon/util/data.py b/ipsilon/util/data.py
index 251c998..cbd3b49 100755
--- a/ipsilon/util/data.py
+++ b/ipsilon/util/data.py
@@ -188,3 +188,69 @@ class Store(object):
         finally:
             if con:
                 con.close()
+
+    def get_data(self, plugin):
+        con = None
+        rows = []
+        try:
+            con = sqlite3.connect(self._admin_dbname)
+            cur = con.cursor()
+            cur.execute("CREATE TABLE IF NOT EXISTS " +
+                        plugin + "_data (id INTEGER, name TEXT, value TEXT)")
+            cur.execute("SELECT * FROM " + plugin + "_data")
+            rows = cur.fetchall()
+            con.commit()
+        except sqlite3.Error, e:
+            if con:
+                con.rollback()
+            cherrypy.log.error("Failed to load %s data: [%s]" % (plugin, e))
+        finally:
+            if con:
+                con.close()
+
+        data = dict()
+        for row in rows:
+            if row[0] not in data:
+                data[row[0]] = dict()
+
+            item = data[row[0]]
+            if row[1] in item:
+                if item[row[1]] is list:
+                    item[row[1]].append(row[2])
+                else:
+                    v = item[row[1]]
+                    item[row[1]] = [v, row[2]]
+            else:
+                item[row[1]] = row[2]
+
+        return data
+
+    def save_data(self, plugin, data):
+        SELECT = "SELECT name, value FROM %s_data WHERE id=?" % plugin
+        UPDATE = "UPDATE %s_data SET value=? WHERE id=? AND name=?" % plugin
+        INSERT = "INSERT INTO %s_data VALUES(?,?,?)" % plugin
+        con = None
+        try:
+            con = sqlite3.connect(self._admin_dbname)
+            cur = con.cursor()
+            for idval in data:
+                curvals = dict()
+                for row in cur.execute(SELECT, (idval,)):
+                    curvals[row[0]] = row[1]
+
+                datum = data[idval]
+                for name in datum:
+                    if name in curvals:
+                        cur.execute(UPDATE, (datum[name], idval, name))
+                    else:
+                        cur.execute(INSERT, (idval, name, datum[name]))
+
+            con.commit()
+        except sqlite3.Error, e:
+            if con:
+                con.rollback()
+            cherrypy.log.error("Failed to store %s data: [%s]" % (plugin, e))
+            raise
+        finally:
+            if con:
+                con.close()
diff --git a/ipsilon/util/plugin.py b/ipsilon/util/plugin.py
index 2077023..16a086a 100755
--- a/ipsilon/util/plugin.py
+++ b/ipsilon/util/plugin.py
@@ -102,6 +102,7 @@ class PluginObject(object):
         self.name = None
         self._config = None
         self._options = None
+        self._data = Store()
 
     def get_config_desc(self):
         """ The configuration description is a dictionary that provides
@@ -136,3 +137,9 @@ class PluginObject(object):
         if not self._config:
             self._config = dict()
         self._config[option] = value
+
+    def get_data(self):
+        return self._data.get_data(self.name)
+
+    def save_data(self, data):
+        self._data.save_data(self.name, data)