在日常的開(kāi)發(fā)和使用過(guò)程中，我們常常會(huì)編寫(xiě)一些實(shí)用的Python腳本，比如用來(lái)批量處理文件、獲取系統(tǒng)信息等。然而，每次都要輸入python script_name.py來(lái)運(yùn)行腳本，時(shí)間一長(zhǎng)難免覺(jué)得繁瑣。要是能像使用系統(tǒng)自帶的命令（如ls、cd）一樣，直接在命令行輸入自定義的命令就能執(zhí)行腳本，那該多方便！實(shí)際上，通過(guò)幾個(gè)簡(jiǎn)單的步驟，我們就能將Python腳本變成真正的命令行命令，大幅提升使用效率。

就像類(lèi) Unix 系統(tǒng)中常用的wc -l命令用于統(tǒng)計(jì)文件行數(shù)一樣，接下來(lái)，我將通過(guò)創(chuàng)建一個(gè)功能與之等同的自定義命令count_lines ，分別在類(lèi) Unix 系統(tǒng)（如 Linux、macOS）和 Windows 系統(tǒng)中展開(kāi)講解，詳細(xì)說(shuō)明將 Python 腳本變?yōu)槊钚忻畹膶?shí)現(xiàn)過(guò)程。

案例腳本準(zhǔn)備

我們先編寫(xiě)一個(gè)簡(jiǎn)單的Python腳本count_lines.py，它的功能是讀取指定文本文件，并輸出文件的行數(shù)。代碼如下：

import sys

if len(sys.argv) < 2:
    print("請(qǐng)輸入要統(tǒng)計(jì)行數(shù)的文件路徑")
    sys.exit(1)

file_path = sys.argv[1]
try:
    with open(file_path, 'r', encoding='utf-8') as file:
        lines = file.readlines()
        print(f"文件 {file_path} 的行數(shù)為: {len(lines)}")
except FileNotFoundError:
    print(f"文件 {file_path} 不存在")

目前，我們可以通過(guò)python count_lines.py test.txt（假設(shè)test.txt是要統(tǒng)計(jì)行數(shù)的文件）來(lái)運(yùn)行這個(gè)腳本。

$ python count_lines.py test.txt
文件 test.py 的行數(shù)為: 177

但是這明顯不夠便捷。接下來(lái)，我們就一步步把它變成真正的命令行命令。

一. 類(lèi)Unix系統(tǒng)實(shí)現(xiàn)步驟

步驟一：賦予腳本可執(zhí)行權(quán)限

在類(lèi)Unix系統(tǒng)（如Linux、macOS）中，文件默認(rèn)是沒(méi)有可執(zhí)行權(quán)限的。我們需要使用chmod命令為腳本賦予可執(zhí)行權(quán)限，具體操作如下：

$ chmod +x count_lines.py

執(zhí)行完上述命令后，count_lines.py就具備了可執(zhí)行屬性。此時(shí)，我們可以嘗試運(yùn)行腳本，但需要注意，由于當(dāng)前目錄默認(rèn)不在系統(tǒng)的PATH環(huán)境變量中，我們需要使用相對(duì)路徑來(lái)運(yùn)行，即：

$./count_lines.py test.txt

不過(guò)，直接運(yùn)行可能會(huì)遇到錯(cuò)誤，因?yàn)橄到y(tǒng)此時(shí)還不知道該用Python解釋器來(lái)執(zhí)行這個(gè)腳本，它會(huì)嘗試以Shell腳本的方式去執(zhí)行，從而導(dǎo)致語(yǔ)法錯(cuò)誤。別擔(dān)心，我們接著進(jìn)行下一步操作。

步驟二：添加解釋器shebang

Shebang（也稱為Hashbang）是一個(gè)特殊的符號(hào)，它的作用是告訴系統(tǒng)應(yīng)該使用哪個(gè)解釋器來(lái)執(zhí)行腳本。對(duì)于Python腳本，我們通常添加#!/usr/bin/env python這一行作為腳本的第一行。使用env的好處在于，它可以根據(jù)系統(tǒng)的PATH環(huán)境變量找到正確的Python解釋器，避免了因Python安裝路徑不同而導(dǎo)致的問(wèn)題。

修改后的count_lines.py腳本如下：

#!/usr/bin/env python
import sys

if len(sys.argv) < 2:
    print("請(qǐng)輸入要統(tǒng)計(jì)行數(shù)的文件路徑")
    sys.exit(1)

file_path = sys.argv[1]
try:
    with open(file_path, 'r', encoding='utf-8') as file:
        lines = file.readlines()
        print(f"文件 {file_path} 的行數(shù)為: {len(lines)}")
except FileNotFoundError:
    print(f"文件 {file_path} 不存在")

添加完Shebang后，我們?cè)俅螄L試運(yùn)行腳本：

$./count_lines.py test.txt

此時(shí)，腳本就能正常運(yùn)行了。而且，我們還可以去掉腳本的.py擴(kuò)展名，讓它看起來(lái)更像一個(gè)系統(tǒng)命令：

$ mv count_lines.py count_lines
$./count_lines test.txt

步驟三：將命令添加到PATH環(huán)境變量

