add ZIP batch export extension

1 files changed, 80 insertions, 0 deletions

diff --git a/lib/extensions/zip.py b/lib/extensions/zip.py
new file mode 100644
index 00000000..4720ad1e
--- /dev/null
+++ b/lib/extensions/zip.py
@@ -0,0 +1,80 @@
+import sys
+import traceback
+import os
+import inkex
+import tempfile
+from zipfile import ZipFile
+from libembroidery import *
+
+from .base import InkstitchExtension
+from ..i18n import _
+from ..output import write_embroidery_file
+from ..stitch_plan import patches_to_stitch_plan
+from ..svg import render_stitch_plan, PIXELS_PER_MM
+
+
+class Zip(InkstitchExtension):
+    def __init__(self, *args, **kwargs):
+        InkstitchExtension.__init__(self)
+        self.OptionParser.add_option("-c", "--collapse_len_mm",
+                                     action="store", type="float",
+                                     dest="collapse_length_mm", default=3.0,
+                                     help="max collapse length (mm)")
+
+        # it's kind of obnoxious that I have to do this...
+        self.formats = []
+        formatList = embFormatList_create()
+        curFormat = formatList
+        while(curFormat):
+            # extension includes the dot, so we'll remove it
+            extension = embFormat_extension(curFormat)[1:]
+            description = embFormat_description(curFormat)
+            writer_state = embFormat_writerState(curFormat)
+
+            if writer_state.strip() and embFormat_type(curFormat) != EMBFORMAT_OBJECTONLY:
+                self.OptionParser.add_option('--format-%s' % extension, type="inkbool", dest=extension)
+                self.formats.append(extension)
+            curFormat = curFormat.next
+
+    def effect(self):
+        if not self.get_elements():
+            return
+
+        patches = self.elements_to_patches(self.elements)
+        stitch_plan = patches_to_stitch_plan(patches, self.options.collapse_length_mm * PIXELS_PER_MM)
+
+        base_file_name = self.get_base_file_name()
+        path = tempfile.mkdtemp()
+
+        files = []
+
+        for format in self.formats:
+            if getattr(self.options, format):
+                output_file = os.path.join(path, "%s.%s" % (base_file_name, format))
+                write_embroidery_file(output_file, stitch_plan, self.document.getroot())
+                files.append(output_file)
+
+        if not files:
+            self.errormsg(_("No embroidery file formats selected."))
+
+        temp_file = tempfile.NamedTemporaryFile(suffix=".zip", delete=False)
+
+        # in windows, failure to close here will keep the file locked
+        temp_file.close()
+
+        with ZipFile(temp_file.name, "w") as zip_file:
+            for file in files:
+                zip_file.write(file)
+
+        # inkscape will read the file contents from stdout and copy
+        # to the destination file that the user chose
+        with open(temp_file.name) as output_file:
+            sys.stdout.write(output_file.read())
+
+        os.remove(temp_file.name)
+        for file in files:
+            os.remove(file)
+        os.rmdir(path)
+
+        # don't let inkex output the SVG!
+        sys.exit(0)