本文整理汇总了Python中six.BytesIO.truncate方法的典型用法代码示例。如果您正苦于以下问题：Python BytesIO.truncate方法的具体用法？Python BytesIO.truncate怎么用？Python BytesIO.truncate使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类six.BytesIO的用法示例。
在下文中一共展示了BytesIO.truncate方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: CsvDictsAdapter
# 需要导入模块: from six import BytesIO [as 别名]
# 或者: from six.BytesIO import truncate [as 别名]
class CsvDictsAdapter(object):
"""Provide a DataChange generator and it provides a file-like object
which returns csv data.
"""
def __init__(self, source_generator):
self.source = source_generator
self.buffer = BytesIO()
self.csv = None
self.add_header = False
def __iter__(self):
return self
def write_header(self):
self.add_header = True
def next(self):
row = self.source.next()
self.buffer.truncate(0)
self.buffer.seek(0)
if not self.csv:
self.csv = csv.DictWriter(self.buffer, row.keys(), quoting=csv.QUOTE_NONNUMERIC)
self.add_header = True
if self.add_header:
if hasattr(self.csv, 'writeheader'):
self.csv.writeheader()
else:
self.csv.writerow(dict((fn, fn) for fn in self.csv.fieldnames))
self.add_header = False
self.csv.writerow(row)
self.buffer.seek(0)
return self.buffer.read()