Textbase loader for visidata.org

textbase.py

from visidata import Column, TableSheet


def open_dmp(p):
    return TextBaseSheet(p.name, source=p)


class TextBaseSheet(TableSheet):
    rowtype = "records"  # rowdef: a list, of collections.OrderedDict objects

    def iterload(self):
        import textbase

        for line in textbase.parse(self.source.given):
            for col in line:
                c = Column(
                    col, getter=lambda col, row: "\n".join(row[0].get(col.name, []))
                )
                if c not in self.columns:
                    self.addColumn(c)
            yield [line]