fix(tools): fix fix_encode error

spider_brew_kit/tools/text_encode.py

@@ -1,16 +1,20 @@
 from spider_brew_kit.contants import ENCODINGS
 
 
-def fix_encode(garbled_text: str, decoding='utf-8') -> (str, str):
+def fix_encode(garbled_text: str, decoding='utf-8', return_encoding=False):
     """
     修复乱码
     :param garbled_text: 乱码文本
     :param decoding: 解码方式
+    :param return_encoding: 是否返回编码
     :return: 正常文本, 编码
     """
     for encoding in ENCODINGS:
         try:
             res = garbled_text.encode(encoding).decode(decoding)
-            return res, encoding
+            if return_encoding:
+                return res, encoding
+            else:
+                return res
         except (UnicodeEncodeError, UnicodeDecodeError):
             continue

test/unit/tools/text_encode.py

@@ -6,5 +6,9 @@ def test_fix_encode(self):
         garbled_text = "亿元"
         expected_text = "亿元"
         expected_encoding = "cp1252"
+
         result = fix_encode(garbled_text)
+        assert result == expected_text
+
+        result = fix_encode(garbled_text, return_encoding=True)
         assert result == (expected_text, expected_encoding)