17 个必备工作场景的 Python 自动化脚本示例

17 个必备工作场景的 Python 自动化脚本示例 | 极客日志

# Python script to sort files in a directory by their extension
import os
from shutil import move

def sort_files(directory_path):
    for filename in os.listdir(directory_path):
        if os.path.isfile(os.path.join(directory_path, filename)):
            file_extension = filename.split('.')[-1]
            destination_directory = os.path.join(directory_path, file_extension)
            if not os.path.exists(destination_directory):
                os.makedirs(destination_directory)
            move(os.path.join(directory_path, filename), os.path.join(destination_directory, filename))

# Python script to remove empty folders in a directory
import os

def remove_empty_folders(directory_path):
    for root, dirs, files in os.walk(directory_path, topdown=False):
        for folder in dirs:
            folder_path = os.path.join(root, folder)
            if not os.listdir(folder_path):
                os.rmdir(folder_path)

# Python script to rename multiple files in a directory
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))

# Python script for web scraping to extract data from a website
import requests
from bs4 import BeautifulSoup

def scrape_data(url):
    response = requests.get(url)
    soup = BeautifulSoup(response.text, 'html.parser')
    # Extract relevant data here, e.g., title
    title = soup.find('title').get_text() if soup.find('title') else "No Title"
    print(f"Extracted Title: {title}")

# Python script to download images in bulk from a website
import requests
import os

def download_images(url, save_directory):
    os.makedirs(save_directory, exist_ok=True)
    response = requests.get(url)
    if response.status_code == 200:
        try:
            images = response.json() # Assuming the API returns a JSON array of image URLs
            for index, image_url in enumerate(images):
                image_response = requests.get(image_url)
                if image_response.status_code == 200:
                    with open(f"{save_directory}/image_{index}.jpg", "wb") as f:
                        f.write(image_response.content)
        except ValueError:
            print("Response is not valid JSON")

# Python script to automate form submissions on a website
import requests

def submit_form(url, form_data):
    response = requests.post(url, data=form_data)
    if response.status_code == 200:
        print("Form submitted successfully")
        return response.text
    else:
        print(f"Failed to submit form: {response.status_code}")

# Python script to count words in a text file

def count_words(file_path):
    with open(file_path, 'r', encoding='utf-8') as f:
        text = f.read()
        word_count = len(text.split())
    return word_count

# Python script to find and replace text in a file

def find_replace(file_path, search_text, replace_text):
    with open(file_path, 'r', encoding='utf-8') as f:
        text = f.read()
        modified_text = text.replace(search_text, replace_text)
    with open(file_path, 'w', encoding='utf-8') as f:
        f.write(modified_text)

# Python script to generate random text
import random
import string

def generate_random_text(length):
    letters = string.ascii_letters + string.digits + string.punctuation
    random_text = ''.join(random.choice(letters) for i in range(length))
    return random_text

# Python script to send personalized emails to a list of recipients
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart

def send_personalized_email(sender_email, sender_password, recipients, subject, body):
    server = smtplib.SMTP('smtp.gmail.com', 587)
    server.starttls()
    server.login(sender_email, sender_password)
    for recipient_email in recipients:
        message = MIMEMultipart()
        message['From'] = sender_email
        message['To'] = recipient_email
        message['Subject'] = subject
        message.attach(MIMEText(body, 'plain'))
        server.sendmail(sender_email, recipient_email, message.as_string())
    server.quit()

# Python script to send emails with file attachments
import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.base import MIMEBase
from email import encoders

def send_email_with_attachment(sender_email, sender_password, recipient_email, subject, body, file_path):
    server = smtplib.SMTP('smtp.gmail.com', 587)
    server.starttls()
    server.login(sender_email, sender_password)
    message = MIMEMultipart()
    message['From'] = sender_email
    message['To'] = recipient_email
    message['Subject'] = subject
    message.attach(MIMEText(body, 'plain'))
    with open(file_path, "rb") as attachment:
        part = MIMEBase('application', 'octet-stream')
        part.set_payload(attachment.read())
        encoders.encode_base64(part)
        part.add_header('Content-Disposition', f'attachment; filename={file_path}')
        message.attach(part)
    server.sendmail(sender_email, recipient_email, message.as_string())
    server.quit()

