Python自动化办公之从Excel/Word处理到浏览器操控的全面指南

from openpyxl import Workbook, load_workbook
from openpyxl.styles import Font, Color, PatternFill

# 1. 创建新 workbook
wb = Workbook()
ws = wb.active
ws.title = "Sales Data"

# 2. 写入数据
ws['A1'] = "Product"
ws['B1'] = "Price"
ws.append(['Laptop', 1200]) # 追加一行
ws.append(['Mouse', 25])

# 添加公式
# 3. 设置样式
ws['A1'].font = Font(bold=True, color="FFFFFF")
ws['A1'].fill = PatternFill(start_color="4472C4", fill_type="solid")
ws['C1'] = "Total"
ws['C2'] = "=A2*B2" # 注意：openpyxl 只写公式字符串，不计算结果

# 4. 读取现有文件
wb_load = load_workbook('report.xlsx')
ws_load = wb_load['Sheet1']
value = ws_load['A1'].value
# 读取现有文件
wb = load_workbook("report.xlsx", data_only=True) # data_only=True 读取公式计算后的值
ws = wb["Sales Data"]
print(ws['A2'].value)
# 5. 保存
wb.save('new_report.xlsx')

from docx import Document
from docx.shared import Inches, Pt
from docx.enum.text import WD_ALIGN_PARAGRAPH

# 1. 创建文档
doc = Document()

# 2. 添加标题
heading = doc.add_heading('Monthly Report', 0)
heading.alignment = WD_ALIGN_PARAGRAPH.CENTER

# 3. 添加段落与样式 (Run 的概念)
p = doc.add_paragraph('This is an automated report generated by Python.')
p.add_run(' Bold text').bold = True
p.add_run(' Italic text').italic = True

# 4. 添加表格
table = doc.add_table(rows=3, cols=3)
table.style = 'Table Grid'
cell = table.cell(0, 0)
cell.text = 'Header 1'

# 5. 保存
doc.save('report.docx')

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
import time

# 1. 配置 Chrome 选项 (无头模式)
options = Options()
options.add_argument("--headless=new") # 2026 新版无头模式
options.add_argument("--disable-gpu")

# 初始化驱动 (需确保 chromedriver 版本匹配)
driver = webdriver.Chrome(options=options)

try:
    # 2. 打开网页
    driver.get("https://www.example.com")
    
    # 3.查找元素 (推荐 WebDriverWait 显式等待，此处简化演示)
    search_box = driver.find_element(By.NAME, "q")
    # 4. 模拟操作
    search_box.send_keys("Python Automation")
    search_box.send_keys(Keys.RETURN)
    
    time.sleep(2) # 实际生产请用 WebDriverWait
    
    # 5. 获取数据
    results = driver.find_elements(By.CSS_SELECTOR, ".result-title")
    print(f"Found {len(results)} results")
    
    # 6.截图
    driver.save_screenshot("result.png")
    
finally:
    # 7.关闭浏览器
    driver.quit() # 务必关闭浏览器

import os
import openpyxl
from openpyxl.utils import get_column_letter

def merge_excel_files(folder_path, output_file):
    master_wb = openpyxl.Workbook()
    master_ws = master_wb.active
    master_ws.title = "All_Data"
    
    headers_written = False
    row_offset = 1
    
    for filename in os.listdir(folder_path):
        if filename.startswith("sales_") and filename.endswith(".xlsx"):
            wb = load_workbook(os.path.join(folder_path, filename))
            ws = wb.active
            
            for i, row in enumerate(ws.iter_rows(values_only=True), start=1):
                if i == 1: # 表头
                    if not headers_written:
                        master_ws.append(row)
                        headers_written = True
                    continue
                
                # 添加来源文件名作为新的一列
                new_row = list(row) + [filename]
                master_ws.append(new_row)
    
    # 添加汇总公式示例
    last_row = master_ws.max_row
    col_b = get_column_letter(2) # 假设第2列是金额
    master_ws[f'B{last_row+1}'] = f"=SUM(B2:B{last_row})"
    
    master_wb.save(output_file)
    print(f"Merged into {output_file}")

# merge_excel_files('./sales_reports', 'total_sales.xlsx')

import openpyxl
from docx import Document

def generate_contracts(excel_path, template_path, output_folder):
    wb = load_workbook(excel_path)
    ws = wb.active
    
    # 加载模板
    template = Document(template_path)
    
    for row in ws.iter_rows(min_row=2, values_only=True):
        name, position, salary = row[0], row[1], row[2]
        
        # 复制模板内容 (深拷贝避免污染)
        import copy
        doc = copy.deepcopy(template)
        
        # 替换占位符 (简单文本替换，复杂情况需遍历 paragraphs)
        for paragraph in doc.paragraphs:
            if '{{NAME}}' in paragraph.text:
                paragraph.text = paragraph.text.replace('{{NAME}}', name)
            if '{{POSITION}}' in paragraph.text:
                paragraph.text = paragraph.text.replace('{{POSITION}}', position)
            if '{{SALARY}}' in paragraph.text:
                paragraph.text = paragraph.text.replace('{{SALARY}}', str(salary))
        
        # 处理表格内的替换 (略，逻辑类似，需遍历 table.rows)
        
        doc.save(f"{output_folder}/Contract_{name}.docx")
    print("Contracts generated!")

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.chrome.options import Options
import os

