class QgsTreePanel(QtGui.QWidget): def __init__(self, explorer): QtGui.QWidget.__init__(self, None) self.explorer = explorer self.projectItem = QgsProjectItem() verticalLayout = QtGui.QVBoxLayout() verticalLayout.setSpacing(2) verticalLayout.setMargin(0) self.toptoolbar = QtGui.QToolBar() self.toptoolbar.setToolButtonStyle(QtCore.Qt.ToolButtonIconOnly) verticalLayout.addWidget(self.toptoolbar) self.tree = ExplorerTreeWidget(self.explorer) verticalLayout.addWidget(self.tree) actions = self.projectItem.contextMenuActions(self.tree, explorer) for action in actions: self.toptoolbar.addAction(action) self.toolbar = QtGui.QToolBar() self.toolbar.setToolButtonStyle(QtCore.Qt.ToolButtonIconOnly) layersIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/layer.png") self.layersAction = QtGui.QAction(layersIcon, "Layers", explorer) self.layersAction.triggered.connect(self.showLayers) stylesIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/style.png") self.stylesAction = QtGui.QAction(stylesIcon, "Styles", explorer) self.stylesAction.triggered.connect(self.showStyles) groupsIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/group.gif") self.groupsAction = QtGui.QAction(groupsIcon, "Groups", explorer) self.groupsAction.triggered.connect(self.showGroups) self.toolbar.addAction(self.layersAction) self.toolbar.addAction(self.stylesAction) self.toolbar.addAction(self.groupsAction) verticalLayout.addWidget(self.toolbar) self.setLayout(verticalLayout) self.lastAction = None self.layersAction.trigger() def currentItem(self): self.tree.currentItem() def showGroups(self): if self.sender() == self.lastAction: return self.tree.clear() groups = qgislayers.getGroups() for group in groups: groupItem = QgsGroupItem(group) groupItem.populate() self.tree.addTopLevelItem(groupItem) self.lastAction = self.sender() def showStyles(self): if self.sender() == self.lastAction: return self.tree.clear() styles = qgislayers.getVectorLayers() for style in styles: styleItem = QgsStyleItem(style) self.tree.addTopLevelItem(styleItem) self.lastAction = self.sender() def showLayers(self): if self.sender() == self.lastAction: return self.tree.clear() layers = qgislayers.getAllLayers() for layer in layers: layerItem = QgsLayerItem(layer) self.tree.addTopLevelItem(layerItem) self.lastAction = self.sender() def refreshContent(self): self.toptoolbar.clear() actions = self.projectItem.contextMenuActions(self.tree, self.explorer) for action in actions: self.toptoolbar.addAction(action) action = self.lastAction self.lastAction = None if action is not None: action.trigger()
class QgsTreePanel(QtGui.QWidget): def __init__(self, explorer): QtGui.QWidget.__init__(self, None) self.explorer = explorer self.projectItem = QgsProjectItem() verticalLayout = QtGui.QVBoxLayout() verticalLayout.setSpacing(2) verticalLayout.setMargin(0) self.toptoolbar = QtGui.QToolBar() self.toptoolbar.setToolButtonStyle(QtCore.Qt.ToolButtonIconOnly) verticalLayout.addWidget(self.toptoolbar) self.tree = ExplorerTreeWidget(self.explorer) verticalLayout.addWidget(self.tree) actions = self.projectItem.contextMenuActions(self.tree, explorer) for action in actions: self.toptoolbar.addAction(action) self.toolbar = QtGui.QToolBar() self.toolbar.setToolButtonStyle(QtCore.Qt.ToolButtonIconOnly) layersIcon = QtGui.QIcon( os.path.dirname(__file__) + "/../images/layer.png") self.layersAction = QtGui.QAction(layersIcon, "Layers", explorer) self.layersAction.triggered.connect(self.showLayers) stylesIcon = QtGui.QIcon( os.path.dirname(__file__) + "/../images/style.png") self.stylesAction = QtGui.QAction(stylesIcon, "Styles", explorer) self.stylesAction.triggered.connect(self.showStyles) groupsIcon = QtGui.QIcon( os.path.dirname(__file__) + "/../images/group.gif") self.groupsAction = QtGui.QAction(groupsIcon, "Groups", explorer) self.groupsAction.triggered.connect(self.showGroups) self.toolbar.addAction(self.layersAction) self.toolbar.addAction(self.stylesAction) self.toolbar.addAction(self.groupsAction) verticalLayout.addWidget(self.toolbar) self.setLayout(verticalLayout) self.lastAction = None self.layersAction.trigger() def currentItem(self): self.tree.currentItem() def showGroups(self): if self.sender() == self.lastAction: return self.tree.clear() groups = qgislayers.getGroups() for group in groups: groupItem = QgsGroupItem(group) groupItem.populate() self.tree.addTopLevelItem(groupItem) self.lastAction = self.sender() def showStyles(self): if self.sender() == self.lastAction: return self.tree.clear() styles = qgislayers.getVectorLayers() for style in styles: styleItem = QgsStyleItem(style) self.tree.addTopLevelItem(styleItem) self.lastAction = self.sender() def showLayers(self): if self.sender() == self.lastAction: return self.tree.clear() layers = qgislayers.getAllLayers() for layer in layers: layerItem = QgsLayerItem(layer) self.tree.addTopLevelItem(layerItem) self.lastAction = self.sender() def refreshContent(self): self.toptoolbar.clear() actions = self.projectItem.contextMenuActions(self.tree, self.explorer) for action in actions: self.toptoolbar.addAction(action) action = self.lastAction self.lastAction = None if action is not None: action.trigger()
class GsTreePanel(QtGui.QWidget): def __init__(self, explorer): QtGui.QWidget.__init__(self, None) self.explorer = explorer self.catalogs = {} verticalLayout = QtGui.QVBoxLayout() verticalLayout.setSpacing(2) verticalLayout.setMargin(0) horizontalLayout = QtGui.QHBoxLayout() horizontalLayout.setSpacing(2) horizontalLayout.setMargin(0) self.comboBox = QtGui.QComboBox() self.comboBox.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum) self.comboBox.currentIndexChanged.connect(self.catalogHasChanged) horizontalLayout.addWidget(self.comboBox) self.addButton = QtGui.QPushButton() self.addButton.clicked.connect(self.addCatalog) addIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/add.png") self.addButton.setIcon(addIcon) self.addButton.setSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Minimum) self.refreshButton = QtGui.QPushButton() self.refreshButton.clicked.connect(self.refreshContent) refreshIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/refresh.png") self.refreshButton.setIcon(refreshIcon) self.refreshButton.setSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Minimum) self.refreshButton.setEnabled(False) horizontalLayout.addWidget(self.refreshButton) horizontalLayout.addWidget(self.addButton) verticalLayout.addLayout(horizontalLayout) self.toptoolbar = QtGui.QToolBar() self.toptoolbar.setToolButtonStyle(QtCore.Qt.ToolButtonIconOnly) self.toptoolbar.setVisible(False) verticalLayout.addWidget(self.toptoolbar) self.tree = ExplorerTreeWidget(self.explorer) verticalLayout.addWidget(self.tree) self.toolbar = QtGui.QToolBar() self.toolbar.setToolButtonStyle(QtCore.Qt.ToolButtonIconOnly) layersIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/layer.png") self.layersAction = QtGui.QAction(layersIcon, "Layers", explorer) self.layersAction.triggered.connect(lambda: self.toggleVisibility((GsLayerItem), self.containerItems[0].contextMenuActions(self.tree, self.explorer))) workspacesIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/workspace.png") self.workspacesAction = QtGui.QAction(workspacesIcon, "Workspaces", explorer) self.workspacesAction.triggered.connect(lambda: self.toggleVisibility((GsWorkspaceItem), self.containerItems[1].contextMenuActions(self.tree, self.explorer))) stylesIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/style.png") self.stylesAction = QtGui.QAction(stylesIcon, "Styles", explorer) self.stylesAction.triggered.connect(lambda: self.toggleVisibility((GsStyleItem), self.containerItems[2].contextMenuActions(self.tree, self.explorer))) groupsIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/group.gif") self.groupsAction = QtGui.QAction(groupsIcon, "Groups", explorer) self.groupsAction.triggered.connect(lambda: self.toggleVisibility((GsGroupItem), self.containerItems[3].contextMenuActions(self.tree, self.explorer))) gwcIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/gwc.png") self.gwcAction = QtGui.QAction(gwcIcon, "GeoWebCache", explorer) self.gwcAction.triggered.connect(lambda: self.toggleVisibility((GwcLayerItem), self.containerItems[4].contextMenuActions(self.tree, self.explorer))) wpsIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/process.png") self.wpsAction = QtGui.QAction(wpsIcon, "Processes", explorer) self.wpsAction.triggered.connect(lambda: self.toggleVisibility((GsProcessItem), self.containerItems[5].contextMenuActions(self.tree, self.explorer))) self.layersAction.setEnabled(False) self.workspacesAction.setEnabled(False) self.stylesAction.setEnabled(False) self.groupsAction.setEnabled(False) self.gwcAction.setEnabled(False) self.wpsAction.setEnabled(False) self.updateActionButtons() #======================================================================= # self.toolbar.addAction(self.layersAction) # self.toolbar.addAction(self.workspacesAction) # self.toolbar.addAction(self.stylesAction) # self.toolbar.addAction(self.groupsAction) # self.toolbar.addAction(self.gwcAction) # self.toolbar.addAction(self.wpsAction) #======================================================================= verticalLayout.addWidget(self.toolbar) self.setLayout(verticalLayout) def updateActionButtons(self): actions = [self.layersAction, self.workspacesAction, self.stylesAction, self.groupsAction, self.gwcAction, self.wpsAction] self.toolbar.clear() for action in actions: button = QtGui.QPushButton() button.setIcon(action.icon()) button.setToolTip(action.text()) button.setEnabled(action.isEnabled()) button.clicked.connect(action.trigger) self.toolbar.addWidget(button) def catalogHasChanged(self): self.catalog = self.comboBox.itemData(self.comboBox.currentIndex()) self.fillTree() self.toggleVisibility((GsLayerItem)) def currentItem(self): self.tree.currentItem() def fillTree(self): self.tree.clear() QtGui.QApplication.setOverrideCursor(QtGui.QCursor(QtCore.Qt.WaitCursor)) self.explorer.setProgressMaximum(7, "Filling explorer tree") try: groups = self.catalog.get_layergroups() for group in groups: groupItem = GsGroupItem(group) groupItem.populate() self.tree.addTopLevelItem(groupItem) self.explorer.setProgress(1) cat = self.catalog try: defaultWorkspace = cat.get_default_workspace() defaultWorkspace.fetch() defaultName = defaultWorkspace.dom.find('name').text except: defaultName = None self.explorer.setProgress(2) workspaces = cat.get_workspaces() for workspace in workspaces: workspaceItem = GsWorkspaceItem(workspace, workspace.name == defaultName) workspaceItem.populate() self.tree.addTopLevelItem(workspaceItem) self.explorer.setProgress(3) styles = self.catalog.get_styles() for style in styles: styleItem = GsStyleItem(style, False) self.tree.addTopLevelItem(styleItem) self.explorer.setProgress(4) layers = self.catalog.get_layers() for layer in layers: layerItem = GsLayerItem(layer) layerItem.populate() self.tree.addTopLevelItem(layerItem) self.explorer.setProgress(5) gwc = Gwc(self.catalog) layers = gwc.layers() for layer in layers: item = GwcLayerItem(layer) self.tree.addTopLevelItem(item) self.explorer.setProgress(6) self.element = Wps(self.catalog) try: processes = self.catalog.processes() for process in processes: item = GsProcessItem(process) self.tree.addTopLevelItem(item) self.explorer.setProgress(7) except: #ignore this section if catalog does not have WPS installed pass self.explorer.resetActivity() self.tree.invisibleRootItem().sortChildren(0, QtCore.Qt.AscendingOrder) #we keep a list of container items, to use their actions self.containerItems = [GsLayersItem(self.catalog), GsWorkspacesItem(self.catalog), GsStylesItem(self.catalog),GsGroupsItem(self.catalog), GwcLayersItem(self.catalog), GsProcessesItem(self.catalog)] finally: self.explorer.setProgress(0) QtGui.QApplication.restoreOverrideCursor() def toggleVisibility(self, visibleItems = (type(None)), visibleActions = []): root = self.tree.invisibleRootItem() for i in range(root.childCount()): item = root.child(i) item.setHidden(not isinstance(item, visibleItems)) self.toptoolbar.setVisible(len(visibleActions)>0) self.toptoolbar.clear() for action in visibleActions: button = QtGui.QPushButton() button.setIcon(action.icon()) button.setToolTip(action.text()) button.setEnabled(action.isEnabled()) button.clicked.connect(action.trigger) self.toptoolbar.addWidget(button) #======================================================================= # for action in visibleActions: # self.toptoolbar.addAction(action) #======================================================================= self.visibleItems = visibleItems self.visibleActions = visibleActions def addCatalog(self, explorer): dlg = DefineCatalogDialog() dlg.exec_() if dlg.ok: try: cat = Catalog(dlg.url, dlg.username, dlg.password) v = cat.gsversion() supported = v.startswith("2.3") or v.startswith("2.4") if not supported: ret = QtGui.QMessageBox.warning(explorer, "GeoServer catalog definition", "The specified catalog seems to be running an older version of GeoServer\n" "That might cause unexpected behaviour.\nDo you want to add the catalog anyway?", QtGui.QMessageBox.Yes | QtGui.QMessageBox.No, QtGui.QMessageBox.No); if ret == QtGui.QMessageBox.No: return name = dlg.name i = 2 while name in self.catalogs.keys(): name = dlg.getName() + "_" + str(i) i += 1 self.catalogs[name] = cat self.catalog = cat self.comboBox.addItem(name, cat) self.layersAction.setEnabled(True) self.workspacesAction.setEnabled(True) self.stylesAction.setEnabled(True) self.groupsAction.setEnabled(True) self.gwcAction.setEnabled(True) self.wpsAction.setEnabled(True) self.refreshButton.setEnabled(True) self.updateActionButtons() self.fillTree() self.layersAction.trigger() self.explorer.updateQgisContent() except Exception, e: explorer.setInfo(str(e), 1) return
class GsTreePanel(QtGui.QWidget): def __init__(self, explorer): QtGui.QWidget.__init__(self, None) self.explorer = explorer self.catalogs = {} verticalLayout = QtGui.QVBoxLayout() verticalLayout.setSpacing(2) verticalLayout.setMargin(0) horizontalLayout = QtGui.QHBoxLayout() horizontalLayout.setSpacing(2) horizontalLayout.setMargin(0) self.comboBox = QtGui.QComboBox() self.comboBox.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum) self.comboBox.currentIndexChanged.connect(self.catalogHasChanged) horizontalLayout.addWidget(self.comboBox) self.addButton = QtGui.QPushButton() self.addButton.clicked.connect(lambda: self.addCatalog(explorer)) addIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/add.png") self.addButton.setIcon(addIcon) self.addButton.setSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Minimum) self.refreshButton = QtGui.QPushButton() self.refreshButton.clicked.connect(self.refreshContent) refreshIcon = QtGui.QIcon( os.path.dirname(__file__) + "/../images/refresh.png") self.refreshButton.setIcon(refreshIcon) self.refreshButton.setSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Minimum) self.refreshButton.setEnabled(False) horizontalLayout.addWidget(self.refreshButton) horizontalLayout.addWidget(self.addButton) verticalLayout.addLayout(horizontalLayout) self.toptoolbar = QtGui.QToolBar() self.toptoolbar.setToolButtonStyle(QtCore.Qt.ToolButtonIconOnly) self.toptoolbar.setVisible(False) verticalLayout.addWidget(self.toptoolbar) self.tree = ExplorerTreeWidget(self.explorer) verticalLayout.addWidget(self.tree) self.toolbar = QtGui.QToolBar() self.toolbar.setToolButtonStyle(QtCore.Qt.ToolButtonIconOnly) layersIcon = QtGui.QIcon( os.path.dirname(__file__) + "/../images/layer.png") self.layersAction = QtGui.QAction(layersIcon, "Layers", explorer) self.layersAction.triggered.connect(lambda: self.toggleVisibility( (GsLayerItem), self.containerItems[0].contextMenuActions( self.tree, self.explorer))) workspacesIcon = QtGui.QIcon( os.path.dirname(__file__) + "/../images/workspace.png") self.workspacesAction = QtGui.QAction(workspacesIcon, "Workspaces", explorer) self.workspacesAction.triggered.connect(lambda: self.toggleVisibility( (GsWorkspaceItem), self.containerItems[1].contextMenuActions( self.tree, self.explorer))) stylesIcon = QtGui.QIcon( os.path.dirname(__file__) + "/../images/style.png") self.stylesAction = QtGui.QAction(stylesIcon, "Styles", explorer) self.stylesAction.triggered.connect(lambda: self.toggleVisibility( (GsStyleItem), self.containerItems[2].contextMenuActions( self.tree, self.explorer))) groupsIcon = QtGui.QIcon( os.path.dirname(__file__) + "/../images/group.gif") self.groupsAction = QtGui.QAction(groupsIcon, "Groups", explorer) self.groupsAction.triggered.connect(lambda: self.toggleVisibility( (GsGroupItem), self.containerItems[3].contextMenuActions( self.tree, self.explorer))) gwcIcon = QtGui.QIcon(os.path.dirname(__file__) + "/../images/gwc.png") self.gwcAction = QtGui.QAction(gwcIcon, "GeoWebCache", explorer) self.gwcAction.triggered.connect(lambda: self.toggleVisibility( (GwcLayerItem), self.containerItems[4].contextMenuActions( self.tree, self.explorer))) wpsIcon = QtGui.QIcon( os.path.dirname(__file__) + "/../images/process.png") self.wpsAction = QtGui.QAction(wpsIcon, "Processes", explorer) self.wpsAction.triggered.connect(lambda: self.toggleVisibility( (GsProcessItem), self.containerItems[5].contextMenuActions( self.tree, self.explorer))) self.layersAction.setEnabled(False) self.workspacesAction.setEnabled(False) self.stylesAction.setEnabled(False) self.groupsAction.setEnabled(False) self.gwcAction.setEnabled(False) self.wpsAction.setEnabled(False) self.updateActionButtons() verticalLayout.addWidget(self.toolbar) self.setLayout(verticalLayout) def updateActionButtons(self): actions = [ self.layersAction, self.workspacesAction, self.stylesAction, self.groupsAction, self.gwcAction, self.wpsAction ] self.toolbar.clear() for action in actions: button = QtGui.QPushButton() button.setIcon(action.icon()) button.setToolTip(action.text()) button.setEnabled(action.isEnabled()) button.clicked.connect(action.trigger) self.toolbar.addWidget(button) def catalogHasChanged(self): self.catalog = self.comboBox.itemData(self.comboBox.currentIndex()) self.fillTree() self.toggleVisibility((GsLayerItem)) def currentItem(self): self.tree.currentItem() def fillTree(self): self.tree.clear() QtGui.QApplication.setOverrideCursor( QtGui.QCursor(QtCore.Qt.WaitCursor)) self.explorer.setProgressMaximum(7, "Filling explorer tree") try: groups = self.catalog.get_layergroups() for group in groups: groupItem = GsGroupItem(group) groupItem.populate() self.tree.addTopLevelItem(groupItem) self.explorer.setProgress(1) cat = self.catalog try: defaultWorkspace = cat.get_default_workspace() defaultWorkspace.fetch() defaultName = defaultWorkspace.dom.find('name').text except: defaultName = None self.explorer.setProgress(2) workspaces = cat.get_workspaces() for workspace in workspaces: workspaceItem = GsWorkspaceItem(workspace, workspace.name == defaultName) workspaceItem.populate() self.tree.addTopLevelItem(workspaceItem) self.explorer.setProgress(3) styles = self.catalog.get_styles() for style in styles: styleItem = GsStyleItem(style, False) self.tree.addTopLevelItem(styleItem) self.explorer.setProgress(4) layers = self.catalog.get_layers() for layer in layers: layerItem = GsLayerItem(layer) layerItem.populate() self.tree.addTopLevelItem(layerItem) self.explorer.setProgress(5) gwc = Gwc(self.catalog) layers = gwc.layers() for layer in layers: item = GwcLayerItem(layer) self.tree.addTopLevelItem(item) self.explorer.setProgress(6) self.element = Wps(self.catalog) try: processes = self.catalog.processes() for process in processes: item = GsProcessItem(process) self.tree.addTopLevelItem(item) self.explorer.setProgress(7) except: #ignore this section if catalog does not have WPS installed pass self.explorer.resetActivity() self.tree.invisibleRootItem().sortChildren( 0, QtCore.Qt.AscendingOrder) #we keep a list of container items, to use their actions self.containerItems = [ GsLayersItem(self.catalog), GsWorkspacesItem(self.catalog), GsStylesItem(self.catalog), GsGroupsItem(self.catalog), GwcLayersItem(self.catalog), GsProcessesItem(self.catalog) ] finally: self.explorer.setProgress(0) QtGui.QApplication.restoreOverrideCursor() def toggleVisibility(self, visibleItems=(type(None)), visibleActions=[]): root = self.tree.invisibleRootItem() for i in range(root.childCount()): item = root.child(i) item.setHidden(not isinstance(item, visibleItems)) self.toptoolbar.setVisible(len(visibleActions) > 0) self.toptoolbar.clear() for action in visibleActions: button = QtGui.QPushButton() button.setIcon(action.icon()) button.setToolTip(action.text()) button.setEnabled(action.isEnabled()) button.clicked.connect(action.trigger) self.toptoolbar.addWidget(button) self.visibleItems = visibleItems self.visibleActions = visibleActions def addCatalog(self, explorer): dlg = DefineCatalogDialog(explorer) dlg.exec_() if dlg.ok: try: cat = Catalog(dlg.url, dlg.username, dlg.password) v = cat.gsversion() supported = v.startswith("2.3") or v.startswith("2.4") if not supported: ret = QtGui.QMessageBox.warning( explorer, "GeoServer catalog definition", "The specified catalog seems to be running an older version of GeoServer\n" "That might cause unexpected behaviour.\nDo you want to add the catalog anyway?", QtGui.QMessageBox.Yes | QtGui.QMessageBox.No, QtGui.QMessageBox.No) if ret == QtGui.QMessageBox.No: return self.catalogs[dlg.name] = cat self.catalog = cat self.comboBox.addItem(dlg.name, cat) self.layersAction.setEnabled(True) self.workspacesAction.setEnabled(True) self.stylesAction.setEnabled(True) self.groupsAction.setEnabled(True) self.gwcAction.setEnabled(True) self.wpsAction.setEnabled(True) self.refreshButton.setEnabled(True) self.updateActionButtons() self.fillTree() self.layersAction.trigger() self.explorer.updateQgisContent() except Exception, e: explorer.setError(str(e)) return