diff options
Diffstat (limited to 'share/extensions/replace_font.py')
| -rw-r--r-- | share/extensions/replace_font.py | 260 |
1 files changed, 260 insertions, 0 deletions
diff --git a/share/extensions/replace_font.py b/share/extensions/replace_font.py new file mode 100644 index 000000000..f0f25ab8e --- /dev/null +++ b/share/extensions/replace_font.py @@ -0,0 +1,260 @@ +#!/usr/bin/env python +''' +replace_font.py + +Copyright (C) 2010 Craig Marshall, craig9 [at] gmail.com + +This program is free software; you can redistribute it and/or modify +it under the terms of the GNU General Public License as published by +the Free Software Foundation; either version 2 of the License, or +(at your option) any later version. + +This program is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. + +You should have received a copy of the GNU General Public License +along with this program; if not, write to the Free Software +Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + +----------------------- + +This script finds all fonts in the current drawing that match the +specified find font, and replaces them with the specified replacement +font. + +It can also replace all fonts indiscriminately, and list all fonts +currently being used. +''' + +import os +import sys +import inkex +import simplestyle +import gettext +_ = gettext.gettext + +text_tags = ['{http://www.w3.org/2000/svg}tspan', + '{http://www.w3.org/2000/svg}text', + '{http://www.w3.org/2000/svg}flowRoot', + '{http://www.w3.org/2000/svg}flowPara', + '{http://www.w3.org/2000/svg}flowSpan'] +font_attributes = ['font-family', '-inkscape-font-specification'] + +def set_font(node, new_font, style=None): + ''' + Sets the font attribute in the style attribute of node, using the + font name stored in new_font. If the style dict is open already, + it can be passed in, otherwise it will be optned anyway. + + Returns a dirty boolean flag + ''' + dirty = False + if not style: + style = get_style(node) + if style: + for att in font_attributes: + if att in style: + style[att] = new_font + set_style(node, style) + dirty = True + return dirty + +def find_replace_font(node, find, replace): + ''' + Searches the relevant font attributes/styles of node for find, and + replaces them with replace. + + Returns a dirty boolean flag + ''' + dirty = False + style = get_style(node) + if style: + for att in font_attributes: + if att in style and style[att].strip().lower() == find: + set_font(node, replace, style) + dirty = True + return dirty + +def is_styled_text(node): + ''' + Returns true if the tag in question is a "styled" element that + can hold text. + ''' + return node.tag in text_tags and 'style' in node.attrib + +def is_text(node): + ''' + Returns true if the tag in question is an element that + can hold text. + ''' + return node.tag in text_tags + + +def get_style(node): + ''' + Sugar coated way to get style dict from a node + ''' + if 'style' in node.attrib: + return simplestyle.parseStyle(node.attrib['style']) + +def set_style(node, style): + ''' + Sugar coated way to set the style dict, for node + ''' + node.attrib['style'] = simplestyle.formatStyle(style) + +def get_fonts(node): + ''' + Given a node, returns a list containing all the fonts that + the node is using. + ''' + fonts = [] + s = get_style(node) + if not s: + return fonts + for a in font_attributes: + if a in s: + fonts.append(s[a]) + return fonts + +def die(msg = "Dying!"): + inkex.errormsg(msg) + sys.exit(0) + +def report_replacements(num): + ''' + Sends a message to the end user showing success of failure + of the font replacement + ''' + if num == 0: + die(_('Couldn\'t find anything using that font, please ensure the spelling and spacing is correct.')) + +def report_findings(findings): + ''' + Tells the user which fonts were found, if any + ''' + if len(findings) == 0: + inkex.errormsg(_("Didn't find any fonts in this document/selection.")) + else: + if len(findings) == 1: + inkex.errormsg(_("Found the following font only: %s") % findings[0]) + else: + inkex.errormsg(_("Found the following fonts:\n%s") % '\n'.join(findings)) + +class ReplaceFont(inkex.Effect): + ''' + Replaces all instances of one font with another + ''' + def __init__(self): + inkex.Effect.__init__(self) + self.OptionParser.add_option("--fr_find", action="store", + type="string", dest="fr_find", + default=None, help="") + + self.OptionParser.add_option("--fr_replace", action="store", + type="string", dest="fr_replace", + default=None, help="") + + self.OptionParser.add_option("--r_replace", action="store", + type="string", dest="r_replace", + default=None, help="") + + self.OptionParser.add_option("--action", action="store", + type="string", dest="action", + default=None, help="") + + self.OptionParser.add_option("--scope", action="store", + type="string", dest="scope", + default=None, help="") + + def find_child_text_items(self, node): + ''' + Recursive method for appending all text-type elements + to self.selected_items + ''' + if is_text(node): + self.selected_items.append(node) + for child in node: + self.find_child_text_items(child) + + def relevant_items(self, scope): + ''' + Depending on the scope, returns all text elements, or all + selected text elements including nested children + ''' + items = [] + to_return = [] + if scope == "selection_only": + self.selected_items = [] + for item in self.selected.iteritems(): + self.find_child_text_items(item[1]) + items = self.selected_items + if len(items) == 0: + die(_("There was nothing selected")) + else: + items = self.document.getroot().getiterator() + to_return.extend(filter(is_text, items)) + return to_return + + def find_replace(self, nodes, find, replace): + ''' + Walks through nodes, replacing fonts as it goes according + to find and replace + ''' + replacements = 0 + for node in nodes: + if find_replace_font(node, find, replace): + replacements += 1 + report_replacements(replacements) + + def replace_all(self, nodes, replace): + ''' + Walks through nodes, setting fonts indiscriminately. + ''' + replacements = 0 + for node in nodes: + if set_font(node, replace): + replacements += 1 + report_replacements(replacements) + + def list_all(self, nodes): + ''' + Walks through nodes, building a list of all fonts found, then + reports to the user with that list + ''' + fonts_found = [] + for node in nodes: + for f in get_fonts(node): + if not f in fonts_found: + fonts_found.append(f) + report_findings(sorted(fonts_found)) + + def effect(self): + action = self.options.action.strip("\"") # TODO Is this a bug? (Extra " characters) + scope = self.options.scope + + relevant_items = self.relevant_items(scope) + + if action == "find_replace": + find = self.options.fr_find + if find is None or find == "": + die(_("Please enter a search string in the find box.")); + find = find.strip().lower() + replace = self.options.fr_replace + if replace is None or replace == "": + die(_("Please enter a replacement font in the replace with box.")); + self.find_replace(relevant_items, find, replace) + elif action == "replace_all": + replace = self.options.r_replace + if replace is None or replace == "": + die(_("Please enter a replacement font in the replace all box.")); + self.replace_all(relevant_items, replace) + elif action == "list_only": + self.list_all(relevant_items) + sys.exit(0) + +if __name__ == "__main__": + e = ReplaceFont() + e.affect() |