Table

Complexity

Total Complexity

11

Size/Duplication

Total Lines	62
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	add_row()	0	17	4
A	add_by_list()	0	14	3
A	__init__()	0	4	1
A	set_header()	0	10	1
A	sort()	0	6	1
A	__str__()	0	3	1

#
# Copyright (c) 2015 SUSE Linux GmbH
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of version 3 of the GNU General Public License as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, contact SUSE LLC.
#
# To contact SUSE about this file by physical or electronic mail,
# you may find current contact information at www.suse.com

import prettytable

class Table(object):
    """A Table instance represents the result of the search
    """
    def __init__(self):
        """Initialize Table class"""
        self.__table = prettytable.PrettyTable()
        self.__header = []

    def add_row(self, name, values):
        """Add row into table

        :param str name:
        :param dict values:
        """
        row = [name]
        for key in self.__header:
            #ignore the Files item we have the name already
            if key != "Files":
                value = (values.get(key))
                #has the file this value?
                if value is not None:
                    row.append(value)
                else:
                    row.append("-")
        self.__table.add_row(row)

    def set_header(self, keys):
        """Set header of table

        :param str keys:
        """
        #prepare header first column is filename + keys
        header = ["Files"]
        header = header + list(keys)
        self.__table.field_names = header
        self.__header = header

    def add_by_list(self, file_list):
        """TODO:

        :param list file_list:
        """
        #first gather all header items
        keys = set()
        for filename, values in file_list.items():
            keys.update(values.keys())
        self.set_header(keys)

        #now we can create the rows
        for filename, values in file_list.items():
            self.add_row(filename, values)

    def sort(self, sortby):
        """Sort table

        :param sortby:
        """
        self.__table.sortby = sortby

    def __str__(self):
        """Return formatted table"""
        return self.__table