未验证 提交 3f796510 编写于 作者: P pukkandan

[utils] HTTPHeaderDict: Handle byte values

上级 de20687e
......@@ -2344,6 +2344,8 @@ def test_traverse_obj(self):
def test_http_header_dict(self):
headers = HTTPHeaderDict()
headers['ytdl-test'] = b'0'
self.assertEqual(list(headers.items()), [('Ytdl-Test', '0')])
headers['ytdl-test'] = 1
self.assertEqual(list(headers.items()), [('Ytdl-Test', '1')])
headers['Ytdl-test'] = '2'
......
......@@ -65,6 +65,8 @@ def __init__(self, *args, **kwargs):
self.update(kwargs)
def __setitem__(self, key, value):
if isinstance(value, bytes):
value = value.decode('latin-1')
super().__setitem__(key.title(), str(value))
def __getitem__(self, key):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册