Merge pull request #15 from gboulay/dict-update

Add update() to LRU

Author: amitdev

lru.c

@@ -332,6 +332,28 @@ get_key(Node *node)
     return node->key;
 }
 
+static PyObject *
+LRU_update(LRU *self, PyObject *args, PyObject *kwargs)
+{
+	PyObject *key, *value;
+	PyObject *arg = NULL;
+	Py_ssize_t pos = 0;
+
+	if ((PyArg_ParseTuple(args, "|O", &arg))) {
+		if (arg && PyDict_Check(arg)) {
+			while (PyDict_Next(arg, &pos, &key, &value))
+				lru_ass_sub(self, key, value);
+		}
+	}
+	
+	if (kwargs != NULL && PyDict_Check(kwargs)) {
+		while (PyDict_Next(kwargs, &pos, &key, &value))
+			lru_ass_sub(self, key, value);
+	}
+
+	Py_RETURN_NONE;
+}
+
 static PyObject *
 LRU_peek_first_item(LRU *self)
 {
@@ -483,6 +505,8 @@ static PyMethodDef LRU_methods[] = {
                     PyDoc_STR("L.peek_first_item() -> returns the MRU item (key,value) without changing key order")},
     {"peek_last_item", (PyCFunction)LRU_peek_last_item, METH_NOARGS,
                     PyDoc_STR("L.peek_last_item() -> returns the LRU item (key,value) without changing key order")},
+    {"update", (PyCFunction)LRU_update, METH_VARARGS | METH_KEYWORDS,
+                    PyDoc_STR("L.update() -> update value for key in LRU")},
     {NULL,	NULL},
 };
 

test/test_lru.py

@@ -106,6 +106,24 @@ def test_access(self):
                 self.assertEqual(l[i], str(i))
                 self.assertEqual(l.get(i,None), str(i))
 
+
+    def test_update(self):
+        l = LRU(2)
+        l['a'] = 1
+        self.assertEqual(l['a'], 1)
+        l.update(a=2)
+        self.assertEqual(l['a'], 2)
+        l['b'] = 2
+        self.assertEqual(l['b'], 2)
+        l.update(b=3)
+        self.assertEqual(l['a'], 2)
+        self.assertEqual(l['b'], 3)
+        l.update({'a':1, 'b':2})
+        self.assertEqual(l['a'], 1)
+        self.assertEqual(l['b'], 2)
+        l.update()
+
+
     def test_peek_first_item(self):
         l = LRU(2)
         self.assertEqual(None, l.peek_first_item())