Introduce a U flag to collapse username to ~ #923 #915

vatlab · Mar 15, 2018 · 7923806 · gaow · Mar 15, 2018 · 7923806
1 parent ec11785
commit 7923806
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/sos/targets.py b/src/sos/targets.py
@@ -405,11 +405,21 @@ def __format__(self, format_spec):
         else:
             return str(self).__format__(format_spec)
 
+def collapseuser(path):
+    home = os.path.expanduser('~')
+    if path == home:
+        return '~'
+    elif path.startswith(home + os.sep):
+        return '~' + path[len(home):]
+    else:
+        return path
+
 class path(type(Path())):
     '''A regular target for files.
     '''
     CONVERTERS = {
         'u': os.path.expanduser,
+        'U': collapseuser,
         'e': lambda x: x.replace(' ', '\\ '),
         'a': lambda x: os.path.abspath(os.path.expanduser(x)),
         'l': lambda x: os.path.realpath(os.path.expanduser(x)),