# _*_ coding: utf-8 _*_

"""
python_csv.py by xianhu
"""

import csv
import datetime

# 数据
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)],
    [4, "zh'n", 11.937, datetime.datetime(2012, 1, 9)],
    [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="mydialect")
    # writer.writerows(data)
    for item in data:
        writer.writerow(item)
exit()

# 读文件
with open("test.csv", "r") as file:
    reader = csv.reader(file, dialect="excel")
    for item in reader:
        print(item)

# 读文件
with open("test.csv", "r") as file:
    reader = csv.DictReader(file, fieldnames=["id", "name", "float", "datetime"], dialect="excel")
    data = [item for item in reader]
    print(data)

# 写文件
with open("test.csv", "w") as file:
    writer = csv.DictWriter(file, fieldnames=["id", "name", "float", "datetime"], dialect="excel")
    writer.writeheader()
    for item in data:
        writer.writerow(item)