使用Python編寫簡單的命令行工具

一、Python命令行參數解析模塊

Python中提供了argparse模塊來解析命令行參數，在命令行使用時可以傳入一些參數，程序可以根據這些參數來進行不同的處理。

import argparse

def main():
    parser = argparse.ArgumentParser(description='這是一個命令行工具')
    parser.add_argument('--name', dest='name', help='輸入姓名')
    parser.add_argument('--age', dest='age', help='輸入年齡')
    args = parser.parse_args()
    print('姓名：', args.name)
    print('年齡：', args.age)

if __name__ == '__main__':
    main()

其中，argparse.ArgumentParser()用來創建一個ArgumentParser對象，可以自定義工具的描述信息；parser.add_argument()用來添加命令行參數，指定參數的名稱、默認值、類型、幫助信息等；parser.parse_args()用來解析命令行參數，返回一個Namespace對象，其中包含了傳入的參數的值。

二、使用Click庫構建命令行工具

Click是Python的一個第三方庫，用於構建命令行工具，使用簡單，功能強大，支持命令嵌套、參數類型驗證、自動生成幫助信息等。

import click

@click.command()
@click.option('--name', prompt='請輸入姓名', help='輸入姓名')
@click.option('--age', prompt='請輸入年齡', help='輸入年齡', type=int)
def main(name, age):
    click.echo('姓名：%s' % name)
    click.echo('年齡：%d' % age)

if __name__ == '__main__':
    main()

其中，@click.command()用來修飾函數，將其轉換為Click命令行工具；@click.option()用來添加命令行參數，指定參數名稱、默認值、提示信息、參數類型等；click.echo()用來輸出信息。

三、使用Python標準庫os和subprocess調用系統命令

Python中的os和subprocess庫提供了調用系統命令的介面，在Python程序中可以方便的調用系統命令。

import os
import subprocess

def main():
    os.system('echo "Hello World!"')
    subprocess.call(['echo', 'Hello World!'])

if __name__ == '__main__':
    main()

其中，os.system()函數可以調用系統命令並返回命令執行結果，subprocess.call()函數可以調用系統命令並等待命令執行完成。

四、使用Python的requests庫發送HTTP請求

Python中的requests庫提供了方便的HTTP請求介面，可以進行GET、POST等操作，並且支持代理和SSL驗證。

import requests

def main():
    response = requests.get('http://www.baidu.com')
    print(response.text)

if __name__ == '__main__':
    main()

其中，requests.get()函數用來發送GET請求，並返回一個Response對象，可以通過response.text獲取返回的結果。

五、使用Python的psutil庫獲取系統信息

Python中的psutil庫提供了獲取系統信息的介面，包括CPU、內存、磁碟、網路等信息。

import psutil

def main():
    cpu_percent = psutil.cpu_percent()
    memory_percent = psutil.virtual_memory().percent
    disk_percent = psutil.disk_usage('/').percent
    print('CPU：%f%%' % cpu_percent)
    print('內存：%f%%' % memory_percent)
    print('磁碟：%f%%' % disk_percent)

if __name__ == '__main__':
    main()

其中，psutil.cpu_percent()函數用來獲取CPU使用率，psutil.virtual_memory().percent用來獲取內存使用率，psutil.disk_usage(‘/’).percent用來獲取磁碟使用率。