Add 'mbed cache ls' implementation and related functions

Author: screamerbg

mbed/mbed.py

@@ -255,6 +255,12 @@ def remove_readonly(func, path, _):
             func(path)
         shutil.rmtree(directory, onerror=remove_readonly)
 
+def sizeof_fmt(num, suffix='B'):
+    for unit in ['','K','M','G','T','P','E','Z']:
+        if abs(num) < 1024.0:
+            return "%3.1f%s%s" % (num, unit, suffix)
+        num /= 1024.0
+    return "%.1f%s%s" % (num, 'Yi', suffix)
 
 # Directory navigation
 @contextlib.contextmanager
@@ -2801,21 +2807,42 @@ def cache_(on=False, off=False, ls=False, purge=False, cache_dir=None, global_cf
         g.set_cfg('CACHE_DIR', cache_dir)
         action('Repository cache location set to \"%s\"' % cache_dir)
 
+    cfg = g.cache_cfg()
     if cmd == 'off' or cmd == 'on':
-        g.set_cfg('CACHE', 'enabled' if cmd == 'on' else 'disabled');
+        g.set_cfg('CACHE', 'enabled' if cmd == 'on' else 'disabled')
         cfg = g.cache_cfg()
         action('Repository cache is now %s.' % str(cfg['cache']).upper())
         action('Cache location \"%s\"' % cfg['cache_dir'])
     elif cmd == 'ls':
-        print 'list'
+        def get_size_(path):
+            size = 0
+            for dirpath, dirs, files in os.walk(path):
+                for f in files:
+                    size += os.path.getsize(os.path.join(dirpath, f))
+            return size
+
+        action('Listing cached repositories in \"%s\"' % cfg['cache_base'])
+        repos = []
+        total_size = 0
+        for dirpath, dirs, files in os.walk(cfg['cache_dir']):
+            dirs[:] = [d for d in dirs  if not d.startswith('.')]
+            if Repo.isrepo(dirpath):
+                repo = Repo().fromrepo(dirpath)
+                url = repo.url
+                size = get_size_(repo.path)
+                total_size += size
+                log('* %s %s\n' % ('{:60}'.format(url), sizeof_fmt(size).rjust(8)))
+                for d in dirs:
+                    dirs.remove(d)
+        log('%s %s\n' % ('{:62}'.format('Total:'), sizeof_fmt(total_size).rjust(8)))
+
     elif cmd == 'purge':
-        cfg = g.cache_cfg()
         action('Purging cached repositories in \"%s\"' % cfg['cache_base'])
-        if os.path.isdir(path):
+        if os.path.isdir(cfg['cache_dir']):
             rmtree_readonly(cfg['cache_dir'])
-        action('Complete')
+
+        action('Purge complete')
     else:
-        cfg = g.cache_cfg()
         action('Repository cache is %s.' % str(cfg['cache']).upper())
         action('Cache location \"%s\"' % cfg['cache_base'])