mirror of
https://github.com/kennethreitz/tablib.git
synced 2026-06-05 23:10:17 +00:00
Kenneth Reitz
86 lines
1.9 KiB
Python
86 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
""" Tablib - XLS Support.
|
|
"""
|
|
|
|
import cStringIO
|
|
|
|
try:
|
|
import xlwt
|
|
except ImportError:
|
|
import tablib.packages.xlwt as xlwt
|
|
|
|
|
|
title = 'xls'
|
|
extentions = ('xls',)
|
|
|
|
# special styles
|
|
wrap = xlwt.easyxf("alignment: wrap on")
|
|
bold = xlwt.easyxf("font: bold on")
|
|
|
|
|
|
def export_set(dataset):
|
|
"""Returns XLS representation of Dataset."""
|
|
|
|
wb = xlwt.Workbook(encoding='utf8')
|
|
ws = wb.add_sheet(dataset.title if dataset.title else 'Tabbed Dataset')
|
|
|
|
dset_sheet(dataset, ws)
|
|
|
|
stream = cStringIO.StringIO()
|
|
wb.save(stream)
|
|
return stream.getvalue()
|
|
|
|
|
|
def export_book(databook):
|
|
"""Returns XLS representation of DataBook."""
|
|
|
|
wb = xlwt.Workbook(encoding='utf8')
|
|
|
|
for i, dset in enumerate(databook._datasets):
|
|
ws = wb.add_sheet(dset.title if dset.title else 'Sheet%s' % (i))
|
|
|
|
dset_sheet(dset, ws)
|
|
|
|
|
|
stream = cStringIO.StringIO()
|
|
wb.save(stream)
|
|
return stream.getvalue()
|
|
|
|
|
|
def dset_sheet(dataset, ws):
|
|
"""Completes given worksheet from given Dataset."""
|
|
_package = dataset._package(dicts=False)
|
|
|
|
for i, sep in enumerate(dataset._separators):
|
|
_offset = i
|
|
_package.insert((sep[0] + _offset), (sep[1],))
|
|
|
|
for i, row in enumerate(_package):
|
|
for j, col in enumerate(row):
|
|
|
|
# bold headers
|
|
if (i == 0) and dataset.headers:
|
|
ws.write(i, j, col, bold)
|
|
|
|
# frozen header row
|
|
ws.panes_frozen = True
|
|
ws.horz_split_pos = 1
|
|
|
|
|
|
# bold separators
|
|
elif len(row) < dataset.width:
|
|
ws.write(i, j, col, bold)
|
|
|
|
# wrap the rest
|
|
else:
|
|
try:
|
|
if '\n' in col:
|
|
ws.write(i, j, col, wrap)
|
|
else:
|
|
ws.write(i, j, col)
|
|
except TypeError:
|
|
ws.write(i, j, col)
|
|
|
|
|