add apply attribute extension (#3983)

1 files changed, 47 insertions, 0 deletions

diff --git a/lib/extensions/apply_attribute.py b/lib/extensions/apply_attribute.py
new file mode 100644
index 00000000..0d3e30e6
--- /dev/null
+++ b/lib/extensions/apply_attribute.py
@@ -0,0 +1,47 @@
+# Authors: see git history
+#
+# Copyright (c) 2025 Authors
+# Licensed under the GNU GPL version 3.0 or later.  See the file LICENSE for details.
+
+from inkex import Boolean, errormsg
+
+from ..i18n import _
+from .base import InkstitchExtension
+
+
+class ApplyAttribute(InkstitchExtension):
+    '''
+    Applies a given attribute to all selected elements
+    '''
+    def __init__(self, *args, **kwargs):
+        InkstitchExtension.__init__(self, *args, **kwargs)
+        self.arg_parser.add_argument("--notebook")
+        self.arg_parser.add_argument("-n", "--namespace", dest="namespace", type=str, default='inkstitch')
+        self.arg_parser.add_argument("-k", "--key", dest="key", type=str, default='')
+        self.arg_parser.add_argument("-v", "--value", dest="value", type=str, default='')
+        self.arg_parser.add_argument("-r", "--remove", dest="remove", type=Boolean, default=False)
+
+    def effect(self):
+        self.get_elements()
+        if not self.elements:
+            errormsg(_("Please select at least one element."))
+            return
+
+        if not self.options.key:
+            errormsg(_("Please enter the attribute name."))
+            return
+
+        key = ''
+        if self.options.namespace:
+            key = f'{self.options.namespace}:'
+        key += self.options.key
+
+        if self.options.remove:
+            for element in self.elements:
+                element.node.pop(key)
+        else:
+            if not self.options.value:
+                errormsg(_("Please enter a value."))
+                return
+            for element in self.elements:
+                element.node.set(key, self.options.value)