Bootstrap Table Helper

BootstrapTable.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# author: Panagiotis Mavrogiorgos
# email: gmail, pmav99

"""
A help class for generating bootstrap tables.
"""

from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
from __future__ import absolute_import

from jinja2 import Template

table_template = """
<table class="{{ table_classes }}">
  {%- if caption %}<caption>{{ caption }}</caption>{% endif -%}
  {# If there is a header row, display it! #}
  {%- if headers %}
  <thead>
    <tr class="">
    {%- for header in headers %}
      <th>{{ header }}</th>
    {%- endfor %}
    </tr>
  </thead>
  {%- endif %}
  <tbody>
  {%- for row in rows %}
    <tr class="">
      {%- for cell in row %}
      <td>{{cell}}</td>
      {%- endfor %}
    </tr>
  {%- endfor %}
  </tbody>
  {# If there is a footer row, display it! #}
  {%- if footers -%}
  <tfoot>
    <tr class="">
    {%- for footer in footers %}
      <th class="">{{ footer }}</th>
    {%- endfor %}
    </tr>
  </tfoot>
  {%- endif %}
</table>
"""


class BootstrapTable(object):
    pad_value = "&nbsp;"

    def __init__(self, rows, headers=None, footers=None, caption=None,
                 hover=False, condensed=False, bordered=False, stripped=False, row_padding=True):
        self.template = Template(table_template)
        self.caption = caption
        self.headers = headers
        self.footers = footers
        self.rows = rows

        # Determine table classes.
        self.table_classes = table_classes = ["table"]
        if hover:
            table_classes.append("table-hover")
        if condensed:
            table_classes.append("table-condensed")
        if bordered:
            table_classes.append("table-bordered")
        if stripped:
            table_classes.append("table-stripped")
        self.table_classes = " ".join(table_classes)

        if row_padding:
            self.pad_rows()

    def pad_rows(self):
        """
        Appends <tds> with "&nbsp;" to the <tfoot>, <thead> and the <tbody> rows.
        We do that in order to maintain visual uniformity.
        """
        nbsp = "&nbsp;"
        max_number_of_columns = self.calc_max_number_of_columns()
        if len(self.headers) < max_number_of_columns:
            self.headers.append(nbsp)
        for row in self.rows:
            while len(row) < max_number_of_columns:
                row.append(nbsp)
        if len(self.footers) < max_number_of_columns:
            self.footers.append(nbsp)

    def calc_max_number_of_columns(self):
        """ Returns the max number of columns between the <thead>, the <tfoot> and the <tbody> rows. """
        data_rows = [self.headers, self.footers] + self.rows
        return max(map(len, (data_rows)))

    def render(self):
        return self.template.render(
            table_classes=self.table_classes,
            caption=self.caption,
            headers=self.headers,
            footers=self.footers,
            rows=self.rows
        )


B = BootstrapTable(
    headers=["a", "b", "c", "d", "e"],
    condensed=True,
    stripped=True,
    rows=[
        [1, 2, 3, 4],
        [5, 6, 7],
        [8, 9, 10, 11],
    ],
    footers=[14, 17, 20, 15],
)

print(B.render())