# Python script to send automatic email reminders
import smtplib
from email.mime.text import MIMEText
from datetime import datetime, timedelta

def send_reminder_email(sender_email, sender_password, recipient_email, subject, body, reminder_date):
    server = smtplib.SMTP('smtp.gmail.com', 587)
    server.starttls()
    server.login(sender_email, sender_password)
    now = datetime.now()
    reminder_date = datetime.strptime(reminder_date, '%Y-%m-%d')
    if now.date() == reminder_date.date():
        message = MIMEText(body, 'plain')
        message['From'] = sender_email
        message['To'] = recipient_email
        message['Subject'] = subject
        server.sendmail(sender_email, recipient_email, message.as_string())
    server.quit()

# Python script to read and write data to an Excel spreadsheet
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)

# Python script for data analysis and visualization with pandas and matplotlib
import pandas as pd
import matplotlib.pyplot as plt

def analyze_and_visualize_data(data):
    df = pd.DataFrame(data)
    # Example visualization
    df.plot(kind='bar')
    plt.show()

# Python script to merge multiple Excel sheets into a single sheet
import pandas as pd

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

# Python script to connect to a database and execute queries
import sqlite3

def connect_to_database(database_path):
    connection = sqlite3.connect(database_path)
    return connection

def execute_query(connection, query):
    cursor = connection.cursor()
    cursor.execute(query)
    result = cursor.fetchall()
    return result

# Python script to execute SQL queries on a database
import sqlite3

def execute_query(connection, query):
    cursor = connection.cursor()
    cursor.execute(query)
    result = cursor.fetchall()
    return result

import shutil

def backup_database(database_path, backup_directory):
    shutil.copy(database_path, backup_directory)

def restore_database(backup_path, database_directory):
    shutil.copy(backup_path, database_directory)

# Python script to automate posting on Twitter and Facebook
from twython import Twython
import facebook

def post_to_twitter(api_key, api_secret, access_token, access_token_secret, message):
    twitter = Twython(api_key, api_secret, access_token, access_token_secret)
    twitter.update_status(status=message)

def post_to_facebook(api_key, api_secret, access_token, message):
    graph = facebook.GraphAPI(access_token)
    graph.put_object(parent_object='me', connection_name='feed', message=message)

# Python script to automatically share content on social media platforms
import random

def get_random_content():
    contents = ["Check out this update!", "New article published.", "Happy coding!"]
    return random.choice(contents)

def post_random_content_to_twitter(api_key, api_secret, access_token, access_token_secret):
    content = get_random_content()
    # Call post_to_twitter function here
    print(f"Posting to Twitter: {content}")

def post_random_content_to_facebook(api_key, api_secret, access_token):
    content = get_random_content()
    # Call post_to_facebook function here
    print(f"Posting to Facebook: {content}")

# Python script for scraping data from social media platforms
import requests

def scrape_social_media_data(url):
    response = requests.get(url)
    # Use BeautifulSoup or similar to parse response
    return response.text

# Python script to manage system processes
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']):
        if p.info['name'] == process_name:
            p.kill()

# Python script to schedule tasks using cron syntax
from crontab import CronTab

def schedule_task(command, schedule):
    cron = CronTab(user=True)
    job = cron.new(command=command)
    job.setall(schedule)
    cron.write()

# Python script to monitor disk space and send an alert if it's low
import psutil

def check_disk_space(minimum_threshold_gb):
    disk = psutil.disk_usage('/')
    free_space_gb = disk.free / (1024**3) # Convert bytes to GB
    if free_space_gb < minimum_threshold_gb:
        print("Warning: Low disk space!")

# Python script to resize and crop images
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.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)

# Python script to add watermarks to images
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont

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

# Python script to create image thumbnails
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)

# Python script to check the status of a website
import requests

def check_website_status(url):
    response = requests.get(url)
    if response.status_code == 200:
        print(f"Website {url} is up")
    else:
        print(f"Website {url} returned status code {response.status_code}")

# Python script to automate FTP file transfers
from ftplib import FTP

