-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
unified_edit.py
125 lines (104 loc) · 4.69 KB
/
unified_edit.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
# -*- coding: utf-8 -*-
"""
/***************************************************************************
UnifiedEdit
A QGIS plugin
Replaces the layer toggle edit button with a unique edit button that toggles editing on all layers, backed by a transaction.
-------------------
begin : 2015-11-12
git sha : $Format:%H$
copyright : (C) 2015 by Matthias Kuhn, OPENGIS.ch LLC
email : matthias@opengis.ch
***************************************************************************/
/***************************************************************************
* *
* 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. *
* *
***************************************************************************/
"""
from PyQt4.QtCore import QSettings, QTranslator, qVersion, QCoreApplication
from PyQt4.QtGui import QAction, QIcon
from qgis.core import QgsMapLayerRegistry, QgsTransaction, QgsMapLayer
# Initialize Qt resources from file resources.py
import resources
import os.path
class UnifiedEdit:
"""QGIS Plugin Implementation."""
def __init__(self, iface):
"""Constructor.
:param iface: An interface instance that will be passed to this class
which provides the hook by which you can manipulate the QGIS
application at run time.
:type iface: QgsInterface
"""
# Save reference to the QGIS interface
self.iface = iface
# initialize plugin directory
self.plugin_dir = os.path.dirname(__file__)
# initialize locale
locale = QSettings().value('locale/userLocale')[0:2]
locale_path = os.path.join(
self.plugin_dir,
'i18n',
'UnifiedEdit_{}.qm'.format(locale))
if os.path.exists(locale_path):
self.translator = QTranslator()
self.translator.load(locale_path)
QCoreApplication.installTranslator(self.translator)
self.toolbar = self.iface.digitizeToolBar()
self.toggleEditSession = QAction(QIcon(':/edit.svg'), 'Start Transaction', self.toolbar)
self.toggleEditSession.toggled.connect(self.toggleEditSessionToggled)
self.transaction = None
def initGui(self):
"""Create the menu entries and toolbar icons inside the QGIS GUI."""
self.toggleEditSession.setCheckable(True)
self.toolbar.insertAction(
self.iface.actionToggleEditing(), self.toggleEditSession)
self.toolbar.removeAction(self.iface.actionToggleEditing())
def toggleEditSessionToggled(self, checked):
if checked:
self.beginTransaction()
else:
self.commitChanges()
def beginTransaction(self):
layers = list()
for l in QgsMapLayerRegistry.instance().mapLayers().values():
if l.type() != QgsMapLayer.VectorLayer:
continue
if not self.transaction:
self.transaction = QgsTransaction.create([l.id()])
if self.transaction:
layers.append(l)
else:
if self.transaction.addLayer(l.id()):
layers.append(l)
if self.transaction:
self.transaction.begin()
for l in layers:
l.startEditing()
self.toggleEditSession.setIcon(QIcon(':commit-end.svg'))
def commitChanges(self):
for l in QgsMapLayerRegistry.instance().mapLayers().values():
l.commitChanges()
self.transaction.commit()
self.toggleEditSession.setIcon(QIcon(':edit.svg'))
def rollbak(self):
self.transaction.rollBack()
def unload(self):
"""Removes the plugin menu item and icon from QGIS GUI."""
self.toolbar.insertAction(
self.toggleEditSession, self.iface.actionToggleEditing())
def run(self):
"""Run method that performs all the real work"""
# show the dialog
self.dlg.show()
# Run the dialog event loop
result = self.dlg.exec_()
# See if OK was pressed
if result:
# Do something useful here - delete the line containing pass and
# substitute with your code.
pass