分享4个方便且好用的Python自动化脚本

分享4个方便且好用的Python自动化脚本攻略

简介

Python是一种流行的编程语言，可以用于编写自动化脚本来简化重复性任务。下面将分享4个方便且好用的Python自动化脚本。

1. Python脚本示例一: 文件批量重命名

import os

def batch_rename(path, old_ext, new_ext):
    for filename in os.listdir(path):
        if filename.endswith(old_ext):
            os.rename(os.path.join(path, filename), 
                      os.path.join(path, filename.replace(old_ext, new_ext)))

batch_rename('/path/to/files', '.txt', '.md')

这个脚本用于批量重命名指定目录下的文件，将指定后缀的文件名从旧后缀改为新后缀。

2. Python脚本示例二: 网页数据抓取

import requests
from bs4 import BeautifulSoup

def scrape_website(url):
    response = requests.get(url)
    soup = BeautifulSoup(response.text, 'html.parser')
    # 进行网页数据解析和提取
    # ...
    # 返回所需数据

data = scrape_website('https://example.com')

这个脚本使用requests库发送HTTP请求，然后使用BeautifulSoup库解析网页内容，从而实现简单的网页数据抓取。

3. Python脚本示例三: 自动发送邮件

import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.mime.image import MIMEImage

def send_email(subject, message, from_email, to_email):
    msg = MIMEMultipart()
    msg['From'] = from_email
    msg['To'] = to_email
    msg['Subject'] = subject
    msg.attach(MIMEText(message, 'plain'))

    smtp = smtplib.SMTP('smtp.example.com', 587)
    smtp.starttls()
    smtp.login('username', 'password')
    smtp.sendmail(from_email, to_email, msg.as_string())
    smtp.quit()

send_email('Hello', 'This is a test email', '[e-mail]', '[e-mail]')

这个脚本使用smtplib库和email库实现自动发送邮件的功能。

4. Python脚本示例四: 自动化测试脚本

import unittest

class TestMathFunctions(unittest.TestCase):
    def test_add(self):
        self.assertEqual(1 + 1, 2)

    def test_multiply(self):
        self.assertEqual(3 * 5, 15)

if __name__ == '__main__':
    unittest.main()

这个脚本使用unittest库编写自动化测试用例，用于测试代码的功能是否正常。

希望上述示例可以帮助您了解如何使用Python编写方便且好用的自动化脚本。