move bytes2human() into psutil._common and reused it from scripts dir

1 files changed, 1 insertions, 17 deletions

diff --git a/scripts/disk_usage.py b/scripts/disk_usage.py
index 37f4da0c..1860401f 100755
--- a/scripts/disk_usage.py
+++ b/scripts/disk_usage.py
@@ -18,23 +18,7 @@ Device               Total     Used     Free  Use %      Type  Mount
 import sys
 import os
 import psutil
-
-
-def bytes2human(n):
-    # http://code.activestate.com/recipes/578019
-    # >>> bytes2human(10000)
-    # '9.8K'
-    # >>> bytes2human(100001221)
-    # '95.4M'
-    symbols = ('K', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y')
-    prefix = {}
-    for i, s in enumerate(symbols):
-        prefix[s] = 1 << (i + 1) * 10
-    for s in reversed(symbols):
-        if n >= prefix[s]:
-            value = float(n) / prefix[s]
-            return '%.1f%s' % (value, s)
-    return "%sB" % n
+from psutil._common import bytes2human
 
 
 def main():