def auto_download_report(url, username, password, download_dir):
    options = Options()
    options.add_argument("--headless=new")
    prefs = {"download.default_directory": download_dir, "download.prompt_for_download": False}
    options.add_experimental_option("prefs", prefs)
    
    driver = webdriver.Chrome(options=options)
    wait = WebDriverWait(driver, 10)
    
    try:
        driver.get(url)
        
        # 等待并输入账号密码
        user_input = wait.until(EC.presence_of_element_located((By.ID, "username")))
        user_input.send_keys(username)
        
        pass_input = driver.find_element(By.ID, "password")
        pass_input.send_keys(password)
        
        driver.find_element(By.ID, "login-btn").click()
        
        # 等待导航栏出现
        wait.until(EC.presence_of_element_located((By.LINK_TEXT, "Daily Reports")))
        driver.find_element(By.LINK_TEXT, "Daily Reports").click()
        
        # 点击下载按钮
        download_btn = wait.until(EC.element_to_be_clickable((By.CLASS_NAME, "download-excel")))
        download_btn.click()
        
        print("Download triggered. Waiting for completion...")
        # 实际项目中需轮询检查文件是否下载完成
        
    finally:
        driver.quit()

from openpyxl.styles import Font, PatternFill

# 假设 data 是从 Selenium 或 requests 抓取到的列表 [(name, price, change), ...]
data = [("AAPL", 150.2, -0.5), ("TSLA", 200.5, -6.2), ("NVDA", 400.0, 2.1)]

wb = Workbook()
ws = wb.active
ws.append(["Stock", "Price", "Change%"])

red_fill = PatternFill(start_color="FFC7CE", end_color="FFC7CE", fill_type="solid")
red_font = Font(color="9C0006", bold=True)

for item in data:
    row_idx = ws.max_row + 1
    ws.append(item)
    
    # 条件格式化：如果跌幅 > 5% (即 change < -5)
    if item[2] < -5.0:
        for cell in ws[row_idx]:
            cell.fill = red_fill
            cell.font = red_font

wb.save("stock_monitor.xlsx")

import os
from openpyxl import load_workbook
from openpyxl.styles import Font, Color

def merge_reports(folder_path):
    summary_wb = Workbook()
    summary_ws = summary_wb.active
    summary_ws.append(['Company', 'Item', 'Amount', 'Status'])

    for filename in os.listdir(folder_path):
        if filename.endswith('.xlsx'):
            wb = load_workbook(os.path.join(folder_path, filename))
            ws = wb.active
            company_name = filename.split('.')[0]
            
            for row in ws.iter_rows(min_row=2, values_only=True):
                item, amount = row[0], row[1]
                status = "Normal"
                color = "000000"
                
                # 业务逻辑：金额大于 10000 标红
                if amount > 10000:
                    status = "High Risk"
                    color = "FF0000"
                
                summary_ws.append([company_name, item, amount, status])
                
                # 动态样式设置 (需要获取单元格对象)
                cell = summary_ws.cell(row=summary_ws.max_row, column=4)
                cell.font = Font(color=color, bold=True)

    summary_wb.save('Monthly_Summary.xlsx')

from docx import Document
import pandas as pd

def generate_contracts():
    # 读取数据源
    df = pd.read_excel('clients.xlsx')
    
    # 加载模板
    template = Document('contract_template.docx')
    
    for index, row in df.iterrows():
        # 复制模板内容 (深拷贝避免污染)
        doc = Document() 
        # 实际项目中通常直接复制模板文件再打开，此处简化逻辑
        # 更好的做法：shutil.copy('template.docx', f'contract_{row["name"]}.docx')
        # 然后打开该文件进行修改
        
        # 模拟替换逻辑 (实际需遍历 paragraph 和 table 进行文本替换)
        # 这里展示核心思路：查找占位符 {{NAME}} 并替换
        for para in template.paragraphs:
            if '{{NAME}}' in para.text:
                para.text = para.text.replace('{{NAME}}', row['Name'])
            if '{{DATE}}' in para.text:
                para.text = para.text.replace('{{DATE}}', str(row['Date']))
        
        # 保存
        doc.save(f"Contracts/Contract_{row['Name']}.docx")

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import time

def price_monitor_and_entry():
    driver = webdriver.Chrome()
    wait = WebDriverWait(driver, 15)
    
    # 1. 抓取竞品价格
    driver.get("https://competitor-site.com/product/123")
    price_elem = wait.until(EC.visibility_of_element_located((By.CLASS_NAME, "price-tag")))
    competitor_price = float(price_elem.text.replace('$', ''))
    
    # 2. 登录内部系统
    driver.get("https://internal-system.com/login")
    driver.find_element(By.ID, "user").send_keys("admin")
    driver.find_element(By.ID, "pass").send_keys("secret")
    driver.find_element(By.ID, "submit").click()
    
    # 3. 填入数据
    wait.until(EC.presence_of_element_located((By.NAME, "price_input")))
    input_box = driver.find_element(By.NAME, "price_input")
    input_box.clear()
    input_box.send_keys(str(competitor_price * 0.95)) # 定价为竞品的 95%
    
    driver.find_element(By.ID, "save_btn").click()
    
    time.sleep(2)
    driver.quit()

# 读取大文件
wb = load_workbook("huge_file.xlsx", read_only=True)
# 写入大文件
wb = Workbook(write_only=True)