From: Jaime Marquínez Ferrándiz Date: Sat, 19 Mar 2016 10:42:35 +0000 (+0100) Subject: [utils] lookup_unit_table: Match word boundary instead of end of string X-Git-Url: https://git.cielonegro.org/gitweb.cgi?a=commitdiff_plain;h=782b1b5bd1cdaaead6865dee5d300486e7dd8348;p=youtube-dl.git [utils] lookup_unit_table: Match word boundary instead of end of string --- diff --git a/test/test_utils.py b/test/test_utils.py index 9a3a8ddff..325b870cc 100644 --- a/test/test_utils.py +++ b/test/test_utils.py @@ -702,6 +702,8 @@ class TestUtil(unittest.TestCase): self.assertEqual(parse_count('1.000'), 1000) self.assertEqual(parse_count('1.1k'), 1100) self.assertEqual(parse_count('1.1kk'), 1100000) + self.assertEqual(parse_count('1.1kk '), 1100000) + self.assertEqual(parse_count('1.1kk views'), 1100000) def test_version_tuple(self): self.assertEqual(version_tuple('1'), (1,)) diff --git a/youtube_dl/utils.py b/youtube_dl/utils.py index bad1c4ea8..067b8a184 100644 --- a/youtube_dl/utils.py +++ b/youtube_dl/utils.py @@ -1346,7 +1346,7 @@ def format_bytes(bytes): def lookup_unit_table(unit_table, s): units_re = '|'.join(re.escape(u) for u in unit_table) m = re.match( - r'(?P[0-9]+(?:[,.][0-9]*)?)\s*(?P%s)$' % units_re, s) + r'(?P[0-9]+(?:[,.][0-9]*)?)\s*(?P%s)\b' % units_re, s) if not m: return None num_str = m.group('num').replace(',', '.')