75 lines
3.1 KiB
Python
75 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# ***************************************************************************
|
|
# * *
|
|
# * Copyright (c) 2017 sliptonic <shopinthewoods@gmail.com> *
|
|
# * *
|
|
# * This program is free software; you can redistribute it and/or modify *
|
|
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
# * as published by the Free Software Foundation; either version 2 of *
|
|
# * the License, or (at your option) any later version. *
|
|
# * for detail see the LICENCE text file. *
|
|
# * *
|
|
# * 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 Library General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Library 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 *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
import importlib
|
|
|
|
__title__ = "Path Icon ViewProvider"
|
|
__author__ = "sliptonic (Brad Collette)"
|
|
__url__ = "http://www.freecadweb.org"
|
|
__doc__ = "ViewProvider who's main and only task is to assign an icon."
|
|
|
|
class ViewProvider(object):
|
|
'''Generic view provider to assign an icon.'''
|
|
|
|
def __init__(self, vobj, icon):
|
|
self.icon = icon
|
|
vobj.Proxy = self
|
|
|
|
def attach(self, vobj):
|
|
self.vobj = vobj
|
|
self.obj = vobj.Object
|
|
|
|
def __getstate__(self):
|
|
attrs = {'icon': self.icon }
|
|
if hasattr(self, 'editModule'):
|
|
attrs['editModule'] = self.editModule
|
|
attrs['editCallback'] = self.editCallback
|
|
return attrs
|
|
|
|
def __setstate__(self, state):
|
|
self.icon = state['icon']
|
|
if state.get('editModule', None):
|
|
self.editModule = state['editModule']
|
|
self.editCallback = state['editCallback']
|
|
|
|
def getIcon(self):
|
|
return ":/icons/Path-{}.svg".format(self.icon)
|
|
|
|
def onEdit(self, callback):
|
|
self.editModule = callback.__module__
|
|
self.editCallback = callback.__name__
|
|
|
|
def _onEditCallback(self, edit):
|
|
if hasattr(self, 'editModule'):
|
|
mod = importlib.import_module(self.editModule)
|
|
callback = getattr(mod, self.editCallback)
|
|
callback(self.obj, self.vobj, edit)
|
|
|
|
def setEdit(self, vobj, mode=0):
|
|
self._onEditCallback(True)
|
|
|
|
def unsetEdit(self, arg1, arg2):
|
|
self._onEditCallback(False)
|
|
|