[utils] Recognize lowercase units in parse_filesize
authorSergey M․ <dstftw@gmail.com>
Thu, 18 Aug 2016 16:32:00 +0000 (23:32 +0700)
committerSergey M․ <dstftw@gmail.com>
Thu, 18 Aug 2016 16:32:00 +0000 (23:32 +0700)
test/test_utils.py
youtube_dl/utils.py

index 74fcf91c0cb622662d3a7c2d4985377a0ea6ce7b..cb578cd536ab0db6161306ccb795206851f24234 100644 (file)
@@ -817,7 +817,9 @@ class TestUtil(unittest.TestCase):
         self.assertEqual(parse_filesize('2 MiB'), 2097152)
         self.assertEqual(parse_filesize('5 GB'), 5000000000)
         self.assertEqual(parse_filesize('1.2Tb'), 1200000000000)
+        self.assertEqual(parse_filesize('1.2tb'), 1200000000000)
         self.assertEqual(parse_filesize('1,24 KB'), 1240)
+        self.assertEqual(parse_filesize('1,24 kb'), 1240)
 
     def test_parse_count(self):
         self.assertEqual(parse_count(None), None)
index b3b687a314681de17547104f7f99e883a312c322..35362e7674205a62fd71ba347829a55e1f4f15a4 100644 (file)
@@ -1508,34 +1508,42 @@ def parse_filesize(s):
         'KB': 1000,
         'kB': 1024,
         'Kb': 1000,
+        'kb': 1000,
         'MiB': 1024 ** 2,
         'MB': 1000 ** 2,
         'mB': 1024 ** 2,
         'Mb': 1000 ** 2,
+        'mb': 1000 ** 2,
         'GiB': 1024 ** 3,
         'GB': 1000 ** 3,
         'gB': 1024 ** 3,
         'Gb': 1000 ** 3,
+        'gb': 1000 ** 3,
         'TiB': 1024 ** 4,
         'TB': 1000 ** 4,
         'tB': 1024 ** 4,
         'Tb': 1000 ** 4,
+        'tb': 1000 ** 4,
         'PiB': 1024 ** 5,
         'PB': 1000 ** 5,
         'pB': 1024 ** 5,
         'Pb': 1000 ** 5,
+        'pb': 1000 ** 5,
         'EiB': 1024 ** 6,
         'EB': 1000 ** 6,
         'eB': 1024 ** 6,
         'Eb': 1000 ** 6,
+        'eb': 1000 ** 6,
         'ZiB': 1024 ** 7,
         'ZB': 1000 ** 7,
         'zB': 1024 ** 7,
         'Zb': 1000 ** 7,
+        'zb': 1000 ** 7,
         'YiB': 1024 ** 8,
         'YB': 1000 ** 8,
         'yB': 1024 ** 8,
         'Yb': 1000 ** 8,
+        'yb': 1000 ** 8,
     }
 
     return lookup_unit_table(_UNIT_TABLE, s)