雖然現(xiàn)在我們可以通過(guò)./count_lines來(lái)運(yùn)行腳本，但還是不夠方便，因?yàn)楸仨氃谀_本所在的目錄下執(zhí)行。為了能在任意目錄下都能運(yùn)行這個(gè)命令，我們需要將腳本所在的目錄添加到系統(tǒng)的PATH環(huán)境變量中。

不建議直接將腳本復(fù)制到系統(tǒng)級(jí)的目錄（如/usr/bin、/usr/local/bin），因?yàn)檫@樣可能會(huì)引發(fā)命名沖突，甚至影響系統(tǒng)穩(wěn)定性。比較好的做法是在用戶主目錄下創(chuàng)建一個(gè)bin目錄，然后將腳本復(fù)制到該目錄，并將bin目錄添加到PATH中。

創(chuàng)建bin目錄：

$ mkdir -p ~/bin

• 復(fù)制腳本到bin目錄：

$ cp count_lines ~/bin

• 臨時(shí)添加bin目錄到PATH（僅對(duì)當(dāng)前會(huì)話有效）：

$ export PATH=$PATH:$HOME/bin

如果希望設(shè)置永久生效，需要將上述命令添加到用戶主目錄下的.profile或.bash_profile文件中。以.bash_profile為例，使用文本編輯器打開(kāi)該文件：

$ vi ~/.bash_profile

在文件末尾添加以下內(nèi)容：

export PATH=$PATH:$HOME/bin

保存并關(guān)閉文件后，執(zhí)行source ~/.bash_profile使配置立即生效。

完成上述步驟后，我們就可以在任意目錄下，像使用系統(tǒng)命令一樣，直接輸入count_lines test.txt來(lái)統(tǒng)計(jì)文件行數(shù)了。

二. Windows系統(tǒng)實(shí)現(xiàn)步驟

在Windows系統(tǒng)中，將Python腳本變成命令行命令的方式與類(lèi)Unix系統(tǒng)有所不同，主要通過(guò)以下幾個(gè)步驟來(lái)實(shí)現(xiàn)：

步驟一：創(chuàng)建批處理文件

批處理文件（.bat）是Windows系統(tǒng)中用于自動(dòng)執(zhí)行一系列命令的文件。我們可以創(chuàng)建一個(gè)批處理文件，在其中指定使用Python解釋器來(lái)運(yùn)行我們的腳本。在腳本count_lines.py所在的目錄下，新建一個(gè)文本文件，命名為count_lines.bat，使用文本編輯器打開(kāi)，并添加以下內(nèi)容：

@echo off
python count_lines.py %*

@echo off用于關(guān)閉批處理文件執(zhí)行過(guò)程中的命令回顯，讓界面更加簡(jiǎn)潔。python count_lines.py %*表示使用Python解釋器運(yùn)行count_lines.py腳本，并將命令行中輸入的參數(shù)傳遞給腳本。

步驟二：配置環(huán)境變量

為了能在任意目錄下運(yùn)行count_lines.bat，我們需要將其所在目錄添加到系統(tǒng)的環(huán)境變量中。具體操作如下：

打開(kāi)“系統(tǒng)屬性”：右鍵點(diǎn)擊“此電腦”，選擇“屬性”，在彈出的窗口中點(diǎn)擊“高級(jí)系統(tǒng)設(shè)置”。

進(jìn)入“環(huán)境變量”設(shè)置：在“系統(tǒng)屬性”窗口的“高級(jí)”選項(xiàng)卡中，點(diǎn)擊“環(huán)境變量”按鈕。

編輯Path環(huán)境變量：在“系統(tǒng)變量”列表中找到Path變量，點(diǎn)擊“編輯”。

添加路徑：點(diǎn)擊“新建”，輸入count_lines.bat所在的目錄路徑（例如C:\scripts，根據(jù)實(shí)際情況填寫(xiě)），然后點(diǎn)擊“確定”保存設(shè)置。

步驟三：測(cè)試命令

完成環(huán)境變量配置后，打開(kāi)命令提示符（CMD）或PowerShell，在任意目錄下輸入count_lines.bat test.txt，即可運(yùn)行腳本統(tǒng)計(jì)文件行數(shù)。如果希望直接輸入count_lines test.txt就能運(yùn)行，可以將count_lines.bat改名為count_lines（去掉.bat擴(kuò)展名），不過(guò)需要注意，這樣可能會(huì)與系統(tǒng)中其他同名文件或命令產(chǎn)生沖突，操作時(shí)需謹(jǐn)慎。

PS E:\> count_lines test.txt
文件 test.py 的行數(shù)為: 177

通過(guò)以上步驟，無(wú)論是在類(lèi)Unix系統(tǒng)還是Windows系統(tǒng)中，我們都成功將Python腳本變成了便捷的命令行命令。當(dāng)然，Python在命令行工具開(kāi)發(fā)方面的潛力遠(yuǎn)不止于此。利用click、argparse等強(qiáng)大的庫(kù)，我們還能開(kāi)發(fā)出功能更豐富、交互更友好的命令行應(yīng)用。

到此這篇關(guān)于Python打造自己的專(zhuān)屬命令行工具的文章就介紹到這了,更多相關(guān)Python 命令行工具內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論