From 4707f32a416dfb8e6a6956bb1e2fe60997cc9f4b Mon Sep 17 00:00:00 2001
From: Lars Wirzenius <liw@liw.fi>
Date: Fri, 18 Feb 2011 23:08:35 +0000
Subject: Implement relative path output.

---
 summain | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'summain')

diff --git a/summain b/summain
index 58ac582..c37e57c 100755
--- a/summain
+++ b/summain
@@ -24,6 +24,10 @@ import summainlib
 
 class Summain(cliapp.Application):
 
+    def add_settings(self):
+        self.add_boolean_setting(['relative-paths', 'r'],
+                                 'print paths relative to arguments')
+
     def files(self, root):
         if os.path.isdir(root):
             for dirname, dirnames, filenames in os.walk(root):
@@ -34,11 +38,12 @@ class Summain(cliapp.Application):
             yield root
 
     def process_args(self, args):
+        relative = self['relative-paths']
         normalizer = summainlib.NumberNormalizer()
         for root in args:
             for filename in self.files(root):
                 o = summainlib.FilesystemObject(filename, normalizer)
-                sys.stdout.write(o.format())
+                sys.stdout.write(o.format(root if relative else None))
                 sys.stdout.write('\n')
 
 
-- 
cgit v1.2.1