✨ exhibitions: "GAM" export

1 files changed, 36 insertions, 1 deletions

diff --git a/archaeological_finds/models_treatments.py b/archaeological_finds/models_treatments.py
index d687199db..ce9066fc7 100644
--- a/archaeological_finds/models_treatments.py
+++ b/archaeological_finds/models_treatments.py
@@ -18,6 +18,12 @@
 # See the file COPYING for details.
 
 import datetime
+import lxml.etree
+import lxml.builder
+import os
+import shutil
+import tempfile
+import zipfile
 
 from django.conf import settings
 from django.contrib.gis.db import models
@@ -1379,6 +1385,36 @@ class TreatmentFile(
     def natural_key(self):
         return (self.year, self.index)
 
+    def is_gam_exportable(self):
+        self.gam_errors = []
+        is_ok = True
+        return is_ok
+
+    def gam_export(self):
+        filename = f"export_gam-{self.exhibition_start_date.year}-{slugify(self.name).replace('-', '_')}.zip"
+        maker = lxml.builder.ElementMaker()
+        document = maker.proposition(
+            maker.action("CREATION"),
+            maker.categorie("PRET"),
+        )
+        content = b'<?xml version="1.0" encoding="UTF-8" standalone="yes"?>\n'
+        content += lxml.etree.tostring(document, pretty_print=True)
+        gam_dir = os.path.join(settings.MEDIA_ROOT, "GAM")
+        if not os.path.exists(gam_dir):
+            os.mkdir(gam_dir)
+        media_dir = os.path.join("GAM", str(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)
+        final_name = os.path.join(full_media_dir, filename)
+        with tempfile.TemporaryDirectory() as tmp_dir_name:
+            new_file = os.path.join(tmp_dir_name, filename)
+            with zipfile.ZipFile(
+                 new_file, mode="a", compression=zipfile.ZIP_DEFLATED) as zf:
+                zf.writestr("pret.xml", content)
+            shutil.move(new_file, final_name)
+        return f"{settings.MEDIA_URL}{media_dir}/{filename}"
+
     @property
     def short_class_name(self):
         return _("Treatment request")
@@ -1598,7 +1634,6 @@ class Exhibition(
             request, "archaeological_finds.add_treatmentfile"
         )
         if can_add_tf:
-
             actions += [
                 (
                     reverse("exhibition-qa-add-loan", args=[self.pk]),