前言
在Windows上編寫python程序時，有時候需要對輸出的文字顏色進行設置，特別是日志顯示，不同級別的日志設置不同的顏色進行展示可以直觀查看。本文主要描述通過ctypes.windll.kernel32調用GetStdHandle()和SetConsoleTextAttribute()兩個API實現(xiàn)來實現(xiàn)Windows控制臺的彩色背景及彩色文字。
基本概念描述
首先分別介紹GetStdHandle()和SetConsoleTextAttribute()這兩個API。
（1） GetStdHandle()的作用是獲取輸入、輸出/錯誤的屏幕緩沖區(qū)的句柄。函數(shù)聲明如下：
HANDLE GetStdHandle(
DWORD nStdHandle
);
其參數(shù)nStdHandle的值為下面幾種類型的一種：
STD_INPUT_HANDLE 標準輸入的句柄
STD_OUTPUT_HANDLE 標準輸出的句柄
STD_ERROR_HANDLE 標準錯誤的句柄
GetStdHandle()返回標準的輸入、輸出或錯誤的設備的句柄，也就是獲得輸入、輸出/錯誤的屏幕緩沖區(qū)的句柄。
（2）SetConsoleTextAttribute() 的作用是在控制臺中設置輸入或輸出文本的文本顏色和背景顏色。其函數(shù)聲明如下：
BOOL SetConsoleTextAttribute(
HANDLE hConsoleOutput, // console 屏幕緩沖區(qū)的句柄
WORD wAttributes // 文本及背景的顏色
);
文本與背景顏色設置可以參考Windows上"color /?"命令的說明。顏色屬性由兩個十六進制數(shù)字指定 -- 第一個為背景，第二個則為前景。每個數(shù)字可以為以下任何值之一，如下
= 黑色 8 = 灰色
= 藍色 9 = 淡藍色
= 綠色 A = 淡綠色
= 淺綠色 B = 淡淺綠色
= 紅色 C = 淡紅色
= 紫色 D = 淡紫色
= 黃色 E = 淡黃色
= 白色 F = 亮白色
如果函數(shù)設置文本及背景顏色成功，則返回非0。如果設置失敗返回0。
prt_cmd_color.py參考代碼如下：
#-*- coding:utf-8 -*-#
#filename: prt_cmd_color.py
import ctypes,sys
STD_INPUT_HANDLE = -10
STD_OUTPUT_HANDLE = -11
STD_ERROR_HANDLE = -12
#字體顏色定義 text colors
FOREGROUND_BLUE = 0x09 # blue.
FOREGROUND_GREEN = 0x0a # green.
FOREGROUND_RED = 0x0c # red.
FOREGROUND_YELLOW = 0x0e # yellow.
# 背景顏色定義 background colors
BACKGROUND_YELLOW = 0xe0 # yellow.
# get handle
std_out_handle = ctypes.windll.kernel32.GetStdHandle(STD_OUTPUT_HANDLE)
def set_cmd_text_color(color, handle=std_out_handle):
Bool = ctypes.windll.kernel32.SetConsoleTextAttribute(handle, color)
return Bool
#reset white
def resetColor():
set_cmd_text_color(FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE)
#green
def printGreen(mess):
set_cmd_text_color(FOREGROUND_GREEN)
sys.stdout.write(mess + '\n')
resetColor()
#red
def printRed(mess):
set_cmd_text_color(FOREGROUND_RED)
sys.stdout.write(mess + '\n')
resetColor()
#yellow
def printYellow(mess):
set_cmd_text_color(FOREGROUND_YELLOW)
sys.stdout.write(mess + '\n')
resetColor()
#white bkground and black text
def printYellowRed(mess):
set_cmd_text_color(BACKGROUND_YELLOW | FOREGROUND_RED)
sys.stdout.write(mess + '\n')
resetColor()
if __name__ == '__main__':
printGreen('printGreen:Gree Color Text')
printRed('printRed:Red Color Text')
printYellow('printYellow:Yellow Color Text')
輸出展示：
說明：如果需要增加不同前景色或者背景色的組合輸出，可以參考代碼和上述顏色編號增加。
使用實例
1、將上述prt_cmd_color.py文件放在與待執(zhí)行腳本ping_ip.py同一目錄下?；蛘咄ㄟ^
sys.path.append()將prt_cmd_color.py
文件所在的目錄添加到搜索路徑下。
2、ping_ip.py腳本中
import prt_cmd_color
導入模塊或者執(zhí)行
from prt_cmd_color import printGreen, printRed
導入指定函數(shù)。
3、ping_ip.py代碼內容如下：
import os, sys
sys.path.append('d:\temp')
from prt_cmd_color import printGreen, printRed
for ip in ['192.168.1.104', '192.168.1.105']:
ret = os.system('ping -n 2 -w 3 %s > nul' % (ip,))
if ret == 0:
printGreen("%s success" % (ip,))
else:
printRed("%s fail" % (ip,))
4、執(zhí)行結果如下圖所示。命令執(zhí)行成功，以綠色顯示提示信息，執(zhí)行失敗以紅色顯示提示信息。
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。
更多文章、技術交流、商務合作、聯(lián)系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號聯(lián)系： 360901061
您的支持是博主寫作最大的動力，如果您喜歡我的文章，感覺我的文章對您有幫助，請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點擊下面給點支持吧，站長非常感激您！手機微信長按不能支付解決辦法：請將微信支付二維碼保存到相冊，切換到微信，然后點擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對您有幫助就好】元
