最近在用 Python 寫一個小工具,這個工具主要就是用來管理各種資源的信息,比如阿里云的 ECS 等信息,因為我工作的電腦使用的是 LINUX,所以就想著用 python 寫一個命令行的管理工具,基本的功能就是同步阿里云的資源的信息到數(shù)據(jù)庫,然后可以使用命令行查詢。因為信息是展現(xiàn)在命令行中的,眾所周知,命令行展現(xiàn)復(fù)雜的文本看起來著實累人,于是就想著能像表格那樣展示,那看起來就舒服多了。prettytable 庫就是這么一個工具,prettytable 可以打印出美觀的表格,并且對中文支持相當(dāng)好(如果有試圖自己實現(xiàn)打印表格,你就應(yīng)該知道處理中文是多么的麻煩)prettytable 并非 python 的內(nèi)置庫,通過 pip install prettytable
即可安裝。
三、一個小示例
#!/usr/bin/python
#**coding:utf-8**
import sys
from prettytable import PrettyTable
reload(sys)
sys.setdefaultencoding('utf8')
table = PrettyTable(['編號','云編號','名稱','IP地址'])
table.add_row(['1','server01','服務(wù)器01','172.16.0.1'])
table.add_row(['2','server02','服務(wù)器02','172.16.0.2'])
table.add_row(['3','server03','服務(wù)器03','172.16.0.3'])
table.add_row(['4','server04','服務(wù)器04','172.16.0.4'])
table.add_row(['5','server05','服務(wù)器05','172.16.0.5'])
table.add_row(['6','server06','服務(wù)器06','172.16.0.6'])
table.add_row(['7','server07','服務(wù)器07','172.16.0.7'])
table.add_row(['8','server08','服務(wù)器08','172.16.0.8'])
table.add_row(['9','server09','服務(wù)器09','172.16.0.9'])
print(table)
linuxops@deepin:~$ python p.py
+------+----------+----------+------------+
| 編號 |云編號| 名稱 | IP地址 |
+------+----------+----------+------------+
|1 | server01 | 服務(wù)器01 | 172.16.0.1 |
|2 | server02 | 服務(wù)器02 | 172.16.0.2 |
|3 | server03 | 服務(wù)器03 | 172.16.0.3 |
|4 | server04 | 服務(wù)器04 | 172.16.0.4 |
|5 | server05 | 服務(wù)器05 | 172.16.0.5 |
|6 | server06 | 服務(wù)器06 | 172.16.0.6 |
|7 | server07 | 服務(wù)器07 | 172.16.0.7 |
|8 | server08 | 服務(wù)器08 | 172.16.0.8 |
|9 | server09 | 服務(wù)器09 | 172.16.0.9 |
+------+----------+----------+------------+
在以上的示例中,我們通過form
導(dǎo)入了表格庫。 table
實例化了一個表格庫,并且添加了['編號','云編號','名稱','IP地址']
為表頭,如果沒有添加表頭,那么會以默認(rèn)的Field+編號顯示,例如:+---------+----------+----------+------------+
| Field 1 | Field 2| Field 3|Field 4 |
+---------+----------+----------+------------+
四、添加數(shù)據(jù)
prettytable提供了多種的添加數(shù)據(jù)的方式,最常用的應(yīng)該就是按行按列添加數(shù)據(jù)了。A、按行添加數(shù)據(jù) table.add_row
在上面簡單的示例中,我們就是按行添加數(shù)據(jù)的。添加的數(shù)據(jù)必須要是列表的形式,而且數(shù)據(jù)的列表長度要和表頭的長度一樣。在實際的使用中,我們應(yīng)該要關(guān)注到添加的數(shù)據(jù)是否和表頭對應(yīng),這一點很重要。B、按列添加數(shù)據(jù) table.add_column()
#!/usr/bin/python
#**coding:utf-8**
import sys
from prettytable import PrettyTable
reload(sys)
sys.setdefaultencoding('utf8')
table = PrettyTable()
table.add_column('項目', ['編號','云編號','名稱','IP地址'])
table.add_column('值', ['1','server01','服務(wù)器01','172.16.0.1'])
print(table)
+-------+--------+------------+
| index | 項目 |值 |
+-------+--------+------------+
| 1 |編號| 1|
| 2 | 云編號 |server01|
| 3 |名稱|服務(wù)器01 |
| 4 | IP地址 | 172.16.0.1 |
+-------+--------+------------+
以上示例中,我們通過add_column
來按列添加數(shù)據(jù),按列添加數(shù)據(jù)不需要在實例化表格的時候制定表頭,它的表頭是在添加列的時候指定的。table.add_column('項目', ['編號','云編號','名稱','IP地址'])
這一行代碼為例,項目
指定了這個列的表頭名為"項目",['編號','云編號','名稱','IP地址']
為列的值,同樣為列表。PrettyTable不僅提供了手動按行按列添加數(shù)據(jù),也支持直接從csv文件中讀取數(shù)據(jù)。#!/usr/bin/python
#**coding:utf-8**
import sys
from prettytable import PrettyTable
from prettytable import from_csv
reload(sys)
sys.setdefaultencoding('utf8')
table = PrettyTable()
fp = open("res.csv", "r")
table = from_csv(fp)
print(table)
fp.close()
如果要讀取cvs文件數(shù)據(jù),必須要先導(dǎo)入from_csv
,否則無法運行。上面的示例運行結(jié)果如下:PS:csv 文件不能通過 xls 直接重命名得到,會報錯。如果是 xls 文件,請用另存為 csv 獲得 csv 文件D、從sql查詢值添加
從數(shù)據(jù)庫查詢出來的數(shù)據(jù)可以直接導(dǎo)入到表格打印,下面的例子使用了sqlite3
,如果使用的是mysql也是一樣的,只要能查詢到數(shù)據(jù)就能導(dǎo)入到表格中。#!/usr/bin/python
#**coding:utf-8**
import sys
from prettytable import PrettyTable
from prettytable import from_db_cursor
import sqlite3
reload(sys)
sys.setdefaultencoding('utf8')
conn = sqlite3.connect("/tmp/aliyun.db")
cur = conn.cursor()
cur.execute("SELECT * FROM res")
table = from_db_cursor(cur)
print(table)
+------+----------+----------+------------+
| 編號 |云編號| 名稱 | IP地址 |
+------+----------+----------+------------+
|1 | server01 | 服務(wù)器01 | 172.16.0.1 |
|2 | server02 | 服務(wù)器02 | 172.16.0.2 |
|3 | server03 | 服務(wù)器03 | 172.16.0.3 |
|4 | server04 | 服務(wù)器04 | 172.16.0.4 |
|5 | server05 | 服務(wù)器05 | 172.16.0.5 |
|6 | server06 | 服務(wù)器06 | 172.16.0.6 |
|7 | server07 | 服務(wù)器07 | 172.16.0.7 |
|8 | server08 | 服務(wù)器08 | 172.16.0.8 |
|9 | server09 | 服務(wù)器09 | 172.16.0.9 |
+------+----------+----------+------------+
E、從HTML導(dǎo)入數(shù)據(jù)
支持從html的表格中導(dǎo)入,請看下面這個例子:#!/usr/bin/python
#**coding:utf-8**
import sys
from prettytable import PrettyTable
from prettytable import from_html
reload(sys)
sys.setdefaultencoding('utf8')
html_string='''<table>
<tr>
<th>編號</th>
<th>云編號</th>
<th>名稱</th>
<th>IP地址</th>
</tr>
<tr>
<td>1</td>
<td>server01</td>
<td>服務(wù)器01</td>
<td>172.16.0.1</td>
</tr>
<tr>
<td>2</td>
<td>server02</td>
<td>服務(wù)器02</td>
<td>172.16.0.2</td>
</tr>
</table>'''
table = from_html(html_string)
print(table[0])
+------+----------+----------+------------+
| 編號 |云編號| 名稱 | IP地址 |
+------+----------+----------+------------+
|1 | server01 | 服務(wù)器01 | 172.16.0.1 |
|2 | server02 | 服務(wù)器02 | 172.16.0.2 |
+------+----------+----------+------------+
如上示例中,我們可以導(dǎo)入html的表格,但是不一樣的地方是print
語句,使用html表格導(dǎo)入數(shù)據(jù)的時候print的必須是列表中的第一個元素,否則有可能會報[<prettytable.PrettyTable object at 0x7fa87feba590>]
這樣的錯誤。
這是因為table
并不是PrettyTable對象,而是包含單個PrettyTable對象的列表,它通過解析html而來,所以無法直接打印table
,而需要打印table[0]
五、表格輸出格式
正如支持多種輸入一樣,表格的輸出也支持多種格式,我們在上面中的例子中已經(jīng)使用了print的方式輸出,這是一種常用的輸出方式。
A、print
直接通過print
打印出表格。這種方式打印出的表格會帶邊框。B、輸出HTML格式的表格
print(table.get_html_string())
可以打印出html標(biāo)簽的表格。在上面的例子中,使用print(table.get_html_string())
會打印出如下結(jié)果:<table>
<tr>
<th>編號</th>
<th>云編號</th>
<th>名稱</th>
<th>IP地址</th>
</tr>
<tr>
<td>1</td>
<td>server01</td>
<td>服務(wù)器01</td>
<td>172.16.0.1</td>
</tr>
<tr>
<td>2</td>
<td>server02</td>
<td>服務(wù)器02</td>
<td>172.16.0.2</td>
</tr>
</table>
六、選擇性輸出
prettytable在創(chuàng)建表格之后,你依然可以有選擇的輸出某些特定的行.A、輸出指定的列
print table.get_string(fields=["編號", "IP地址"])
可以輸出指定的列B、輸出前兩行
通過print(table.get_string(start = 0, end = 2))
的可以打印出指定的列,當(dāng)然start
和end
參數(shù)讓我可以自由控制顯示區(qū)間。當(dāng)然區(qū)間中包含start
不包含end
,是不是很熟悉這樣的用法?根據(jù)輸出指定行列的功能,我們可以同時指定行和列來輸出,這里就不說明了。
C、將表格切片
從上面的輸出區(qū)間,我們做一個大膽的假設(shè),既然區(qū)間包含start
不包含end
這種規(guī)則和切片的一樣,我們可以不可通過切片來生成一個新的表格然后將其打印。new_table = table[0:2]
print(new_table)
如上代碼段中,我們就可以打印出0到1行共2行的表格,python的切片功能異常強大,配合切片我們可以自由的輸入任意的行。
D、輸出排序
有時候我們需要對輸出的表格進(jìn)行排序,使用print table.get_string(sortby="編號", reversesort=True)
就可以對表格進(jìn)行排序,其中reversesort
指定了是否倒序排序,默認(rèn)為False
,即默認(rèn)正序列排序。
七、表格的樣式
A、內(nèi)置樣式
通過set_style()
可以設(shè)置表格樣式,prettytable內(nèi)置了多種的樣式個人覺得MSWORD_FRIENDLY
,PLAIN_COLUMNS
,DEFAULT
這三種樣式看起來比較清爽,在終端下顯示表格本來看起就很累,再加上一下花里胡哨的東西看起來就更累。除了以上推薦的三種樣式以外,還有一種樣式不得不說,那就是RANDOM
,這是一種隨機的樣式,每一次打印都會在內(nèi)置的樣式中隨機選擇一個,比較好玩。具體內(nèi)置了幾種樣式,請各位參考官網(wǎng)完整自己嘗試輸出看看。#!/usr/bin/python
#**coding:utf-8**
import sys
from prettytable import PrettyTable
from prettytable import MSWORD_FRIENDLY
from prettytable import PLAIN_COLUMNS
from prettytable import RANDOM
from prettytable import DEFAULT
reload(sys)
sys.setdefaultencoding('utf8')
table = PrettyTable(['編號','云編號','名稱','IP地址'])
table.add_row(['1','server01','服務(wù)器01','172.16.0.1'])
table.add_row(['3','server03','服務(wù)器03','172.16.0.3'])
table.add_row(['2','server02','服務(wù)器02','172.16.0.2'])
table.add_row(['9','server09','服務(wù)器09','172.16.0.9'])
table.add_row(['4','server04','服務(wù)器04','172.16.0.4'])
table.add_row(['5','server05','服務(wù)器05','172.16.0.5'])
table.add_row(['6','server06','服務(wù)器06','172.16.0.6'])
table.add_row(['8','server08','服務(wù)器08','172.16.0.8'])
table.add_row(['7','server07','服務(wù)器07','172.16.0.7'])
table.set_style(DEFAULT)
print(table)
B、自定義樣式
除了內(nèi)置的樣式以外,PrettyTable也提供了用戶自定義,例如對齊方式,數(shù)字輸出格式,邊框連接符等等C、設(shè)置對齊方式
align
提供了用戶設(shè)置對齊的方式,值有l
,r
,c
方便代表左對齊,右對齊和居中 如果不設(shè)置,默認(rèn)居中對齊。D、控制邊框樣式
在PrettyTable中,邊框由三個部分組成,橫邊框,豎邊框,和邊框連接符(橫豎交叉的鏈接符號)#!/usr/bin/python
#**coding:utf-8**
import sys
from prettytable import PrettyTable
reload(sys)
sys.setdefaultencoding('utf8')
table = PrettyTable(['編號','云編號','名稱','IP地址'])
table.add_row(['1','server01','服務(wù)器01','172.16.0.1'])
table.add_row(['3','server03','服務(wù)器03','172.16.0.3'])
table.add_row(['2','server02','服務(wù)器02','172.16.0.2'])
table.add_row(['9','server09','服務(wù)器09','172.16.0.9'])
table.add_row(['4','server04','服務(wù)器04','172.16.0.4'])
table.add_row(['5','server05','服務(wù)器05','172.16.0.5'])
table.add_row(['6','server06','服務(wù)器06','172.16.0.6'])
table.add_row(['8','server08','服務(wù)器08','172.16.0.8'])
table.add_row(['7','server07','服務(wù)器07','172.16.0.7'])
table.align[1] = 'l'
table.border = True
table.junction_char='$'
table.horizontal_char = '+'
table.vertical_char = '%'
print(table)
table.border
控制是否顯示邊框,默認(rèn)是True
table.junction_char
控制邊框連接符table.horizontal_char
控制橫邊框符號table.vertical_char
控制豎邊框符號$++++++$++++++++++$++++++++++$++++++++++++$
% 編號 %云編號% 名稱 % IP地址 %
$++++++$++++++++++$++++++++++$++++++++++++$
%1 % server01 % 服務(wù)器01 % 172.16.0.1 %
%3 % server03 % 服務(wù)器03 % 172.16.0.3 %
%2 % server02 % 服務(wù)器02 % 172.16.0.2 %
%9 % server09 % 服務(wù)器09 % 172.16.0.9 %
%4 % server04 % 服務(wù)器04 % 172.16.0.4 %
%5 % server05 % 服務(wù)器05 % 172.16.0.5 %
%6 % server06 % 服務(wù)器06 % 172.16.0.6 %
%8 % server08 % 服務(wù)器08 % 172.16.0.8 %
%7 % server07 % 服務(wù)器07 % 172.16.0.7 %
$++++++$++++++++++$++++++++++$++++++++++++$