Added and tested log.add_attribute method for adding new properties to log records.

Author: Joel Becker

gitnet/gitnet_tests/test_log.py

@@ -102,6 +102,39 @@ def test_commit_log_values(self):
         # Delete the temporary .git folder
         sub.call(["rm", "-rf", ".git"])
 
+class AddAttributeTest(unittest.TestCase):
+    """
+    Tests for the add_attribute method.
+    """
+    def setUp(self):
+        data = {"Bob": {"author": 'Bob',
+                        "email": 'bob@gmail.com',
+                        "type": 'author',
+                        "loc": 'Waterloo',
+                        "books": ['BookA', 'BookB']},
+                "Bobby": {"author": 'Bobby',
+                          "email": 'bobby@gmail.com',
+                          "type": 'author',
+                          "loc": 'Kitchener',
+                          "books": ['BookC', 'BookD']},
+                "Robert": {"author": 'Robert',
+                           "email": 'robby@gmail.com',
+                           "type": 'author',
+                           "loc": 'Kitchener',
+                           "books": ['BookC', 'BookD']}}
+        self.log = gitnet.Log(data)
+
+    def test_new_attr_1(self):
+        new_log = self.log.add_attribute("letter", lambda d: d["author"][0])
+        self.assertEqual(new_log["Bob"]["letter"], "B")
+        self.assertEqual(new_log["Bobby"]["letter"], "B")
+        self.assertEqual(new_log["Robert"]["letter"], "R")
+
+    def test_new_attr_1(self):
+        new_log = self.log.add_attribute("edge",lambda d: gitnet.net_edges_simple("v1","v2",d,["loc"]))
+        self.assertEqual(new_log["Bob"]["edge"], ("v1","v2",{"loc":"Waterloo"}))
+        self.assertEqual(new_log["Bobby"]["edge"], ("v1","v2",{"loc":"Kitchener"}))
+        self.assertEqual(new_log["Robert"]["edge"], ("v1","v2",{"loc":"Kitchener"}))
 
 class DescribeTests(unittest.TestCase):
     def setUp(self):

gitnet/log.py

@@ -136,6 +136,29 @@ def attributes(self):
             attr_list = attr_list + sorted(list(attr))
         return attr_list
 
+    def add_attribute(self,name,helper):
+        """
+        Creates a new record attribute.
+
+        **Parameters** :
+
+        > *name* : `string`
+
+        >> The name of the new attribute.
+
+        > *helper* : `None`
+
+        >> A helper function, which takes an attribute dict and produces the new attribute.
+
+        **Return** :
+
+        > A new Log object, identical to self but with the desired attribute.
+        """
+        self_copy = copy.deepcopy(self)
+        for n in self_copy:
+            self_copy[n][name] = helper(self_copy[n])
+        return self_copy
+
     def author_email_list(self):
         """
         Gathers each unique author email combination from the log, and then prints them in a list.