1
import os
2
import shutil
3
import zipfile
4
import tempfile
5
import argparse
6
from lxml import etree
7
from concurrent.futures import ProcessPoolExecutor, as_completed
8
import multiprocessing
9

10
def remove_modify_verifier(input_pptx, output_pptx):
11
    try:
12
        with tempfile.TemporaryDirectory() as tmpdir:
13
            with zipfile.ZipFile(input_pptx, 'r') as zip_ref:
14
                zip_ref.extractall(tmpdir)
15

16
            presentation_xml_path = os.path.join(tmpdir, 'ppt', 'presentation.xml')
17

18
            if not os.path.exists(presentation_xml_path):
19
                return False, "找不到 ppt/presentation.xml 文件。"
20

21
            parser = etree.XMLParser(remove_blank_text=True)
22
            tree = etree.parse(presentation_xml_path, parser)
23
            root = tree.getroot()
24

25
            namespaces = {'p': 'http://schemas.openxmlformats.org/presentationml/2006/main'}
26

27
            modify_verifiers = root.findall('.//p:modifyVerifier', namespaces)
28

29
            if not modify_verifiers:
30
                pass
31
            else:
32
                for verifier in modify_verifiers:
33
                    parent = verifier.getparent()
34
                    parent.remove(verifier)
35

36
                tree.write(presentation_xml_path, xml_declaration=True, encoding='UTF-8', pretty_print=True)
37

38
            with zipfile.ZipFile(output_pptx, 'w', zipfile.ZIP_DEFLATED) as zip_out:
39
                for foldername, subfolders, filenames in os.walk(tmpdir):
40
                    for filename in filenames:
41
                        file_path = os.path.join(foldername, filename)
42
                        rel_path = os.path.relpath(file_path, tmpdir)
43
                        zip_out.write(file_path, rel_path)
44

45
        return True, None
46
    except Exception as e:
47
        return False, str(e)
48

49
def process_file(file_path, output_dir):
50
    try:
51
        filename = os.path.basename(file_path)
52
        output_path = os.path.join(output_dir, filename)
53
        success, error = remove_modify_verifier(file_path, output_path)
54
        if success:
55
            return (file_path, True, None)
56
        else:
57
            return (file_path, False, error)
58
    except Exception as e:
59
        return (file_path, False, str(e))
60

61
def main():
62
    parser = argparse.ArgumentParser(description='将只读 PPTX/PPT 文件转换为可编辑的 PPTX 文件。')
63
    parser.add_argument('path', nargs='?', help='输入的PPTX/PPT文件或文件夹路径')
64
    args = parser.parse_args()
65

66
    input_folder = 'input'
67
    output_folder = 'output'
68

69
    if os.path.isdir(input_folder):
70
        input_paths = []
71
        for root_dir, _, files in os.walk(input_folder):
72
            for file in files:
73
                if file.lower().endswith(('.pptx', '.ppt')):
74
                    input_paths.append(os.path.join(root_dir, file))
75
    elif args.path:
76
        if os.path.isdir(args.path):
77
            input_paths = []
78
            for root_dir, _, files in os.walk(args.path):
79
                for file in files:
80
                    if file.lower().endswith(('.pptx', '.ppt')):
81
                        input_paths.append(os.path.join(root_dir, file))
82
        elif os.path.isfile(args.path) and args.path.lower().endswith(('.pptx', '.ppt')):
83
            input_paths = [args.path]
84
        else:
85
            print("错误：指定的路径不是有效的PPTX/PPT文件或文件夹。")
86
            return
87
    else:
88
        print("错误：当前目录下不存在 'input' 文件夹，且未提供输入路径。请指定输入文件或文件夹路径。")
89
        return
90

91
    if not input_paths:
92
        print("没有找到需要处理的PPTX/PPT文件。")
93
        return
94

95
    os.makedirs(output_folder, exist_ok=True)
96

97
    total = len(input_paths)
98
    success_count = 0
99
    fail_count = 0
100
    fail_details = []
101

102
    cpu_count = multiprocessing.cpu_count()
103
    with ProcessPoolExecutor(max_workers=cpu_count) as executor:
104
        future_to_file = {executor.submit(process_file, file_path, output_folder): file_path for file_path in input_paths}
105
        for future in as_completed(future_to_file):
106
            file_path = future_to_file[future]
107
            try:
108
                _, success, error = future.result()
109
                if success:
110
                    success_count += 1
111
                else:
112
                    fail_count += 1
113
                    fail_details.append((file_path, error))
114
            except Exception as e:
115
                fail_count += 1
116
                fail_details.append((file_path, str(e)))
117

118
    print(f"遍历 {total} 个文件，已成功 {success_count} 个文件，失败 {fail_count} 个文件。")
119
    if fail_count > 0:
120
        print("失败的文件及原因：")
121
        for file, reason in fail_details:
122
            print(f"{file}: {reason}")
123

124
if __name__ == "__main__":
125
    main()