Python PrettyTable在同一行打印表

Question

我使用PrettyTable'from_db_cursor'模块来很好地打印我从sql请求中得到的响应。一切都很好，我唯一的问题是，对于一些我想打印的查询，它们被附加到同一行，如下所示：

countUnivers1:    countUnivers2:    countUnivers3:
+----------+      +----------+      +----------+
| COUNT(*) |      | COUNT(*) |      | COUNT(*) |
+----------+      +----------+      +----------+
| 1681943  |      |  51954   |      | 4140984  |
+----------+      +----------+      +----------+

但我无法找到如何做到这一点，这是我目前使用的代码：

i = 0
tables = []
with open(output, 'w+') as file:
    file.write(str(current_time) + 'n')
    for query in QUERIES.items():
        cur.execute(query[1])
        table = from_db_cursor(cur)
        if not re.search('countUnivers' ,query[0]):
            file.write('n' + query[0] + ':n')
            file.write(str(table) + 'n')
        else:
            if i < 6:
                file.write(query[0] + ':t')
                tables.append(str(table))
                i += 1
            elif i == 6:
                file.write('n')
                for t in tables:
                    file.write(str(table) + 't')
                i = 0
                tables = []
    file.write('nDatabase:n' + json.dumps(dbParams, indent=4) + 'n')

此代码输出我：

countUnivers1:  countUnivers2:  countUnivers3:
+----------+
| COUNT(*) |
+----------+
| 1681943  |
+----------+    +----------+
| COUNT(*) |
+----------+
|  51954   |
+----------+    +----------+
| COUNT(*) |
+----------+
| 4140984  |
+----------+

QUERIES这里是一个OrderedDict，里面装满了SQL请求：

('countUnivers1', "SELECT COUNT(*) 
                        FROM ourson_offer o 
                        WHERE o.cat LIKE '1%' 
                        AND CHARACTER_LENGTH(o.cat) = 7"),

就像这个。

Answer 1

文本是二维的：行+行中的字符。"table"是一个python字符串，其中有新行——它是一个文本！每行结束字符（CRLF、CR或LF，具体取决于操作系统/平台）后会开始一行新行。

您想要做的事情需要将字符串的结构理解为文本，并垂直对齐新的块。

def pad_lines_vertically(lines, size):
    ''' List of lines of exactly `size` length.
    Extended with empty lines if needed.
    '''
    orig_lines = list(lines)
    assert size >= len(orig_lines)
    return orig_lines + [''] * (size - len(orig_lines))
def pad_lines_horizontally(lines):
    ''' Pad lines to the lenght of the longest line.
    '''
    line_length = max(len(line) for line in lines)
    return [
        line.ljust(line_length)
        for line in lines
    ]
def text_add(text1, text2, padding=' '):
    lines1 = text1.splitlines()
    lines2 = text2.splitlines()
    line_count = max(len(lines1), len(lines2))
    def pad_lines(lines):
        return pad_lines_horizontally(
            pad_lines_vertically(lines, line_count)
        )
    return 'n'.join(
        ''.join(line1 + padding + line2)
        for line1, line2 in zip(pad_lines(lines1), pad_lines(lines2))
    )

像这样使用：

text1 = '''
countUnivers1:
+----------+
| COUNT(*) |
+-------
| 1681943  |
+----------+
'''
text2 = '''
countUnivers2:
+----------+
| COUNT(*) |
+----------+
|  51954   |
+----------+
 plus a
 comment
'''
print text_add(text1, text2, padding='t')

countUnivers1:  countUnivers2:
+----------+    +----------+
| COUNT(*) |    | COUNT(*) |
+-------        +----------+
| 1681943  |    |  51954   |
+----------+    +----------+
                 plus a
                 comment