vieyahn2017 / iBlog

44 stars 0 forks source link

12.2 python使用zipfile模块 #375

Closed vieyahn2017 closed 3 months ago

vieyahn2017 commented 3 years ago

【python】详解zipfile模块读取处理压缩文件实例

https://blog.csdn.net/brucewong0516/article/details/79064384

vieyahn2017 commented 3 years ago

xls本质也是zip文件

这边搜到了一个例子 https://github.com/dreamzxs/myconfig/blob/aa4305b17496459668eec4265d054f5720fc19d0/bin/vba_extract.py

#!/usr/bin/python
# vba_extract.py

##############################################################################
#
# vba_extract - A simple utility to extract a vbaProject.bin binary from an
# Excel 2007+ xlsm file for insertion into an XlsxWriter file.
#
# Copyright 2013-2016, John McNamara, jmcnamara@cpan.org
#
import sys
import shutil
from zipfile import ZipFile
from zipfile import BadZipfile

# The VBA project file we want to extract.
vba_filename = 'vbaProject.bin'

#import pdb
#pdb.set_trace()

# Get the xlsm file name from the commandline.
if len(sys.argv) > 1:
    xlsm_file = sys.argv[1]
else:
    xlsm_file = "11.xlsm"

try:
    # Open the Excel xlsm file as a zip file.
    xlsm_zip = ZipFile(xlsm_file, 'r')

    # print(xlsm_zip.namelist())
    # xlsm_zip.extractall()
    # xlsm_zip.extract('xl/vbaProject.bin')

    # Read the xl/vbaProject.bin file.
    vba_data = xlsm_zip.read('xl/' + vba_filename)

    # Write the vba data to a local file.
    vba_file = open(vba_filename, "wb")
    vba_file.write(vba_data)
    vba_file.close()

except IOError:
    # Use exc_info() for Python 2.5+ compatibility.
    e = sys.exc_info()[1]
    print("File error: %s" % str(e))
    exit()

except KeyError:
    # Usually when there isn't a xl/vbaProject.bin member in the file.
    e = sys.exc_info()[1]
    print("File error: %s" % str(e))
    print("File may not be an Excel xlsm macro file: '%s'" % xlsm_file)
    exit()

except BadZipfile:
    # Usually if the file is an xls file and not an xlsm file.
    e = sys.exc_info()[1]
    print("File error: %s: '%s'" % (str(e), xlsm_file))
    print("File may not be an Excel xlsm macro file.")
    exit()

except:
    # Catch any other exceptions.
    e = sys.exc_info()[1]
    print("File error: %s" % str(e))
    exit()

finally:
    xlsm_zip.close()

print("Extracted: %s" % vba_filename)