mirror of https://github.com/yt-dlp/yt-dlp.git
[utils] Add remove_quotes
This commit is contained in:
parent
eb0bdc2c3e
commit
31b2051e21
|
@ -43,6 +43,7 @@ from youtube_dl.utils import (
|
|||
sanitize_path,
|
||||
prepend_extension,
|
||||
replace_extension,
|
||||
remove_quotes,
|
||||
shell_quote,
|
||||
smuggle_url,
|
||||
str_to_int,
|
||||
|
@ -200,6 +201,15 @@ class TestUtil(unittest.TestCase):
|
|||
self.assertEqual(replace_extension('.abc', 'temp'), '.abc.temp')
|
||||
self.assertEqual(replace_extension('.abc.ext', 'temp'), '.abc.temp')
|
||||
|
||||
def test_remove_quotes(self):
|
||||
self.assertEqual(remove_quotes(None), None)
|
||||
self.assertEqual(remove_quotes('"'), '"')
|
||||
self.assertEqual(remove_quotes("'"), "'")
|
||||
self.assertEqual(remove_quotes(';'), ';')
|
||||
self.assertEqual(remove_quotes('";'), '";')
|
||||
self.assertEqual(remove_quotes('""'), '')
|
||||
self.assertEqual(remove_quotes('";"'), ';')
|
||||
|
||||
def test_ordered_set(self):
|
||||
self.assertEqual(orderedSet([1, 1, 2, 3, 4, 4, 5, 6, 7, 3, 5]), [1, 2, 3, 4, 5, 6, 7])
|
||||
self.assertEqual(orderedSet([]), [])
|
||||
|
|
|
@ -1406,6 +1406,15 @@ def remove_end(s, end):
|
|||
return s
|
||||
|
||||
|
||||
def remove_quotes(s):
|
||||
if s is None or len(s) < 2:
|
||||
return s
|
||||
for quote in ('"', "'", ):
|
||||
if s[0] == quote and s[-1] == quote:
|
||||
return s[1:-1]
|
||||
return s
|
||||
|
||||
|
||||
def url_basename(url):
|
||||
path = compat_urlparse.urlparse(url).path
|
||||
return path.strip('/').split('/')[-1]
|
||||
|
|
Loading…
Reference in New Issue