CSV操作增加自定义dialect

xianhu · xianhu · commit 12b1ecaaa454 · 2018-06-28T14:15:59.000+08:00
diff --git a/python_csv.py b/python_csv.py
@@ -11,17 +11,29 @@
 data = [
     [1, "a,bc", 19.353, datetime.datetime(2001, 3, 17)],
     [2, "ei,f", 13.287, datetime.datetime(2011, 4, 27)],
-    [3, "q\"ij", 15.852, datetime.datetime(2003, 7, 14)],
+    [3, 'q"ij', 15.852, datetime.datetime(2003, 7, 14)],
     [4, "zh'n", 11.937, datetime.datetime(2012, 1, 9)],
-    [5, "i\'op", 12.057, datetime.datetime(2009, 5, 18)],
+    [5, "i'op", 12.057, datetime.datetime(2009, 5, 18)],
 ]
 
+# 自己创建dialect
+csv.register_dialect(
+    "mydialect",
+    delimiter=',',              # 字段分隔符
+    escapechar='\\',            # 转义字符
+    quotechar='"',              # 包裹字符
+    doublequote=False,          # 使转义字符生效
+    lineterminator='\n',        # 行与行之间的分隔符
+    quoting=csv.QUOTE_ALL       # 包裹模式
+)
+
 # 写文件
 with open("test.csv", "w") as file:
-    writer = csv.writer(file, dialect="excel")
+    writer = csv.writer(file, dialect="mydialect")
     # writer.writerows(data)
     for item in data:
         writer.writerow(item)
+exit()
 
 # 读文件
 with open("test.csv", "r") as file:
