From f94ab37d1f577f3cea14525b40efe3df1f707f7e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fr=C3=A9d=C3=A9ric=20P=C3=A9ters?= <fpeters@entrouvert.com>
Date: Fri, 15 Apr 2022 11:01:02 +0200
Subject: [PATCH] add support for all kinds of categories (#63943)

---
 README                       |  4 ++-
 wcs/imio_import_directory.py | 48 +++++++++++++++++++++++++-----------
 2 files changed, 36 insertions(+), 16 deletions(-)

diff --git a/README b/README
index bcbf1fb..e03d7c0 100644
--- a/README
+++ b/README
@@ -13,7 +13,9 @@ Expected directory layout
 
 * blocks: list of w.c.s. blocks (XML files)
 * carddefs: list of w.c.s. cards (XML files)
-* category: list of w.c.s. categories (XML files)
+* category, carddef_category, workflow_category, block_category, mail_template_category,
+  data_source_category: list of w.c.s. categories, for their respective objects
+  (XML files)
 * datasources: list of w.c.s. data sources (XML files)
 * forms: list of w.c.s. forms (XML files)
 * workflows: list of w.c.s. workflows (XML files)
diff --git a/wcs/imio_import_directory.py b/wcs/imio_import_directory.py
index 32ef6e7..2709f59 100644
--- a/wcs/imio_import_directory.py
+++ b/wcs/imio_import_directory.py
@@ -3,7 +3,14 @@ import xml.etree.ElementTree as ET
 
 from wcs.blocks import BlockDef
 from wcs.carddef import CardDef
-from wcs.categories import Category
+from wcs.categories import (
+    Category,
+    CardDefCategory,
+    WorkflowCategory,
+    BlockCategory,
+    MailTemplateCategory,
+    DataSourceCategory,
+)
 from wcs.data_sources import NamedDataSource
 from wcs.formdef import FormDef
 from wcs.mail_templates import MailTemplate
@@ -41,20 +48,31 @@ class Cmd(Command):
         self.import_formdefs()
 
     def import_categories(self):
-        if not os.path.exists(os.path.join(self.directory, 'category')):
-            return
-        for filename in os.listdir(os.path.join(self.directory, 'category')):
-            category = Category.import_from_xml(open(os.path.join(self.directory, 'category', filename)))
-            try:
-                existing_category = Category.get_by_urlname(category.url_name)
-            except KeyError:
-                category.store()
-            except Exception as e:
-                raise Exception('failed to load existing category %r' % filename) from e
-            else:
-                # replace
-                category.id = existing_category.id
-                category.store()
+        for category_klass in (
+            Category,
+            CardDefCategory,
+            WorkflowCategory,
+            BlockCategory,
+            MailTemplateCategory,
+            DataSourceCategory,
+        ):
+            dirname = category_klass.xml_root_node
+            if not os.path.exists(os.path.join(self.directory, dirname)):
+                continue
+            for filename in os.listdir(os.path.join(self.directory, dirname)):
+                category = category_klass.import_from_xml(
+                    open(os.path.join(self.directory, dirname, filename))
+                )
+                try:
+                    existing_category = category_klass.get_by_urlname(category.url_name)
+                except KeyError:
+                    category.store()
+                except Exception as e:
+                    raise Exception('failed to load existing category %r' % filename) from e
+                else:
+                    # replace
+                    category.id = existing_category.id
+                    category.store()
 
     def import_datasources(self):
         if not os.path.exists(os.path.join(self.directory, 'datasources')):