day20-自动化办公2

1.excel 文件的读操作

import openpyxl

(1) 加载文件(只能加载后缀是.xlsx文件)

wb = openpyxl.load_workbook('files/学生信息.xlsx')

(2) 从工作簿中获取工作表相关信息

• 获取所有工作表的表名

names = wb.sheetnames
print(names)

• 获取活跃表

sheet1 = wb.active
print(sheet1)

• 跟据表名获取指定表
• 工作表对象[表名] - 获取指定工作表

sheet2 = wb['sheet1']
print(sheet2)

(3) 从工作表中获取表相关内容

• 获取表名: 工作表对象.title

print(sheet1.title)      # 'sheet1'

• 获取最大行数

print(sheet1.max_row)

• 获取最大列数

print(sheet1.max_column)

(4) 通过表获取单元格

• 获取指定一个单元格
• a. 工作表[‘列号行号’] - 获取指定位置对应的单元格(列号是字母)

cell1 = sheet1['A2']
print(cell1)

• b. 工作表.cell (行号，列号) - 获取指定位置对应的单元格(列号是数字)

cell2 = sheet1.cell(2, 1)

• 获取单元格的内容
• 单元格对象.value

print(cell1, value)
print(cell2, value)

2.excel 文件写操作

• 注意：excel 所有的写操作，如果要有效必须在操作完成以后对excel文件对应的工作簿进行保存操作

(1) 新建工作簿

if os.path.exists('files/test.xlsx')
	print('打开')
    wb = openpyxl.load_workbook('files/test.xlsx')
else:
    print('新建')
    wb = openpyxl.Workbook()
    wb.save('files/test.xlsx')

(2) 保存工作簿

• 工作簿对象.save(文件路径)

wb.save('files/test.xlsx')

(3) 针对表的写操作

• 新建表；工作簿对象.create_sheet(表名，位置)

wb.create_sheet('老师表')
wb.create_sheet('学生表', 0)

• 删除表；工作簿对象.remove (表对象)

try:
    wb.remove(wb['老师表'])
except KeyError:
    pass

• 修改表的表名：表对象.title = 新的表名

if 'Sheet' in wb.sheetnames:
    wb['Sheet'].title = '课程表'

(4) 修改单元格中的内容

• 表对象[‘列号行号’] = 数据
• 单元格对象.value = 数据

sheet1 = wb.active
sheet1['A1'] = '姓名'
sheet1['B1'] = '年龄'
sheet1['A1'] = ''

sheet1.cell(1, 2).value = 'Age'
print(sheet1.cell(1, 2).value)

3.csv 文件操作

(1) 创建文件对象的 reader 对象

• csv.reader (文件对象) - 读数据的时候每一行内容对应一个列表(返回值是包含文件每行内容的迭代器)
• csv.DictReader(文件对象) - 读数据的时候每一行内容对应一个字典(返回值是包含文件每行内容的迭代器)

f = open('files/2018年北京积分落户数据.csv', encoding='utf-8')
# reader = csv.reader(f)
reader = csv.DictReader(f)

• 通过 reader 读数据

print(next(reader))
print(next(reader))

(2) 创建文件对应的 writer 对象

• csv.writer(文件对象)

f = open('files/test.csv', 'w', encoding='utf-8')
writer = csv.writer(f)

• 写数据：writer.writerow (列表) - 一次写一行
• csv.DictWriter (文件对象)

f = open('files/test2.csv', 'w', encoding='utf-8')
writer = csv.DictWriter(f, ['name', 'age', 'score', 'gender'])

# 写数据
writer.writeheader()     # 将原来设定好的字典的键作为第一行内容写入文件
writer.writerow({'name': '小明', 'age': 18, 'score': 100, 'gender': '男'})
writer.writerows([
    {'name': '小花', 'age': 30, 'score': 92, 'gender': '女'},
    {'name': '小红', 'age': 28, 'score': 67, 'gender': '女'},
    {'name': '老王', 'age': 25, 'score': 88, 'gender': '男'}
])

5.pdf 文件操作

import PyPDF2, openpyxl

(1) 读操作 - 获取 pdf 文件内容

• 以读的方式打开 pdf 文件

reader = PyPDF2.PdfFileReader('files/HEU_KMS_Activator_v20.0.0用户使用手册.pdf')

• 获取总的页数

total_page = reader.getNumPages()
print(total_page)

• 获取指定页面 (Pageobject)

page0 = reader.getPage(0)
page1 = reader.getPage(1)
page2 = reader.getPage(2)
page3 = page1.rotateClockwise(90)    # 旋转

(2) 写操作

• 创建空的 pdf 文件对象
• PyPDF2.PdfFileWriter() - 创建一个空的 pdf 文件对象

writer1 = PyPDF2.PdfFileWriter()

• 添加页面
• writer.addPage(页面对象) - 指定页
• writer.addBlankPage() - 添加空白页

writer1.addPage(page0)
writer1.addBlankPage()
writer1.addPage(page1)
writer1.addBlankPage()
writer1.addPage(page2)
writer1.addBlankPage()

• 保存 pdf 文件

f = open('files/test.pdf', 'wb')
writer1.write(f)

6. 添加水印

import PyPDF2

# 1. 准备需要添加水印的pdf
reader1 = PyPDF2.PdfFileReader('files/存储引擎的讲解.pdf')
# 2. 准备水印文件
walter_page = PyPDF2.PdfFileReader('files/demo1.pdf').getPage(0)
# 3. 准备空的pdf，用来存放添加完水印的页
new_pdf = PyPDF2.PdfFileWriter()
# 4. 遍历原文件的每页内容，添加水印后添加到空的pdf中
for page_num in range(reader1.getNumPages()):
    # 获取原文件的一页内容
    page = reader1.getPage(page_num)
    # 添加水印
    page.mergePage(walter_page)
    # 将新的页添加到空的pdf中
    new_pdf.addPage(page)
# 5.保存新的pdf文件
f = open('files/test3.pdf', 'wb')
new_pdf.write(f)

7.创建文印文件

from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
from reportlab.pdfgen import canvas

# 1. 注册字体文件
pdfmetrics.registerFont(TTFont('Font1', 'files/Vera.ttf'))
pdfmetrics.registerFont(TTFont('Font2', 'files/青呱石头体.ttf'))

# 2.生成文字
# 1）创建空的pdf文件
walter_pdf = canvas.Canvas('files/walter.pdf')

# 2）设置字体
walter_pdf.setFont('Font1', 40)

# 3）设置文字颜色
walter_pdf.setFillColorRGB(1, 0, 0, 1)

# 4）旋转
walter_pdf.rotate(45)

# 5）渲染文字内容
walter_pdf.drawString(200, 100, 'hello world!')

walter_pdf.setFillColorRGB(0, 1, 0, 1)
walter_pdf.setFont('Font2', 40)
walter_pdf.drawString(400, 200, 'hello world!')


# 6）保存
walter_pdf.save()