ca1f799a7d
Many Path object forward its editing operation to the job object. It is possible that some editing option change may lead to removing of the editing object (e.g. changing stock type) causing crash. It is possible to just open the task panel without setting any editing object by returning False in ViewProvider.setEdit()
118 lines
4.6 KiB
Python
118 lines
4.6 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__ = "https://www.freecadweb.org"
|
|
__doc__ = "ViewProvider who's main and only task is to assign an icon."
|
|
|
|
PathLog.setLevel(PathLog.Level.INFO, PathLog.thisModule())
|
|
#PathLog.trackModule(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 False
|
|
|
|
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
|
|
|