python_word/wwdocx.py at master · ghosttk/python_word · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import logging
import sys
import os
import zipfile
from zipfile import ZipFile
import shutil
from lxml import etree

class docx():
    def add_elems(self,new_elem,new_text):
        for elem in self.elems:
            new=etree.SubElement(elem,new_elem)
            new.text=new_text
        pass
    def remove_one_elem(self,old_elem):
        logging.debug(old_elem.getparent())
        old_elem.getparent().remove(old_elem)
        pass
    def remove_elems(self,elem_to_remove):
        elems=self.find_elems(elem_to_remove)
        for elem in self.elems:
            self.remove_one_elem(elem)
    def change_one_elem(self,old_elem,new_elem):
        pass
    def change_elems(self,old_elem,new_text):
        pass
    def __init__(self,fname):
        self.fname=fname
        self.extract_file()
        self.parse_document()
    def extract_file(self):
        f=ZipFile(self.fname)
        rtn=f.extractall(path='tmp')
        f.close()
        return rtn
    def parse_document(self):
        f=open('tmp/word/document.xml')
        tree=etree.parse(f)
        root=tree.getroot()
        self.tree=tree
        self.root=root
        self.nsmap=root.nsmap
        f.close()
        return root
    def find_elems(self,elem):
        elems=self.tree.xpath(elem,namespaces=self.nsmap)
        self.elems=elems
        return elems
    def save_docx(self):
        f=open('tmp/word/document.xml',mode='w')
        self.tree.write(f)
        f.close()
        self.zip_dir('tmp')
        pass
    def zip_dir(self,dirname):
        new_zfname='%s_new.%s'%(self.fname.split('.')[0],'docx')
        filelist = []
        if os.path.isfile(dirname):
            filelist.append(dirname)
        else :
            for root, dirs, files in os.walk(dirname):
                for name in files:
                    filelist.append(os.path.join(root, name))
        zf =ZipFile(new_zfname, "w",)
        for tar in filelist:
            arcname = tar[len(dirname):]
            zf.write(tar,arcname)
        zf.close()
        #shutil.rmtree('tmp')


if __name__=='__main__':
    logging.basicConfig(level=logging.DEBUG)
    mydoc=docx('tmp.docx')
    mydoc.remove_elems('//wp:positionH/wp:posOffset')
    mydoc.find_elems('//wp:positionH')
    new_elem='{%s}%s'%(mydoc.nsmap['wp'],'align')
    new_text='right'
    mydoc.add_elems(new_elem,new_text)

    mydoc.save_docx()