首先,使用除模块之外的任何技术生成 CSV 数据csv
不会为您赢得很多朋友;这是完成这项工作的正确工具。
使用该模块生成输出的最方便writer.writerows()
的方法是 with ,它采用嵌套列表(或者更确切地说,是序列的迭代器)。生成器是一种很好的表达方式:
def csv_spec(queryset):
for employee in queryset:
yield (
employee.id,
employee.manager.first_name,
employee.user.last_login,
employee.contact.phone_number)
w = csv.writer(outfile)
w.writerows(csv_spec(something))
编辑:所以你说你想要一个模板之类的东西,原因与我们使用模板的原因相同,但它产生的是 CSV 而不是文本。您可以要求 jinja 解析和处理它在占位符中的各种表达式,而无需完整的模板。
假设我们有一些数据:
>>> from collections import namedtuple
>>> from datetime import datetime
>>> Employee = namedtuple('Employee', 'id manager user contact')
>>> Person = namedtuple('Person', 'first_name last_name')
>>> Account = namedtuple('Account', 'username password last_login')
>>> Contact = namedtuple('Contact', 'phone_number email')
>>> data = [
... Employee(0, Person('Alice', 'Brown'),
... Account('abrown', 'secret', datetime(2013, 3, 3)),
... Contact('5551234', 'abrown@example.com')),
... Employee(1, Person('Bob', 'Jones'),
... Account('bjones', 'safe', datetime(2013, 3, 3)),
... Contact('5555678', 'bjones@example.com')),
... Employee(2, Person('Carol', 'Smith'),
... Account('csmith', 'hidden', datetime(2013, 3, 3)),
... Contact('5559012', 'csmith@example.com'))]
和一个模板(你可以很容易地从文件中读取它)
>>> import StringIO
>>> template = StringIO.StringIO('''employee.id
... employee.manager.first_name
... employee.user.last_login
... employee.contact.phone_number''')
要使用表达式编译器,您需要一个环境;无需指定选项:
>>> import jinja2
>>> env = jinja2.Environment()
>>> template_filter = map(env.compile_expression, template)
正如我提到的,使用 csv 模块:
>>> import csv
>>> import sys
>>> writer = csv.writer(sys.stdout)
最后,使用编译后的表达式,我们可以一次运行一个“行”数据,并且对于每一行,应用每个表达式
>>> writer.writerows([[field(employee=row) for field in template_filter]
... for row in data])
0,Alice,2013-03-03 00:00:00,5551234
1,Bob,2013-03-03 00:00:00,5555678
2,Carol,2013-03-03 00:00:00,5559012