Added tableReport class

pym/cl_string.py

@@ -17,6 +17,11 @@
 from re import search, compile, S
 from cl_utils import _toUNICODE
 import cl_overriding
+import cl_lang
+import sys
+tr = cl_lang.lang()
+tr.setLocalDomain('cl_lib')
+tr.setLanguage(sys.modules[__name__])
 
 def prettyColumnStr(*cols):
     '''Функция преобразования строк в текстовые колонки. Если указанный текст
@@ -115,7 +120,7 @@ def prettyColumnStr(*cols):
                     partstr = partstr.ljust(cols[q+1], ' ')
                     cols[q] = ''
 
-                retstr+= partstr + cellspacing
+                retstr+= partstr + cellspacingS
 
                 # остальную часть строки оставить на следующую итерацию
                 # если от строки что то осаталось
@@ -256,3 +261,104 @@ def justify(s,width):
         pos += 3
     # вернуть строку в utf8 если она пришла в utf8
     return s.encode('utf-8')
+
+class tableReport:
+    """Класс для выдачи данных в табличном виде"""
+    def __init__(self, title, headerList, dataList):
+        # Заголовок
+        self.title = title
+        # Cписок элементов первой строки таблицы
+        self.headerList = headerList
+        # Список строк (каждая строка - список элементов)
+        self.dataList = dataList
+        # Список ширин колонок
+        self.columnsWidth = self.getColumsnWidth()
+
+    def getColumsnWidth(self):
+        """Находит максимальную ширину каждой колонки
+
+        результат список ширин колонок
+        """
+        columnsWidth = []
+        lenCol = 0 
+        for s in self.headerList:
+            lenCol += 1
+            columnsWidth.append(len(_toUNICODE(s)))
+        maxLenCol = lenCol
+        # Вычисляем максимальное количество столбцов
+        for s in self.dataList:
+            lenS = len(s)
+            if maxLenCol < lenS:
+                maxLenCol = lenS
+        if maxLenCol > lenCol:
+            appCol = maxLenCol - lenCol
+            # Добавляем элементы в список ширин
+            for i in range(appCol):
+                columnsWidth.append(0)
+        # Вычисляем ширину столбцов
+        for e in self.dataList:
+            i = 0
+            for s in e:
+                lenS = len(_toUNICODE(s))
+                if columnsWidth[i] < lenS:
+                    columnsWidth[i] = lenS
+                i += 1
+        return columnsWidth
+
+    def createFormatStr(self, listStr):
+        """Создает список (текст, ширина ...)"""
+        strList = []
+        lenStr = len(listStr)
+        for i in range(len(self.columnsWidth)):
+            if lenStr > i:
+                strList.append(listStr[i])
+            else:
+                strList.append("")
+            strList.append(self.columnsWidth[i])
+        return strList
+
+    def printReport(self):
+        """Напечатать данные в табличном виде"""
+        cl_overriding.printSUCCESS(self.title)
+        listStrSep = []
+        for lenCol in self.columnsWidth:
+            listStrSep.append("-"*lenCol)
+        printData = ""
+        printData += columnStr(*self.createFormatStr(listStrSep))
+        printData += columnStr(*self.createFormatStr(self.headerList))
+        printData += columnStr(*self.createFormatStr(listStrSep))
+        for s in self.dataList:
+            printData += columnStr(*self.createFormatStr(s))
+        printData += columnStr(*self.createFormatStr(listStrSep))
+        if printData[-1] == "\n":
+            printData = printData[:-1]
+        lines = printData.splitlines()
+        lenCols = map(lambda x: len(x), lines[0].strip().split(" "))
+        convLines = []
+        lenLines = len(lines)
+        for i in range(lenLines):
+            char = " | "
+            if i == 0 or i == 2 or i == lenLines-1:
+                char ="-+-"
+            convLines.append(self._insertStrChar(lines[i], lenCols, char))
+        cl_overriding.printSUCCESS("\n".join(convLines))
+        cl_overriding.printSUCCESS("(%s %s)"%(len(self.dataList), _("rows")))
+
+    def _insertStrChar(self, line, lenCols, char):
+        """Вставляет несколько символов char в указанные позиции
+
+        сначала строки будет вставлено char[1:] в конце строки char[:-1]
+        """
+        lineUnicode = _toUNICODE(line)
+        prevPos = 0
+        convLine = char[1:]
+        lenLenCols = len(lenCols)
+        for i in range(lenLenCols):
+            pos = lenCols[i] + prevPos
+            if i == lenLenCols-1:
+                insertChar = char[:-1]
+            else:
+                insertChar = char
+            convLine += lineUnicode[prevPos:pos] + insertChar
+            prevPos = pos + 1
+        return convLine.encode("UTF-8")