def ftp_file_transfer(host, username, password, local_file_path, remote_file_path):
    with FTP(host) as ftp:
        ftp.login(user=username, passwd=password)
        with open(local_file_path, 'rb') as f:
            ftp.storbinary(f'STOR {remote_file_path}', f)

# Python script to automate network device configuration
from netmiko import ConnectHandler

def configure_network_device(host, username, password, configuration_commands):
    device = {
        'device_type': 'cisco_ios',
        'host': host,
        'username': username,
        'password': password,
    }
    with ConnectHandler(device) as net_connect:
        net_connect.send_config_set(configuration_commands)

# Python script to remove duplicates from data
import pandas as pd

def remove_duplicates(data_frame):
    cleaned_data = data_frame.drop_duplicates()
    return cleaned_data

# Python script for data normalization
import pandas as pd

def normalize_data(data_frame):
    normalized_data = (data_frame - data_frame.min()) / (data_frame.max() - data_frame.min())
    return normalized_data

# Python script to handle missing values in data
import pandas as pd

def handle_missing_values(data_frame):
    filled_data = data_frame.fillna(method='ffill')
    return filled_data

# Python script to extract text from PDFs
import PyPDF2

def extract_text_from_pdf(file_path):
    with open(file_path, 'rb') as f:
        pdf_reader = PyPDF2.PdfReader(f)
        text = ''
        for page_num in range(pdf_reader.num_pages):
            page = pdf_reader.get_page(page_num)
            text += page.extract_text()
        return text

# Python script to merge multiple PDFs into a single 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)

# Python script to add password protection to a 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(pdf_reader.num_pages):
            page = pdf_reader.get_page(page_num)
            pdf_writer.add_page(page)
        pdf_writer.encrypt(password)
    with open(output_path, 'wb') as output_file:
        pdf_writer.write(output_file)

# Python script for GUI automation using pyautogui
import pyautogui

def automate_gui():
    # Simulate mouse movement and click
    pyautogui.moveTo(100, 100, duration=1)
    pyautogui.click()

# Python script to create simple GUI applications using tkinter
import tkinter as tk

def create_simple_gui():
    root = tk.Tk()
    label = tk.Label(root, text="Hello World")
    label.pack()
    root.mainloop()

# Python script to handle GUI events using tkinter
import tkinter as tk

def handle_gui_events():
    def on_button_click():
        print("Button clicked!")
    
    root = tk.Tk()
    button = tk.Button(root, text="Click Me", command=on_button_click)
    button.pack()
    root.mainloop()

# Python script for unit testing with the unittest module
import unittest

def add(a, b):
    return a + b

class TestAddFunction(unittest.TestCase):
    def test_add_positive_numbers(self):
        self.assertEqual(add(2, 3), 5)
    def test_add_negative_numbers(self):
        self.assertEqual(add(-2, -3), -5)
    def test_add_zero(self):
        self.assertEqual(add(5, 0), 5)

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

# Python script for web testing using Selenium
from selenium import webdriver

def perform_web_test():
    driver = webdriver.Chrome()
    driver.get("https://www.example.com")
    # Interact with web elements here
    driver.quit()

# Python script for building test automation frameworks
# Define framework architecture and tools here

def setup_framework():
    print("Setting up test environment...")
    return True

# Python script to automate uploading files to cloud storage
# Requires boto3 for AWS S3 or google-cloud-storage for GCS

def upload_to_cloud(bucket_name, file_path):
    # Implementation depends on specific cloud provider SDK
    print(f"Uploading {file_path} to {bucket_name}")

# Python script to manage AWS resources using Boto3
import boto3

def create_ec2_instance(instance_type, image_id, key_name, security_group_ids):
    ec2 = boto3.resource('ec2')
    instance = ec2.create_instances(
        ImageId=image_id,
        InstanceType=instance_type,
        KeyName=key_name,
        SecurityGroupIds=security_group_ids,
        MinCount=1,
        MaxCount=1
    )
    return instance[0].id

# Python script to automate interactions with Google Drive
# Requires google-api-python-client

def interact_with_drive():
    # Connect to Google Drive API
    print("Connecting to Google Drive...")

