栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

msoffcrypto打开加密Excel和未加密的Excel文件,并用pandas读取

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

msoffcrypto打开加密Excel和未加密的Excel文件,并用pandas读取

文章目录
  • 1. 功能描述
  • 2. 环境安装
  • 3. 代码

1. 功能描述

目前有一个批量读取Excel文件的需求,但是这些文件有些是加了密的,有些是未加密的,当然加密的文件也是知道密码的。
现在要在同一个程序中通过pandas读取这些文件。
通过查找资料,一般来说有win32com的方法,还有就是msoffcrypto的方法。测试了win32com的方法,最终放弃了。
在此我选择的是msoffcrypto-tool的方法,版本为5.0.
python版本为3.6.8
好了,废话不多说,直接上代码

2. 环境安装

可以通过pip直接安装

pip install pandas
pip install msoffcrypto-tool
3. 代码
import pandas as pd  # 读取Excel并进行数据分析
import msoffcrypto   # 读取加密的Excel文件
import io            # 将文件写入到内存中
# 定义一个函数,用来导入Excel文件,并将获取的dataFrame数据结构返回
def read_excel(path, sheet_name, title, header=0, password=''):
    decrypted = io.BytesIO()   # 用来后来将读取到的Excel文件解密之后写入内存中。
    try:
        with open(path, 'rb') as f:   # 用上下文将文件打开
            # 判断Excel文件是否加密,未加密直接pandas读取,加密的就先解密到内存中然后pandas读取。
            if not msoffcrypto.olefile.isOleFile(f):   # mscoffcrypto有可以判断文件是否加密的功能。
                pd_excel = pd.read_excel(f, sheet_name, header=header)
                logging.info("{title}文件读取成功".format(title=title))
            else:
                xlf = msoffcrypto.OfficeFile(f)   # 用msoffcrypto打开加密文件,未加密文件用这个打开会报错
                xlf.load_key(password=password)   # 传入文件的密码
                xlf.decrypt(decrypted)            # 将文件解密并保存到内存中,定义descrypted为io.BytesIO() ,这是msoffcrypto常用方法。
                pd_excel = pd.read_excel(decrypted, sheet_name, header=header)  # 将保存到内存中的文件用pandas打开。
                # 通过pandas读取Excel的内容,指定文件路径名称以及要读取的sheet的名称
                logging.info("{title}文件读取成功".format(title=title))
    except Exception as e:   # 如果文件打开失败,则将报错信息记录到日志中,并向管理员发送邮件。
        error_msg = "Excel文件{title}打开失败,请检查文件路径及权限。".format(title=title)
        logging.critical(error_msg)
        logging.critical(e)
        send_waring_mail(error_msg, title)
        return
    pd.set_option("display.max_columns", None)   # 将pandas输出设置为显示所有列。在辅助print打印时使用
    # 将责任时间的数据格式转换成datetime类型,如此在后期可以参与实践比较
    pd_excel['责任时间'] = pd.to_datetime(pd_excel['责任时间'])
    return pd_excel
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/935608.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号