# -*- coding: utf-8 -*-
import os
import time
import csv
class WriteCSV():
"""定义成员变量"""
def __init__(self):
pass
def init(self, info_list, info_dict_list):
self.info_list = info_list
self.info_dict_list = info_dict_list
def getBaseDir(self):
baseDir = r'D:\tempcmd\migrate'
return baseDir
def getTimestamp(self):
timestamp = time.strftime('%Y%m%d%H%M%S', time.localtime())
return timestamp
def writeFile(self):
pass
def writeFileDict(self):
filename = self.getBaseDir() + os.sep + 'tmp' + os.sep + self.getTimestamp() + '.csv'
header = ['姓名', '年龄', '性别', '籍贯', '电话', '备注'] info_list = [{'姓名': '赵四', '年龄': 36, '性别': '男', '籍贯': '铁岭'},
{'姓名': '刘能', '年龄': 38, '性别': '男', '籍贯': '铁岭'}] with open(filename, 'w', newline='', encoding='utf-8') as f:
dictWriter = csv.DictWriter(f, fieldnames=header)
dictWriter.writeheader()
for item in info_list:
dictWriter.writerow(item)
# 引用导入
# writer = WriteCSV()
if __name__ == '__main__':
writer = WriteCSV()
writer.writeFileDict()
)
)
