python --写入csv文件(原生)

python --写入csv文件(原生)

class ExportExcel(object):
    '''日志导出excel'''

    def __init__(self, queryset):
        self._queryset = queryset

    @property
    def queryset(self):
        return self._queryset

    @queryset.setter
    def queryset(self, value):
        self._queryset = value

    @property
    def log_path(self):
        return os.path.join(os.path.join(self.base_dir, 'static'), 'logs')

    @property
    def base_dir(self):
        return os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

    @property
    def file_name(self):
        return time.strftime('%Y%m%d%H%M%S_logs.csv')

    @property
    def server(self):
        return 'http://192.168.1.60:9001/static/logs/'

    def write(self): # 批量写入   csvwriter.writerow(row)为单行写入
        _name = self.file_name
        with open(_name, 'wb') as f:
            w = csv.writer(f)
            w.writerow(['用户', '用户id', '创建时间', 'ip', '操作内容'])
            w.writerows([
                [i.username, i.user_id, i.ctime.strftime('%Y-%m-%d %H:%M:%S'), i.ip,
                 i.operation_content] for i in self._queryset])
        old, new = os.path.join(self.base_dir, _name), os.path.join(self.log_path, _name)
        print '{} --> {}'.format(old, new)
        shutil.move(old, new)
        return self.server + _name

附录

import csv

with open('abc.csv', 'w', newline='') as csvfile:
    w = csv.writer(csvfile)
    # 写入列头
    w.writerow(["sid", "name", "age"])
    w.writerow(["10010", "橡皮擦", "18"])
    w.writerow(["10086", "发量迷人的乔喻", "18"])

这时如果你在打开文件时，缺少了 newline=‘’，那写入 csv 文件内容之后，会出现多余的换行。除此之外，我们写入 csv 文件的时候，可以启用 dialect（方言）参数。例如使用 | 作为列的分隔符。此时的文件写入代码写成下述内容：

class my_dialect(csv.Dialect):
    lineterminator = '\r\n'
    delimiter = ';'
    quotechar = '"'
    quoting = csv.QUOTE_MINIMAL
 
 
with open('abc.csv', 'w', newline='') as csvfile:
    w = csv.writer(csvfile, dialect=my_dialect)
    # 写入列头
    w.writerow(["sid", "name", "age"])
    w.writerow(["10010", "橡皮擦", "18"])
    w.writerow(["10086", "发量迷人的乔喻", "18"])

上述代码定义了一个新的类 my_dialect，并使其继承 csv.Dialect 类，重写了其中的部分字段。

delimiter：分隔字符，默认为 ,；- lineterminator：writer 方法写入数据时，每行的结尾字符，默认为 \r\n；- quotechar：单字符，用于包裹具有特殊字符的字段，例如 定界符，引号字符，换行符，默认是 "；- quoting：控制 writer 何时生成引号，以及 reader 何时识别引号，默认是 QUOTE_MINIMAL，其余几个值分别是 QUOTE_ALL（全部加引号），QUOTE_MINIMAL（特定情况加），QUOTE_NONNUMERIC（所有非数字加），QUOTE_NONE（都不加）；Writer 对象的方法说明：

csvwriter.writerow(row)：写入单行；- csvwriter.writerows(rows)：写入多行