github/gallery-dl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[formatter] support putting keys in quotes

Browse Source 
i.e. obj["key"] or obj['key']
as in f-strings
This commit is contained in:
Mike Fährmann
2023-03-21 22:06:54 +01:00
parent 46fdf46f21
commit fe41a2b159
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gallery_dl/formatter.py
View File

@@ -256,6 +256,8 @@ def parse_field_name(field_name):
try: try:
if ":" in key: if ":" in key:
key = _slice(key) key = _slice(key)
else:
key = key.strip("\"'")
except TypeError: except TypeError:
pass # key is an integer pass # key is an integer

5
test/test_formatter.py
View File

@@ -128,6 +128,11 @@ class TestFormatter(unittest.TestCase):
self._run_test("{l[0]}" , "a") self._run_test("{l[0]}" , "a")
self._run_test("{a[6]}" , "w") self._run_test("{a[6]}" , "w")
def test_dict_access(self):
self._run_test("{d[a]}" , "foo")
self._run_test("{d['a']}", "foo")
self._run_test('{d["a"]}', "foo")
def test_slicing(self): def test_slicing(self):
v = self.kwdict["a"] v = self.kwdict["a"]
self._run_test("{a[1:10]}" , v[1:10]) self._run_test("{a[1:10]}" , v[1:10])