Make the downloader statistics work.

[quix0rs-apt-p2p.git] / apt_p2p / util.py

diff --git a/apt_p2p/util.py b/apt_p2p/util.py
index 3d1a50eafb4e813eb1e3a360f94dbd52cc546ee9..ec0287cec4bc9790ad348a232d769644844b7445 100644 (file)
--- a/apt_p2p/util.py
+++ b/apt_p2p/util.py
@@ -160,11 +160,15 @@ def byte_format(s):
     @param s: the number of bytes
     @rtype: C{string}
     @return: the formatted size with appropriate units
     @param s: the number of bytes
     @rtype: C{string}
     @return: the formatted size with appropriate units

-    

     """
     """

-    
-    if (s < 1024):
-        r = str(s) + 'B'
+    if (s < 1):
+        r = str(int(s*1000.0)/1000.0) + 'B'
+    elif (s < 10):
+        r = str(int(s*100.0)/100.0) + 'B'
+    elif (s < 102):
+        r = str(int(s*10.0)/10.0) + 'B'
+    elif (s < 1024):
+        r = str(int(s)) + 'B'

     elif (s < 10485):
         r = str(int((s/1024.0)*100.0)/100.0) + 'KiB'
     elif (s < 104857):
     elif (s < 10485):
         r = str(int((s/1024.0)*100.0)/100.0) + 'KiB'
     elif (s < 104857):