File size: 2,002 Bytes
985c397 | 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 | #! python
# SPDX-License-Identifier: LGPL-2.1-or-later
# (c) 2010 Werner Mayer LGPL
# FreeCAD Python script to work with the FCStd file format.
import os
from defusedxml import sax as defused_sax
from xml.sax.handler import ContentHandler
import zipfile
# SAX handler to parse the Document.xml
class DocumentHandler(ContentHandler):
def __init__(self, dirname):
super().__init__()
self.files = []
self.dirname = dirname
def startElement(self, name, attributes):
item = attributes.get("file")
if item is not None:
self.files.append(os.path.join(self.dirname, str(item)))
def characters(self, data):
return
def endElement(self, name):
return
def extractDocument(filename, outpath):
zfile = zipfile.ZipFile(filename)
files = zfile.namelist()
for i in files:
data = zfile.read(i)
dirs = i.split("/")
if len(dirs) > 1:
dirs.pop()
curpath = outpath
for j in dirs:
curpath = curpath + "/" + j
os.mkdir(curpath)
output = open(outpath + "/" + i, "wb")
output.write(data)
output.close()
def createDocument(filename, outpath):
files = getFilesList(filename)
compress = zipfile.ZipFile(outpath, "w", zipfile.ZIP_DEFLATED)
for i in files:
dirs = os.path.split(i)
# print i, dirs[-1]
compress.write(i, dirs[-1], zipfile.ZIP_DEFLATED)
compress.close()
def getFilesList(filename):
dirname = os.path.dirname(filename)
handler = DocumentHandler(dirname)
parser = defused_sax.make_parser()
parser.setContentHandler(handler)
parser.parse(filename)
files = []
files.append(filename)
files.extend(iter(handler.files))
dirname = os.path.join(dirname, "GuiDocument.xml")
if os.path.exists(dirname):
files.append(dirname)
return files
|