unblock.py

import zipfile, re, os, sys, resources.checks as checks
from resources.updatezip import UpdateableZipFile as uzip
from tkinter.filedialog import askopenfilenames, askdirectory
from tkinter import Tk

files = not sys.argv.__contains__('--dir')
cli = sys.argv.__contains__('--cli')
filepaths, folderpath = [],''

if not cli: Tk().withdraw()

# input
if files:
    if not cli:
        filepaths = list(askopenfilenames(filetypes = [('excel files', '.xlsx')]))
    else:
        filepaths.append(input('Enter the path to the xlsx file. Both explicit and relative are valid.\n'))
else:
    folderpath = askdirectory() if not cli else input('Enter the path to the target directory. Both explicit and relative are valid.\n')

#regex pattern to exclude sheetProtection
strclean = re.compile('<sheetProtection.*?>')

def processfile(excelfile):
    sheetfiles = {}
    # get all target sheet paths
    with zipfile.ZipFile(excelfile) as zip:
        for filename in zip.namelist():
            if filename.startswith('xl/worksheets/sheet') and filename.endswith('.xml'):
                ofile = zip.open(filename)
                sheetfiles[filename] = re.sub(strclean, '', ofile.read().decode('utf-8'))
    # alter files
    with uzip(excelfile) as file:
        for fname, fcontent in sheetfiles.items():
            file.writestr(fname, fcontent.encode('utf-8'))

# execute
if files:
    if checks.checkfiles(filepaths):
        for filepath in filepaths:
            processfile(filepath)
        print('\033[92mFinished.\033[0m')
else:
    if checks.checkpath(folderpath):
        for item in os.listdir(folderpath):
            if item.endswith('.xlsx'):
                processfile(item)
        print('\033[92mFinished.\033[0m')