Python实现简单的命令行工具

Python是一种流行的编程语言，它被广泛应用于数据处理、机器学习、Web开发等领域。在命令行下，我们可以使用Python来创建各种实用工具，简化我们的日常工作。本文将介绍如何使用Python实现简单的命令行工具。

一、读取命令行参数

在编写命令行工具时，我们通常需要读取命令行参数。Python提供了一个标准库`argparse`来处理命令行参数。下面是一个简单的示例，从命令行读取两个参数，并将它们相加：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('a', type=int, help='the first number')
parser.add_argument('b', type=int, help='the second number')
args = parser.parse_args()

result = args.a + args.b
print(result)

运行以上代码，示例输出如下：

$ python add.py 2 3
5

在命令行中输入`python add.py 2 3`，程序将读取两个整数2和3，并将它们相加输出结果5。

二、解析CSV文件

CSV（Comma-Separated Values）即逗号分隔值，是一种常用的数据格式。在Python中，我们可以使用标准库`csv`来读取和写入CSV文件。下面是一个示例，读取一个CSV文件并计算其中一列的和：

import csv

total_sales = 0

with open('sales.csv') as file:
    reader = csv.reader(file)
    for row in reader:
        total_sales += float(row[1])

print(total_sales)

在上面的代码中，我们打开一个名为`sales.csv`的文件，在循环中逐行读取每个行，并将第二列的数字加入总销售额中。

三、使用正则表达式

正则表达式是一种强大的文本处理工具，可用于匹配、搜索和替换字符串。Python内置了标准库`re`来处理正则表达式。下面是一个示例，从HTML文本中提取所有链接：

import re

html = '<html><body><p>Website: <a href="http://example.com">http://example.com</a></p></body></html>'
pattern = re.compile('<a.*?href="(.*?)".*?>(.*?)</a>')
links = pattern.findall(html)

for link, text in links:
    print('Link: ', link)
    print('Text: ', text)

在上面的代码中，我们使用正则表达式`<a.*?href=”(.*?)”.*?>(.*?)</a>`匹配URL和链接文本。然后，我们使用`findall`方法以元组列表的形式返回所有匹配项。最后，我们遍历列表并打印每个链接和相应的文本。

四、生成随机数

Python提供了一个`random`模块，可以生成各种类型的随机数。下面是一个示例，生成一个10个元素的随机数列并对其进行排序：

import random

numbers = [random.randint(0, 100) for _ in range(10)]
print('Random numbers:', numbers)

numbers.sort()
print('Sorted numbers:', numbers)

在上面的代码中，我们使用列表推导式生成10个随机整数，其中每个元素都在0和100之间。然后，我们使用`sort`方法对列表进行升序排序，并打印排序后的结果。

五、发送邮件

Python内置了`smtplib`模块，可用于发送电子邮件。下面是一个示例，发送一封包含文本和附件的电子邮件：

import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.mime.application import MIMEApplication

# 邮件配置
smtp_server = 'smtp.example.com'
smtp_port = 587
smtp_username = 'user@example.com'
smtp_password = 'password'

# 构造邮件
msg = MIMEMultipart()
msg['From'] = smtp_username
msg['To'] = 'recipient@example.com'
msg['Subject'] = 'Test Email'

body = 'This is a test email.'
msg.attach(MIMEText(body, 'plain'))

with open('file.txt', 'rb') as file:
    attachment = MIMEApplication(file.read(), _subtype='txt')
    attachment.add_header('Content-Disposition', 'attachment', filename='file.txt')
    msg.attach(attachment)

# 发送邮件
with smtplib.SMTP(smtp_server, smtp_port) as server:
    server.starttls()
    server.login(smtp_username, smtp_password)
    server.sendmail(smtp_username, 'recipient@example.com', msg.as_string())

在上面的代码中，我们首先配置SMTP服务器和用户名/密码。然后，我们构造一个带有文本和附件的电子邮件。最后，我们使用`SMTP`类连接到SMTP服务器并发送电子邮件。

总结

在本文中，我们介绍了如何使用Python实现一些简单的命令行工具，包括读取命令行参数、解析CSV文件、使用正则表达式、生成随机数和发送电子邮件。这些工具可以帮助我们完成各种常见任务，提高我们的工作效率。如果您感兴趣，可以深入研究Python的标准库和第三方库，探索更多的可能性。

原创文章，作者：CJUF，如若转载，请注明出处：https://www.506064.com/n/136916.html