Implement lookup of username corresponding to st_uid. With unit test.

2 files changed, 8 insertions, 0 deletions

diff --git a/summainlib.py b/summainlib.py
index 5ce9f10..ff0bf12 100644
--- a/summainlib.py
+++ b/summainlib.py
@@ -15,6 +15,7 @@
 
 
 import os
+import pwd
 import time
 import urllib
 
@@ -41,11 +42,15 @@ class FilesystemObject(object):
         self['Nlink'] = '%d' % stat_result.st_nlink
         self['Size'] = '%d' % stat_result.st_size
         self['Uid'] = '%d' % stat_result.st_uid
+        self['Username'] = self.lookup_username(stat_result.st_uid)
 
     def format_time(self, timestamp):
         return time.strftime('%Y-%m-%d %H:%M:%S +0000', 
                              time.gmtime(timestamp))
 
+    def lookup_username(self, uid):
+        return pwd.getpwuid(uid).pw_name
+
     def hook_name(self, value):
         return urllib.quote(value)
 
diff --git a/summainlib_tests.py b/summainlib_tests.py
index 7f5a9fd..db2e6b5 100644
--- a/summainlib_tests.py
+++ b/summainlib_tests.py
@@ -69,3 +69,6 @@ class FilesystemObjectTests(unittest.TestCase):
     def test_formats_uid_correctly(self):
         self.assertEqual(self.new('foo')['Uid'], '0')
 
+    def test_formats_username_correctly(self):
+        self.assertEqual(self.new('foo')['Username'], 'root')
+