20个日常工作常用的Python脚本 简单的python脚本

知识星球：写代码那些事

----

如果你有收获｜欢迎｜点赞｜关注｜转发

----

这里会定期更新｜大厂的开发｜架构｜方案设计

这里也会更新｜如何摸鱼｜抓虾

欢迎来到写代码那些事!当涉及到日常工作的自动化和批处理任务时，Python可以成为你的得力助手。以下是一些常用的Python脚本示例，可以帮助你在日常工作中提高效率：

1. 批量文件重命名：

import os

folder_path = '/path/to/folder'
new_prefix = 'new_prefix'

for filename in os.listdir(folder_path):
    if filename.endswith('.txt'):
        new_name = f'{new_prefix}_{filename}'
        os.rename(os.path.join(folder_path, filename), os.path.join(folder_path, new_name))

2. 自动发送邮件：

import smtplib
from email.mime.text import MIMEText

def send_email(subject, body, to_email):
    smtp_server = 'smtp.example.com'
    smtp_port = 587
    sender_email = 'your_email@example.com'
    sender_password = 'your_password'

    msg = MIMEText(body)
    msg['Subject'] = subject
    msg['From'] = sender_email
    msg['To'] = to_email

    server = smtplib.SMTP(smtp_server, smtp_port)
    server.starttls()
    server.login(sender_email, sender_password)
    server.sendmail(sender_email, to_email, msg.as_string())
    server.quit()

send_email('Hello', 'This is a test email', 'recipient@example.com')

3. 自动备份文件：

import shutil
import datetime

source_folder = '/path/to/source'
backup_folder = '/path/to/backup'

current_date = datetime.datetime.now().strftime('%Y-%m-%d')
backup_path = os.path.join(backup_folder, f'backup_{current_date}')

shutil.copytree(source_folder, backup_path)

4. 网络爬虫：

import requests
from bs4 import BeautifulSoup

url = 'https://example.com'
response = requests.get(url)
soup = BeautifulSoup(response.text, 'html.parser')

for link in soup.find_all('a'):
    print(link.get('href'))

5. 数据处理和分析：

import pandas as pd

data = pd.read_csv('data.csv')
filtered_data = data[data['sales'] > 1000]
average_sales = filtered_data['sales'].mean()

print('Average Sales:', average_sales)

6. 数据清洗和转换：

import pandas as pd

data = pd.read_csv('data.csv')
cleaned_data = data.dropna()  # 删除缺失值
transformed_data = cleaned_data.apply(lambda x: x * 2)  # 数据转换

transformed_data.to_csv('cleaned_and_transformed_data.csv', index=False)

7. 自动化测试脚本：

import unittest

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

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

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

8. 自动化GUI操作：

import pyautogui
import time

# 打开文本编辑器
pyautogui.press('win')
pyautogui.write('notepad')
pyautogui.press('enter')

time.sleep(2)

# 输入文本并保存
pyautogui.write('Hello, World!')
pyautogui.hotkey('ctrl', 's')
time.sleep(1)
pyautogui.write('file_name.txt')
pyautogui.press('enter')

9. 数据库操作脚本：

import sqlite3

conn = sqlite3.connect('my_database.db')
cursor = conn.cursor()

cursor.execute('CREATE TABLE users (id INTEGER PRIMARY KEY, name TEXT, age INTEGER)')
cursor.execute('INSERT INTO users (name, age) VALUES (?, ?)', ('Alice', 25))
conn.commit()

cursor.execute('SELECT * FROM users')
for row in cursor.fetchall():
    print(row)

conn.close()

10. 文件监控和处理：

import os
import time

folder_path = '/path/to/folder'

while True:
    for filename in os.listdir(folder_path):
        if filename.endswith('.txt'):
            with open(os.path.join(folder_path, filename), 'r') as file:
                content = file.read()
                # 在这里进行文件内容的处理操作
                print(content)
            os.remove(os.path.join(folder_path, filename))
    time.sleep(10)

11. 自动备份数据库：

import subprocess
import datetime

current_date = datetime.datetime.now().strftime('%Y-%m-%d')
backup_file = f'backup_{current_date}.sql'

subprocess.run(['mysqldump', '-u', 'username', '-p', 'database_name', '>', backup_file], shell=True)

12. 日志文件分析：

import re

log_file = 'app.log'
error_pattern = r'ERROR: (.+)'

errors = []

with open(log_file, 'r') as file:
    for line in file:
        match = re.search(error_pattern, line)
        if match:
            errors.append(match.group(1))

for error in errors:
    print(error)

13. 自动化截屏：

import pyautogui
import time

time.sleep(5)
screenshot = pyautogui.screenshot()
screenshot.save('screenshot.png')

14. 自动化表单填写：

from selenium import webdriver

browser = webdriver.Chrome()
browser.get('https://example.com')

name_field = browser.find_element_by_id('name')
name_field.send_keys('John Doe')

email_field = browser.find_element_by_id('email')
email_field.send_keys('john@example.com')

submit_button = browser.find_element_by_id('submit')
submit_button.click()

browser.quit()

15. 自动下载文件：

import requests

file_url = 'https://example.com/file.pdf'
response = requests.get(file_url)

with open('downloaded_file.pdf', 'wb') as file:
    file.write(response.content)

16. 日志自动分析：

import re
from collections import Counter

log_file = 'app.log'
error_pattern = r'ERROR: (.+)'

errors = []

with open(log_file, 'r') as file:
    for line in file:
        match = re.search(error_pattern, line)
        if match:
            errors.append(match.group(1))

error_counts = Counter(errors)
for error, count in error_counts.items():
    print(f'{error}: {count} occurrences')

17. 批量图片处理：

from PIL import Image
import os

input_folder = '/path/to/input/folder'
output_folder = '/path/to/output/folder'

for filename in os.listdir(input_folder):
    if filename.endswith('.jpg'):
        img = Image.open(os.path.join(input_folder, filename))
        img.thumbnail((300, 300))
        img.save(os.path.join(output_folder, filename))

18. 自动化表格生成：

import pandas as pd

data = pd.DataFrame({
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 22]
})

data.to_excel('data.xlsx', index=False)

19. 定时提醒工具：

import time
from plyer import notification

def set_reminder(message, interval):
    while True:
        time.sleep(interval)
        notification.notify(
            title='Reminder',
            message=message,
            app_name='Reminder App'
        )

reminder_message = 'Take a break and stretch!'
reminder_interval = 3600  # 1 hour

set_reminder(reminder_message, reminder_interval)

20. 文件夹清理脚本：

import os
import shutil

source_folder = '/path/to/source'
destination_folder = '/path/to/destination'

for filename in os.listdir(source_folder):
    if filename.endswith('.txt'):
        shutil.move(os.path.join(source_folder, filename), os.path.join(destination_folder, filename))