# Python script for stock price analysis
# Requires yfinance or similar

def analyze_stock_price(symbol):
    # Fetch data using API
    print(f"Analyzing stock: {symbol}")

# Python script to fetch currency exchange rates
# Requires fixer.io or openexchangerates API

def get_exchange_rates():
    # Fetch rates from API
    print("Fetching exchange rates...")

# Python script for budget tracking and analysis
import pandas as pd

def track_budget(file_path):
    df = pd.read_csv(file_path)
    total_income = df[df['type'] == 'income']['amount'].sum()
    total_expense = df[df['type'] == 'expense']['amount'].sum()
    print(f"Income: {total_income}, Expense: {total_expense}")

# Python script for sentiment analysis using NLTK or other NLP libraries
import nltk
from nltk.sentiment import SentimentIntensityAnalyzer

def analyze_sentiment(text):
    nltk.download('vader_lexicon')
    sia = SentimentIntensityAnalyzer()
    sentiment_score = sia.polarity_scores(text)
    return sentiment_score

# Python script for text summarization using NLP techniques
# Requires sumy or transformers library

def summarize_text(text):
    # Implement summarization logic
    return text[:100] + "..."

# Python script for language translation using NLP libraries
# Requires googletrans or deep_translator

def translate_text(text, target_lang):
    # Connect to translation API
    print(f"Translating to {target_lang}")

17 个必备工作场景的 Python 自动化脚本示例

引言

1. 自动化文件管理

1.1 对目录中的文件进行排序

1.2 删除空文件夹

1.3 重命名多个文件

2. 使用 Python 进行网页抓取

2.1 从网站提取数据

2.2 从网站批量下载图像

2.3 自动提交表单

3. 文本处理和操作

3.1 计算文本文件中的字数

3.2 查找并替换文本

3.3 生成随机文本

4. 电子邮件自动化

4.1 发送个性化电子邮件

4.2 通过电子邮件发送文件附件

4.3 自动邮件提醒

5. 自动化 Excel 电子表格

5.1 Excel 读 & 写

5.2 数据分析和可视化

5.3 合并多个工作表

6. 与数据库交互

6.1 连接到一个数据库

6.2 执行 SQL 查询

6.3 数据备份与恢复

7. 社交媒体自动化

7.1 发布到 Twitter 和 Facebook

7.2 社交媒体自动共享

7.3 抓取社交媒体数据

8. 自动化系统任务

8.1 管理系统进程

8.2 使用 Cron 安排任务

8.3 监控磁盘空间

9. 自动化图像编辑

9.1 图像大小调整和裁剪

9.2 为图像添加水印

9.3 创建图像缩略图

10. 网络自动化

10.1 检查网站状态

10.2 自动 FTP 传输

10.3 网络配置设置

11. 数据清理和转换

11.1 从数据中删除重复项

11.2 数据标准化

11.3 处理缺失值

12. 自动化 PDF 操作

12.1 从 PDF 中提取文本

12.2 合并多个 PDF

12.3 添加密码保护

13. 自动化 GUI

13.1 自动化鼠标和键盘

13.2 创建简单的 GUI 应用程序

13.3 处理 GUI 事件

14. 自动化测试

14.1 使用 Python 进行单元测试

14.2 用于 Web 测试的 Selenium

14.3 测试自动化框架

15. 自动化云服务

15.1 向云空间上传文件

15.2 管理 AWS 资源

15.3 自动化 Google 云端硬盘

16. 财务自动化

16.1 分析股票价格

16.2 货币汇率

16.3 预算追踪

17. 自然语言处理

17.1 情感分析

17.2 文本摘要

17.3 语言翻译

结论

常见问题

1. Python 适合自动化吗？

2. 使用 Python 自动化任务有哪些好处？

3. 我可以在我的项目中使用这些脚本吗？

4. 我需要安装任何库来运行这些脚本吗？

5. 我可以将这些脚本用于商业用途吗？

6. 如何针对我的特定项目进一步优化这些脚本？

7. 我可以使用 Python 自动执行复杂的任务吗？

8. 自动化任务时是否有任何安全考虑？