index b0140719891603f854ff1d7cb05e2ac16f239eea..263cf592505a6489c307041ab4108d445208b2bf 100644 (file)
#!/usr/bin/env python
'''
-Copyright (C) 2005 Aaron Spike, aaron@ekips.org
+Copyright (C) 2005,2007 Aaron Spike, aaron@ekips.org
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
import inkex, os, base64
-class MyEffect(inkex.Effect):
+class Embedder(inkex.Effect):
def __init__(self):
inkex.Effect.__init__(self)
+ self.OptionParser.add_option("-s", "--selectedonly",
+ action="store", type="inkbool",
+ dest="selectedonly", default=False,
+ help="embed only selected images")
def effect(self):
- ctx = inkex.xml.xpath.Context.Context(self.document,processorNss=inkex.NSS)
-
- # if there is a selection only embed selected images
- # otherwise embed all images
+ # if slectedonly is enabled and there is a selection only embed selected
+ # images. otherwise embed all images
+ if (self.options.selectedonly):
+ self.embedSelected(self.document, self.selected)
+ else:
+ self.embedAll(self.document)
+
+ def embedSelected(self, document, selected):
+ self.document=document
+ self.selected=selected
if (self.options.ids):
- for id, node in self.selected.iteritems():
- if node.tagName == 'image':
+ for id, node in selected.iteritems():
+ if node.tag == inkex.addNS('image','svg'):
self.embedImage(node)
- else:
- path = '//image'
- for node in inkex.xml.xpath.Evaluate(path,self.document, context=ctx):
- self.embedImage(node)
+
+ def embedAll(self, document):
+ self.document=document #not that nice... oh well
+ path = '//svg:image'
+ for node in self.document.getroot().xpath(path, namespaces=inkex.NSS):
+ self.embedImage(node)
+
def embedImage(self, node):
- xlink = node.attributes.getNamedItemNS(inkex.NSS[u'xlink'],'href')
- if (xlink.value[:4]!='data'):
- absref=node.attributes.getNamedItemNS(inkex.NSS[u'sodipodi'],'absref')
- if (os.path.isfile(absref.value)):
- file = open(absref.value,"rb").read()
+ xlink = node.get(inkex.addNS('href','xlink'))
+ if (xlink[:4]!='data'):
+ absref=node.get(inkex.addNS('absref','sodipodi'))
+ href=xlink
+ svg=self.document.getroot().xpath('/svg:svg', namespaces=inkex.NSS)[0]
+ docbase=svg.get(inkex.addNS('docbase','sodipodi'))
+
+ path=''
+ #path selection strategy:
+ # 1. href if absolute
+ # 2. sodipodi:docbase + href
+ # 3. realpath-ified href
+ # 4. absref, only if the above does not point to a file
+ if (href != None):
+ if (os.path.isabs(href)):
+ path=os.path.realpath(href)
+ elif (docbase != None):
+ path=os.path.join(docbase,href)
+ else:
+ path=os.path.realpath(href)
+ if (not os.path.isfile(path)):
+ if (absref != None):
+ path=absref
+ if (not os.path.isfile(path)):
+ inkex.debug('No xlink:href or sodipodi:absref attributes found, or they do not point to an existing file! Unable to embed image.')
+
+ if (os.path.isfile(path)):
+ file = open(path,"rb").read()
embed=True
if (file[:4]=='\x89PNG'):
type='image/png'
elif (file[:2]=='\xff\xd8'):
- type='image/jpg'
+ type='image/jpeg'
+ elif (file[:2]=='BM'):
+ type='image/bmp'
+ elif (file[:6]=='GIF87a' or file[:6]=='GIF89a'):
+ type='image/gif'
+ #ico files lack any magic... therefore we check the filename instead
+ elif(path.endswith('.ico')):
+ type='image/x-icon' #official IANA registered MIME is 'image/vnd.microsoft.icon' tho
else:
embed=False
if (embed):
- xlink.value = 'data:%s;base64,%s' % (type, base64.encodestring(file))
- node.removeAttributeNS(inkex.NSS[u'sodipodi'],'absref')
+ node.set(inkex.addNS('href','xlink'), 'data:%s;base64,%s' % (type, base64.encodestring(file)))
+ if (absref != None):
+ del node.attrib[inkex.addNS('absref',u'sodipodi')]
else:
- inkex.debug("%s is not of type image/png or image/jpg" % absref.value)
+ inkex.debug("%s is not of type image/png, image/jpeg, image/bmp, image/gif or image/x-icon" % path)
else:
- inkex.debug("Sorry we could not locate %s" % absref.value)
-e = MyEffect()
-e.affect()
+ inkex.debug("Sorry we could not locate %s" % path)
+
+if __name__ == '__main__':
+ e = Embedder()
+ e.affect()