Add more functionality to postgresql driver

pwman/data/drivers/postgresql.py

@@ -94,6 +94,19 @@ class PostgresqlDatabase(Database):
         #self._setnodetags(self._cur.lastrowid, tags)
         self._con.commit()
 
+    def getnodes(self, ids):
+        sql = "SELECT * FROM NODE WHERE ID IN ({})".format(','.join('%s' for
+                                                                    i in ids))
+        self._cur.execute(sql, (ids))
+        nodes = self._cur.fetchall()
+        nodes_w_tags = []
+        for node in nodes:
+            #tags = list(self._get_node_tags(node))
+            tags = []
+            nodes_w_tags.append(list(node) + tags)
+
+        return nodes_w_tags
+
     def removenodes(self, nodes):
         pass
 

pwman/tests/test_postgresql.py

@@ -66,8 +66,10 @@ class TestPostGresql(unittest.TestCase):
         self.assertEqual(row, ('TOP', 'SECRET'))
 
     def test_5_add_node(self):
-        self.db.add_node(("TBONE", "S3K43T", "example.org", "some note"))
-
+        innode = ["TBONE", "S3K43T", "example.org", "some note"]
+        self.db.add_node(innode)
+        outnode = self.db.getnodes([1])[0]
+        self.assertEqual(innode, outnode[1:])
 
 if __name__ == '__main__':