salfter/youtube-dl
1
0
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl.git synced 2025-10-29 09:26:20 -07:00
Code Issues Packages Projects Releases Wiki Activity

[utils] Allow None in remove_{start,end}

Browse Source
This commit is contained in:
Sergey M․
2016-05-19 04:31:30 +06:00
parent b78531a36a
commit 46bc9b7d7c
2 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/test_utils.py
View File

@@ -50,6 +50,8 @@ from youtube_dl.utils import (
sanitize_path,
prepend_extension,
replace_extension,
remove_start,
remove_end,
remove_quotes,
shell_quote,
smuggle_url,
@@ -215,6 +217,16 @@ class TestUtil(unittest.TestCase):
self.assertEqual(replace_extension('.abc', 'temp'), '.abc.temp')
self.assertEqual(replace_extension('.abc.ext', 'temp'), '.abc.temp')
def test_remove_start(self):
self.assertEqual(remove_start(None, 'A - '), None)
self.assertEqual(remove_start('A - B', 'A - '), 'B')
self.assertEqual(remove_start('B - A', 'A - '), 'B - A')
def test_remove_end(self):
self.assertEqual(remove_end(None, ' - B'), None)
self.assertEqual(remove_end('A - B', ' - B'), 'A')
self.assertEqual(remove_end('B - A', ' - B'), 'B - A')
def test_remove_quotes(self):
self.assertEqual(remove_quotes(None), None)
self.assertEqual(remove_quotes('"'), '"')

8
youtube_dl/utils.py
View File

@@ -1549,15 +1549,11 @@ def setproctitle(title):
def remove_start(s, start):
if s.startswith(start):
return s[len(start):]
return s
return s[len(start):] if s is not None and s.startswith(start) else s
def remove_end(s, end):
if s.endswith(end):
return s[:-len(end)]
return s
return s[:-len(end)] if s is not None and s.endswith(end) else s
def remove_quotes(s):