save_excel.py

from visidata import *

@VisiData.api
def save_xlsx(vd, p, vs, *vsheets):
    import openpyxl

    wb = openpyxl.Workbook()
    wb.remove_sheet(wb['Sheet'])

    for vs in vsheets or [vs]:
        ws = wb.create_sheet(title=vs.name)

        headers = [col.name for col in vs.visibleCols]
        ws.append(headers)

        for dispvals in vs.iterdispvals(format=False):

            row = []
            for v in dispvals.values():
                if type(v) == date:
                    v = datetime.datetime.fromtimestamp(int(v.timestamp()))
                elif not type(v) in [int,float,str]:
                    v = str(v)
                row.append(v)
            ws.append(row)

    wb.active = ws

    wb.save(filename=p)
    status(f'{p} save finished')


@VisiData.api
def save_xls(vd, p, vs, *vsheets):
    import xlwt

    wb = xlwt.Workbook()

    for vs in vsheets or [vs]:

        ws1 = wb.add_sheet(vs.name[:32]) # sheet name can not be longer than 32 characters

        for col_i, col in enumerate(vs.visibleCols):
            ws1.write(0, col_i, col.name)

        for r_i, dispvals in enumerate(vs.iterdispvals(format=False)):
            r_i += 1
            for c_i, v in enumerate(dispvals.values()):

                if not type(v) in [int,float,str]:
                    v = str(v)

                ws1.write(r_i, c_i, v)

    wb.save(p)
    status(f'{p} save finished')