mightyscape-1.2/extensions/fablabchemnitz/edit_attributes/edit_attributes.py

40 lines
1.5 KiB
Python
Raw Normal View History

2022-10-03 03:07:44 +02:00
#!/usr/bin/env python3
import inkex
import sys
class EditAttributes(inkex.EffectExtension):
def add_arguments(self, pars):
pars.add_argument("-a", "--attributeName", help="attribute name to set")
pars.add_argument("-v", "--attributeValue", help="attribute value to set")
pars.add_argument("-m", "--mode", default="set", help="mode of operation")
def effect(self):
if not self.options.attributeName: # if attributeName is not given
inkex.errormsg("Attribute name not given")
return
if not self.options.attributeValue: # required to make proper behaviour
inkex.errormsg("Please define proper attribute value")
return
elements = self.svg.selected.values()
for el in elements:
currentAtt = el.attrib.get(self.options.attributeName)
if currentAtt is None:
currentAtt = ""
if self.options.mode == "set":
el.set(self.options.attributeName, self.options.attributeValue)
elif self.options.mode == "append":
el.attrib[self.options.attributeName] = currentAtt + self.options.attributeValue
elif self.options.mode == "prefix":
el.attrib[self.options.attributeName] = self.options.attributeValue + currentAtt
elif self.options.mode == "subtract":
el.attrib[self.options.attributeName] = currentAtt.replace(self.options.attributeValue, "")
elif self.options.mode == "remove":
if self.options.attributeName in el.attrib:
del el.attrib[self.options.attributeName]
else:
inkex.errormsg("Invalid mode: " + self.options.mode)
if __name__ == '__main__':
EditAttributes().run()