support setitem() for dictionary assignment, e.g. name['first']='jerry'

Author: derek73

nameparser/parser.py

@@ -91,12 +91,16 @@ def __ne__(self, other):
         return not (u(self)).lower() == (u(other)).lower()
 
     def __getitem__(self, key):
-        if isinstance(key, text_type):
-            return getattr(self, key)
         if isinstance(key, slice):
             return [getattr(self, x) for x in self._members[key]]
         else:
-            return getattr(self, self._members[key])
+            return getattr(self, key)
+
+    def __setitem__(self, key, value):
+        if key in self._members:
+            self._set_list(key, value)
+        else:
+            raise KeyError("Not a valid HumanName attribute", key)
 
     def next(self):
         return self.__next__()

tests.py

@@ -140,14 +140,25 @@ def test_slice(self):
         self.m(hn[1:], ['John', 'P.', 'Doe-Ray', 'CLU, CFP, LUTC',''], hn)
         self.m(hn[1:-2], ['John', 'P.', 'Doe-Ray'], hn)
 
-    def test_dictionary_like(self):
+    def test_getitem(self):
         hn = HumanName("Dr. John A. Kenneth Doe, Jr.")
         self.m(hn['title'], "Dr.", hn)
         self.m(hn['first'], "John", hn)
         self.m(hn['last'], "Doe", hn)
         self.m(hn['middle'], "A. Kenneth", hn)
         self.m(hn['suffix'], "Jr.", hn)
 
+    def test_setitem(self):
+        hn = HumanName("Dr. John A. Kenneth Doe, Jr.")
+        hn['title'] = 'test'
+        self.m(hn['title'], "test", hn)
+        hn['last'] = ['test','test2']
+        self.m(hn['last'], "test test2", hn)
+        with self.assertRaises(TypeError):
+            hn["suffix"] = [['test']]
+        with self.assertRaises(TypeError):
+            hn["suffix"] = {"test":"test"}
+
     def test_conjunction_names(self):
         hn = HumanName("johnny y")
         self.m(hn.first, "johnny", hn)