浏览代码

[utils] Improve colon handling (Fixes #4683)

Philipp Hagemeister 10 年之前
父节点
当前提交
2aeb06d6dc
共有 2 个文件被更改,包括 6 次插入0 次删除
  1. 4 0
      test/test_utils.py
  2. 2 0
      youtube_dl/utils.py

+ 4 - 0
test/test_utils.py

@@ -79,6 +79,10 @@ class TestUtil(unittest.TestCase):
         tests = '\u043a\u0438\u0440\u0438\u043b\u043b\u0438\u0446\u0430'
         self.assertEqual(sanitize_filename(tests), tests)
 
+        self.assertEqual(
+            sanitize_filename('New World record at 0:12:34'),
+            'New World record at 0_12_34')
+
         forbidden = '"\0\\/'
         for fc in forbidden:
             for fbc in forbidden:

+ 2 - 0
youtube_dl/utils.py

@@ -287,6 +287,8 @@ def sanitize_filename(s, restricted=False, is_id=False):
             return '_'
         return char
 
+    # Handle timestamps
+    s = re.sub(r'[0-9]+(?::[0-9]+)+', lambda m: m.group(0).replace(':', '_'), s)
     result = ''.join(map(replace_insane, s))
     if not is_id:
         while '__' in result: