125 lines
4.8 KiB
Python
125 lines
4.8 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 PathScripts.PathLog as PathLog
|
|
import PathScripts.PathUtil as PathUtil
|
|
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."
|
|
|
|
LOGLEVEL = False
|
|
|
|
if LOGLEVEL:
|
|
PathLog.setLevel(PathLog.Level.DEBUG, PathLog.thisModule())
|
|
PathLog.trackModule(PathLog.thisModule())
|
|
else:
|
|
PathLog.setLevel(PathLog.Level.INFO, PathLog.thisModule())
|
|
|
|
class ViewProvider(object):
|
|
'''Generic view provider to assign an icon.'''
|
|
|
|
def __init__(self, vobj, icon):
|
|
self.icon = icon
|
|
self.attach(vobj)
|
|
|
|
self.editModule = None
|
|
self.editCallback = None
|
|
|
|
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=None, mode=0):
|
|
# pylint: disable=unused-argument
|
|
if 0 == mode:
|
|
self._onEditCallback(True)
|
|
return True
|
|
|
|
def unsetEdit(self, arg1, arg2):
|
|
# pylint: disable=unused-argument
|
|
self._onEditCallback(False)
|
|
|
|
def setupContextMenu(self, vobj, menu):
|
|
# pylint: disable=unused-argument
|
|
PathLog.track()
|
|
from PySide import QtCore, QtGui
|
|
edit = QtCore.QCoreApplication.translate('Path', 'Edit', None)
|
|
action = QtGui.QAction(edit, menu)
|
|
action.triggered.connect(self.setEdit)
|
|
menu.addAction(action)
|
|
|
|
_factory = {}
|
|
|
|
def Attach(vobj, name):
|
|
'''Attach(vobj, name) ... attach the appropriate view provider to the view object.
|
|
If no view provider was registered for the given name a default IconViewProvider is created.'''
|
|
|
|
PathLog.track(vobj.Object.Label, name)
|
|
global _factory # pylint: disable=global-statement
|
|
for key,value in PathUtil.keyValueIter(_factory):
|
|
if key == name:
|
|
return value(vobj, name)
|
|
PathLog.track(vobj.Object.Label, name, 'PathIconViewProvider')
|
|
return ViewProvider(vobj, name)
|
|
|
|
def RegisterViewProvider(name, provider):
|
|
'''RegisterViewProvider(name, provider) ... if an IconViewProvider is created for an object with the given name
|
|
an instance of provider is used instead.'''
|
|
|
|
PathLog.track(name)
|
|
global _factory # pylint: disable=global-statement
|
|
_factory[name] = provider
|
|
|