20 个 Python 自动化脚本实战：文件管理与办公效率提升

说明： 使用 os.walk 从底层向上遍历，确保删除空目录前已处理完子目录。

1.3 批量重命名文件

根据规则批量修改文件名，例如去除特定前缀或后缀。

import os

def rename_files(directory_path, old_name, new_name):
    for filename in os.listdir(directory_path):
        if old_name in filename:
            new_filename = filename.replace(old_name, new_name)
            os.rename(os.path.join(directory_path, filename), 
                      os.path.join(directory_path, new_filename))

if __name__ == '__main__':
    # 示例：将 'old_' 替换为 'new_'
    rename_files('your_directory_path', 'old_', 'new_')

2. Excel 办公自动化

2.1 读取和写入数据

利用 pandas 库高效处理 Excel 文件的读写操作。

import pandas as pd

def read_excel(file_path):
    df = pd.read_excel(file_path)
    return df

def write_to_excel(data, file_path):
    df = pd.DataFrame(data)
    df.to_excel(file_path, index=False)

if __name__ == '__main__':
    dataframe = read_excel('path_to_your_input_file.xlsx')
    write_to_excel(dataframe, 'path_to_your_output_file.xlsx')

说明： index=False 参数表示在输出的 Excel 文件中不包括行索引，使表格更美观。

2.2 合并多个工作表

将同一个 Excel 文件中的多个 Sheet 合并到一个新的文件中。

import pandas as pd

def merge_sheets(file_path, output_file_path):
    xls = pd.ExcelFile(file_path)
    df = pd.DataFrame()
    for sheet_name in xls.sheet_names:
        sheet_df = pd.read_excel(xls, sheet_name)
        df = pd.concat([df, sheet_df], ignore_index=True)
    df.to_excel(output_file_path, index=False)

if __name__ == '__main__':
    merge_sheets('path_to_your_excel_file.xlsx', 'output_merged.xlsx')

说明： 使用 pd.concat 替代 deprecated 的 append 方法，性能更好且符合新版 Pandas 规范。

3. 图片处理

3.1 图片裁剪与调整大小

使用 Pillow 库进行基础图片编辑。

from PIL import Image

def resize_image(input_path, output_path, width, height):
    image = Image.open(input_path)
    resized_image = image.resize((width, height), Image.Resampling.LANCZOS)
    resized_image.save(output_path)

def crop_image(input_path, output_path, left, top, right, bottom):
    image = Image.open(input_path)
    cropped_image = image.crop((left, top, right, bottom))
    cropped_image.save(output_path)

if __name__ == '__main__':
    resize_image('input.jpg', 'resized.jpg', 800, 600)
    crop_image('input.jpg', 'cropped.jpg', 100, 100, 400, 400)

3.2 添加水印

为图片添加半透明文字水印，保护版权。

from PIL import Image, ImageDraw, ImageFont

def add_watermark(input_path, output_path, watermark_text):
    image = Image.open(input_path).convert('RGBA')
    draw = ImageDraw.Draw(image)
    try:
        font = ImageFont.truetype('arial.ttf', 36)
    except IOError:
        font = ImageFont.load_default()
    draw.text((10, 10), watermark_text, fill=(255, 255, 255, 128), font=font)
    image.save(output_path)

if __name__ == '__main__':
    add_watermark('input.jpg', 'watermarked.jpg', 'Copyright Text')

3.3 创建缩略图

批量生成图片缩略图，节省存储空间。

from PIL import Image

def create_thumbnail(input_path, output_path, size=(128, 128)):
    image = Image.open(input_path)
    image.thumbnail(size)
    image.save(output_path)

if __name__ == '__main__':
    create_thumbnail('input.jpg', 'thumbnail.jpg')

4. 系统任务管理

4.1 进程监控与终止

查看当前运行进程并强制结束指定名称的进程。

import psutil

def get_running_processes():
    return [p.info for p in psutil.process_iter(['pid', 'name', 'username'])]

def kill_process_by_name(process_name):
    for p in psutil.process_iter(['pid', 'name', 'username']):
        try:
            if p.info['name'] == process_name:
                p.kill()
        except (psutil.NoSuchProcess, psutil.AccessDenied):
            pass

if __name__ == '__main__':
    print(get_running_processes())
    # kill_process_by_name('process_name_here')

5. PDF 文件操作

5.1 合并多个 PDF 文件

将多个 PDF 文件合并为一个。

import PyPDF2

def merge_pdfs(input_paths, output_path):
    pdf_merger = PyPDF2.PdfMerger()
    for path in input_paths:
        with open(path, 'rb') as f:
            pdf_merger.append(f)
    with open(output_path, 'wb') as f:
        pdf_merger.write(f)

if __name__ == '__main__':
    input_pdf_paths = ['pdf1.pdf', 'pdf2.pdf']
    merge_pdfs(input_pdf_paths, 'merged.pdf')

5.2 PDF 密码保护

为 PDF 文件设置打开密码。

import PyPDF2

def add_password_protection(input_path, output_path, password):
    with open(input_path, 'rb') as f:
        pdf_reader = PyPDF2.PdfReader(f)
        pdf_writer = PyPDF2.PdfWriter()
        for page_num in range(len(pdf_reader.pages)):
            page = pdf_reader.pages[page_num]
            pdf_writer.add_page(page)
    pdf_writer.encrypt(password)
    with open(output_path, 'wb') as output_file:
        pdf_writer.write(output_file)

if __name__ == '__main__':
    add_password_protection('input.pdf', 'protected.pdf', 'your_password')

6. 邮件自动化（新增）

6.1 发送 HTML 邮件

使用 smtplib 自动发送邮件通知。

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

def send_email(subject, body, to_addr, from_addr, smtp_server, port, user, pwd):
    msg = MIMEMultipart()
    msg['From'] = from_addr
    msg['To'] = to_addr
    msg['Subject'] = subject
    msg.attach(MIMEText(body, 'html'))
    
    server = smtplib.SMTP(smtp_server, port)
    server.starttls()
    server.login(user, pwd)
    server.send_message(msg)
    server.quit()

if __name__ == '__main__':
    send_email(
        subject='Test Report',
        body='<h1>Hello</h1>',
        to_addr='[email protected]',
        from_addr='[email protected]',
        smtp_server='smtp.example.com',
        port=587,
        user='user',
        pwd='password'
    )

7. 网络数据采集（新增）

7.1 简单的网页抓取

使用 requests 和 BeautifulSoup 获取网页标题。

import requests
from bs4 import BeautifulSoup

def scrape_title(url):
    response = requests.get(url)
    soup = BeautifulSoup(response.text, 'html.parser')
    title = soup.title.string if soup.title else 'No Title'
    return title

if __name__ == '__main__':
    print(scrape_title('https://www.python.org'))

总结

本文介绍了多种常用的 Python 自动化脚本场景，涵盖文件管理、Office 处理、图像处理、系统维护及网络交互。通过模块化设计，这些脚本可灵活组合以适应不同需求。在实际应用中，建议增加异常处理机制（try-except），并定期备份重要数据。掌握这些基础脚本后，开发者可进一步探索 Selenium 进行浏览器自动化，或使用 Airtest 进行游戏/应用测试，从而构建更复杂的自动化工作流。