Add a get_first convenience function to ODict.

2 files changed, 12 insertions, 0 deletions

diff --git a/netlib/odict.py b/netlib/odict.py
index afc33caa..629fcade 100644
--- a/netlib/odict.py
+++ b/netlib/odict.py
@@ -80,6 +80,12 @@ class ODict:
         else:
             return d
 
+    def get_first(self, k, d=None):
+        if k in self:
+            return self[k][0]
+        else:
+            return d
+
     def items(self):
         return self.lst[:]
 
diff --git a/test/test_odict.py b/test/test_odict.py
index e7453e2d..f27f6f8b 100644
--- a/test/test_odict.py
+++ b/test/test_odict.py
@@ -85,6 +85,12 @@ class TestODict:
         assert self.od.get("one") == ["two"]
         assert self.od.get("two") == None
 
+    def test_get_first(self):
+        self.od.add("one", "two")
+        self.od.add("one", "three")
+        assert self.od.get_first("one") == "two"
+        assert self.od.get_first("two") == None
+
 
 class TestODictCaseless:
     def setUp(self):