事情是这样的,公司每个月都要处理一堆销售数据的Excel表格,然后根据内容生成不同的PDF通知发给各个部门。手工操作又慢又容易出错,我就琢磨着用Python解放双手。刚开始以为两三个库就能搞定,真动手才发现一堆细节要处理。
先说说环境准备。我用的Python 3.8,主要用了几个库:pandas处理Excel,reportlab生成PDF,还有os和datetime这些内置库。安装的时候reportlab折腾了半天,pip老是超时,后来换成国内镜像才搞定。建议大家装库的时候直接用清华源,速度飞起。
核心代码分三大块。第一块是读取Excel数据,这里有个坑。最开始我用的是pandas的read_excel,但发现日期格式老是识别错误。后来改成了指定dtype参数,才把那些以0开头的工号保留下来了。还有个问题是要处理合并单元格,pandas默认会填充NaN,得用ffill方法向前填充。
第二块是数据处理。需要根据部门分类,每个部门的数据要单独生成PDF。这里用到了groupby方法,但要注意分组后的数据格式转换。我一开始直接遍历groupby对象,结果老是报错,查了半天文档才发现要先用list转换。
第三块是生成PDF。这是最麻烦的部分,reportlab的用法比较特别。要先创建Canvas对象,然后像画画一样在页面上定位每个元素。坐标系统是从左下角开始的,和平时习惯的左上角不一样,我在这里栽了好几次跟头。字体也是个坑,中文字体要单独注册,不然显示出来都是乱码。
具体实现时,我写了个生成单页PDF的函数。要计算每行数据的位置,动态调整行高。如果数据太多还要自动分页,这个逻辑挺烧脑的。我用了个循环来计算当前页剩余空间,不够就新建一页。
调试过程中最头疼的是格式对齐问题。数字有时候会对不齐,后来发现是字体不是等宽的。换成Courier字体就好了,虽然丑了点但至少对齐了。还有个问题是PDF文件名重复,后来加了个时间戳解决了。
代码优化方面,我后来加了进度显示,用tqdm库做了个进度条,这样处理大量文件时心里有数。异常处理也很重要,比如文件被占用或者格式错误时,要能跳过继续处理下一个。
最终效果还不错,原来要忙活半天的活儿,现在五分钟就搞定了。最重要的是准确率百分百,再也不用担心人为失误。部门同事看到后都来找我要脚本,看来这种实用的小工具确实能提升效率。
几点心得体会:第一,开始项目前一定要把需求分析清楚,我中间就因为没考虑分页重写了部分代码。第二,善用官方文档,比到处找教程靠谱。第三,调试时要有耐心,有时候一个小问题能卡半天。
这个项目虽然不大,但涵盖了文件操作、数据处理、格式转换等多个实用场景。建议新手可以试着做做看,做完后对Python的实际应用会有更深的理解。下次我准备在这个基础上加个邮件自动发送功能,到时候再和大家分享经验。
代码已经传到GitHub上了,需要参考的朋友可以私信我。不过建议大家先自己实现,遇到问题再对照,这样进步更快。记住,看十遍不如写一遍,动手才是硬道理!