Merge pull request #1 from mic1on/perf_utils

🎈 perf(utils): `cookie_to_dict` `headers_to_dict` `data_to_dict`

Author: mic1on

example/use_to.py

@@ -3,14 +3,14 @@
 
 def demo_to():
     print("demo_to")
-    print(useTo.to_string('https://www.baidu.com'))
-    print(useTo.to_string(b'https://www.baidu.com'))
-    print(useTo.to_string(123))
-    print(useTo.to_string(123.456))
-    print(useTo.to_string(True))
-    print(useTo.to_string(None))
-    print(useTo.to_string([1, 2, 3]))
-    print(useTo.to_string({'a': 1, 'b': 2, 'c': 3}))
+    print(useTo.string('https://www.baidu.com'))
+    print(useTo.string(b'https://www.baidu.com'))
+    print(useTo.string(123))
+    print(useTo.string(123.456))
+    print(useTo.string(True))
+    print(useTo.string(None))
+    print(useTo.string([1, 2, 3]))
+    print(useTo.string({'a': 1, 'b': 2, 'c': 3}))
     pass
 
 

src/usepy/utils/utils.py

@@ -13,7 +13,7 @@ def cookie_to_dict(cookies: str) -> dict:
     :param cookies: cookie字符串
     :return: dict
     """
-    return {cookie.split('=')[0]: cookie.split('=')[-1] for cookie in cookies.split('; ')}
+    return dict(x.split('=') for x in cookies.split('; '))  # noqa
 
 
 def headers_to_dict(headers: str) -> dict:
@@ -22,12 +22,7 @@ def headers_to_dict(headers: str) -> dict:
     :param headers: headers字符串
     :return: dict
     """
-    header_dict = {}
-    for line in headers.split('\n'):
-        if ':' in line:
-            key, value = line.split(':', 1)
-            header_dict[key.strip()] = value.strip()
-    return header_dict
+    return dict(map(lambda x: x.strip(), line.split(':')) for line in headers.split('\n') if ':' in line)  # noqa
 
 
 def data_to_dict(data: str) -> dict:
@@ -36,7 +31,7 @@ def data_to_dict(data: str) -> dict:
     :param data: data字符串。格式为`key1=value1&key2=value2`
     :return: dict
     """
-    return {item.split('=')[0]: item.split('=')[-1] for item in data.split('&')}
+    return dict(x.split('=') for x in data.split('&'))  # noqa
 
 
 def gen_unique_id():

tests/test_utils/test_utils.py

@@ -33,6 +33,14 @@ def test_cookie_to_dict():
     }
 
 
+def test_data_to_dict():
+    data_str = "key1=value1&key2=value2"
+    assert useDataToDict(data_str) == {
+        "key1": "value1",
+        "key2": "value2",
+    }
+
+
 def test_header_to_dict():
     header_str = """sec-ch-ua: "Not?A_Brand";v="8", "Chromium";v="108", "Microsoft Edge";v="108"
     sec-ch-ua-platform: "macOS"