更新逐行文件读取工具类支持csv

zhanghe06 · zhanghe06 · commit 3e42f1e506e2 · 2015-11-24T11:01:54.000+08:00
diff --git a/tools/html.py b/tools/html.py
@@ -2,6 +2,7 @@
 __author__ = 'zhanghe'
 
 import re
+import json
 
 
 def replace_html(input_html, reg_expression=r'', replace_text=''):
@@ -82,15 +83,19 @@ def read_file(file_path):
         return f.read()
 
 
-def read_file_each_line(file_path):
+def read_file_each_line(file_path, file_type='json'):
     """
     &#36880;&#34892;&#35835;&#21462;&#25991;&#20214;
     :param file_path:
+    :param file_type:
     :return:
     """
     with open(file_path, 'r') as f:
         for each_line in f:
-            yield each_line
+            if file_type == 'json':
+                yield json.loads(each_line)
+            if file_type == 'csv':
+                yield each_line
 
 
 def save_file(file_path, text, save_type='w'):
