Newer
Older
GB_Printer / Dump / share / extensions / extractimage.py
#!/usr/bin/env python 
'''
Copyright (C) 2005 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
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
'''
# standard library
import base64
import os
# local library
import inkex

inkex.localize()

class MyEffect(inkex.Effect):
    def __init__(self):
        inkex.Effect.__init__(self)
        self.OptionParser.add_option("--desc")
        self.OptionParser.add_option("--filepath",
                        action="store", type="string", 
                        dest="filepath", default=None,
                        help="")
    def effect(self):
        mimesubext={
            'png' :'.png',
            'bmp' :'.bmp',
            'jpeg':'.jpg',
            'jpg' :'.jpg', #bogus mime
            'icon':'.ico',
            'gif' :'.gif'
        }
        
        # exbed the first embedded image
        path = self.options.filepath
        if (path != ''):
            if (self.options.ids):
                for id, node in self.selected.iteritems():
                    if node.tag == inkex.addNS('image','svg'):
                        xlink = node.get(inkex.addNS('href','xlink'))
                        if (xlink[:4]=='data'):
                            comma = xlink.find(',')
                            if comma>0:
                                #get extension
                                fileext=''
                                semicolon = xlink.find(';')
                                if semicolon>0:
                                    for sub in mimesubext.keys():
                                        if sub in xlink[5:semicolon].lower():
                                            fileext=mimesubext[sub]
                                            path=path+fileext;
                                            if (not os.path.isabs(path)):
                                                if os.name == 'nt':
                                                    path = os.path.join(os.environ['USERPROFILE'],path)
                                                else:
                                                    path = os.path.join(os.path.expanduser("~"),path)
                                            inkex.errormsg(_('Image extracted to: %s') % path)
                                            break 
                                #save
                                data = base64.decodestring(xlink[comma:])
                                open(path,'wb').write(data)
                                node.set(inkex.addNS('href','xlink'),os.path.realpath(path)) #absolute for making in-mem cycles work
                            else:
                                inkex.errormsg(_('Unable to find image data.'))
                            break

if __name__ == '__main__':
    e = MyEffect()
    e.affect()


# vim: expandtab shiftwidth=4 tabstop=8 softtabstop=4 fileencoding=utf-8 textwidth=99