Auto-generate labels from a base libreoffice template

1 files changed, 93 insertions, 2 deletions

diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index 23416d406..6ac197af4 100644
--- a/ishtar_common/models.py
+++ b/ishtar_common/models.py
@@ -20,6 +20,7 @@
 """
 Models description
 """
+from bs4 import BeautifulSoup
 import copy
 import datetime
 import inspect
@@ -55,6 +56,7 @@ from django.contrib.sites.models import Site
 from django.core.cache import cache
 from django.core.exceptions import ObjectDoesNotExist, ValidationError, \
     MultipleObjectsReturned
+from django.core.files.base import ContentFile
 from django.core.files.uploadedfile import SimpleUploadedFile
 from django.core.urlresolvers import reverse
 from django.db.models import Q, Max, Count
@@ -1436,7 +1438,21 @@ class DocumentTemplate(models.Model):
     slug = models.SlugField(_("Slug"), max_length=100, unique=True)
     associated_model = models.ForeignKey(ImporterModel)
     template = models.FileField(
-        _("Template"), upload_to="templates/%Y/", help_text=max_size_help())
+        _("Template"), upload_to="templates/%Y/", blank=True, null=True,
+        help_text=max_size_help())
+    label_template = models.FileField(
+        _("Base template for labels"), upload_to="templates/%Y/",
+        blank=True, null=True, help_text=max_size_help())
+    label_targets = models.TextField(
+        _("Labels: targets for labels in the LibreOffice file"),
+        blank=True, null=True,
+        help_text=_("Each target is separated by a semi-colon. The first "
+                    "target is the name of the object including the data in "
+                    "base template. Following targets will be filled with the "
+                    "content of the first target. For instance: "
+                    "\"Cadre1;Cadre2;Cadre3;Cadre4;Cadre5;Cadre6\" for a "
+                    "sheet with 6 labels.")
+    )
     available = models.BooleanField(_("Available"), default=True)
     for_labels = models.BooleanField(_("Used for labels"), default=False)
     label_per_page = models.IntegerField(
@@ -1462,10 +1478,85 @@ class DocumentTemplate(models.Model):
             raise ValidationError(_("For label template, you must provide "
                                     "number of label per page."))
 
+    def generate_label_template(self):
+        if not self.label_template.name or not self.label_targets:
+            return
+        targets = self.label_targets.split(";")
+        base_target = targets[0]
+        try:
+            with zipfile.ZipFile(self.label_template.path) as zip:
+                with zip.open('content.xml') as content:
+                    soup = BeautifulSoup(content.read(), 'xml')
+                    base_content = soup.find(
+                        "draw:frame", attrs={"draw:name": base_target})
+                    if not base_content:
+                        return
+                    base_content = base_content.contents
+        except (FileNotFoundError, zipfile.BadZipFile, KeyError):
+            base_content = None
+        if not base_content:
+            return
+        for idx, target in enumerate(targets[1:]):
+            replace_str = "items." + str(idx + 1)
+            new_content = []
+            for content in base_content:
+                content = copy.copy(content)
+                for text in content.find_all(text=re.compile("items.0")):
+                    fixed_text = text.replace("items.0", replace_str)
+                    text.replace_with(fixed_text)
+                for image in content.find_all(
+                        attrs={"draw:name": re.compile("items.0")}):
+                    image["draw:name"] = image["draw:name"].replace("items.0",
+                                                                    replace_str)
+                new_content.append(content)
+            next_target = soup.find(
+                    "draw:frame", attrs={"draw:name": target})
+            if next_target:
+                next_target.contents = new_content
+
+        with tempfile.TemporaryDirectory() as tmp:
+            sp = self.label_template.name.split(os.sep)[-1].split(".")
+            if len(sp) == 1:  # no extension?
+                sp.append("odt")
+            sp[-2] += "-label"
+            new_filename = ".".join(sp)
+            new_file = os.path.join(tmp, new_filename)
+            with zipfile.ZipFile(new_file, 'w') as zip_out:
+                with zipfile.ZipFile(self.label_template.path, 'r') as zip_in:
+                    zip_out.comment = zip_in.comment
+                    for item in zip_in.infolist():
+                        if item.filename != "content.xml":
+                            zip_out.writestr(item,
+                                             zip_in.read(item.filename))
+            with zipfile.ZipFile(new_file, mode='a',
+                                 compression=zipfile.ZIP_DEFLATED) as zf:
+                zf.writestr("content.xml", str(soup))
+
+            media_dir = "templates/{}/".format(datetime.date.today().year)
+            full_media_dir = os.path.join(settings.MEDIA_ROOT, media_dir)
+            if not os.path.exists(full_media_dir):
+                os.mkdir(full_media_dir)
+            media_file = new_filename
+            idx = 0
+            while os.path.exists(os.path.join(settings.MEDIA_ROOT, media_file)):
+                idx += 1
+                sp = media_file.split(".")
+                sub_sp = sp[-2].split("-label")
+                sub_sp[-1] += str(idx)
+                sp[-2] = "-label".join(sub_sp)
+                media_file = ".".join(sp)
+            with open(new_file, "rb") as file:
+                with ContentFile(file.read()) as file_content:
+                    self.template.save(media_file, file_content)
+                    self.save()
+
     def save(self, *args, **kwargs):
         if not self.slug:
             self.slug = create_slug(DocumentTemplate, self.name)
-        return super(DocumentTemplate, self).save(*args, **kwargs)
+        super(DocumentTemplate, self).save(*args, **kwargs)
+        if self.label_template.name and self.label_targets and not \
+                self.template:
+            self.generate_label_template()
 
     @classmethod
     def get_tuples(cls, dct=None, empty_first=True):