def __init__(self, iface):

# Save reference to the QGIS interface

self.iface = iface

# initialize plugin directory

dir = os.path.dirname(__file__)

self.plugin_dir = dir.decode('cp1251')

self.labProp = None

self.fed = None

self.scrc = None

self.codename = None

self.model = None

self.okdict = {}

self.root = QgsProject.instance().layerTreeRoot()

self.style = QgsStyleV2().defaultStyle()

self.dlg = ViewShpForRscCodeDialog()

self.editRend = EditRenderDialog()

self.editRend.listWgt.doubleClicked.connect(self.editSymbol)

self.editRend.closeButton.clicked.connect(self.closeEditRend)

self.editRend.csvExpButton.clicked.connect(self.expCSV)

self.editRend.xmlExpButton.clicked.connect(self.expXML)

self.editRend.saveButton.clicked.connect(functools.partial(self.expCSV, True))

self.editRend.delButton.clicked.connect(self.removeCodeItem)

self.editRend.addButton.clicked.connect(self.addCodeItem)

self.dlg.runButton.clicked.connect(self.setMapStyle)

self.dlg.closeButton.clicked.connect(self.closeDial)

self.dlg.importXmlButton.clicked.connect(self.importXML)

self.dlg.editRenderButton.clicked.connect(self.editRender)

self.dlg.delTabButton.clicked.connect(self.removeTab)

# initialize locale

locale = QSettings().value('locale/userLocale')[0:2]

locale_path = os.path.join(

self.plugin_dir,

'i18n',

'MapStyling_{}.qm'.format(locale))

if os.path.exists(locale_path):

self.translator = QTranslator()

self.translator.load(locale_path)

if qVersion() > '4.3.3':

QCoreApplication.installTranslator(self.translator)

# Declare instance attributes

self.actions = []

self.menu = self.tr(u'&ViewShpForRscCodePlugin')

self.toolbar = self.iface.addToolBar(u'MapStyling')

self.toolbar.setObjectName(u'MapStyling')

def tr(self, message):

return QCoreApplication.translate('MapStyling', message)

def add_action(

self,

icon_path,

text,

callback,

enabled_flag=True,

add_to_menu=True,

add_to_toolbar=True,

status_tip=None,

whats_this=None,

parent=None):

"""Add a toolbar icon to the toolbar.

:param icon_path: Path to the icon for this action. Can be a resource

path (e.g. ':/plugins/foo/bar.png') or a normal file system path.

:type icon_path: str

:param text: Text that should be shown in menu items for this action.

:type text: str

:param callback: Function to be called when the action is triggered.

:type callback: function

:param enabled_flag: A flag indicating if the action should be enabled

by default. Defaults to True.

:type enabled_flag: bool

:param add_to_menu: Flag indicating whether the action should also

be added to the menu. Defaults to True.

:type add_to_menu: bool

:param add_to_toolbar: Flag indicating whether the action should also

be added to the toolbar. Defaults to True.

:type add_to_toolbar: bool

:param status_tip: Optional text to show in a popup when mouse pointer

hovers over the action.

:type status_tip: str

:param parent: Parent widget for the new action. Defaults None.

:type parent: QWidget

:param whats_this: Optional text to show in the status bar when the

mouse pointer hovers over the action.

:returns: The action that was created. Note that the action is also

added to self.actions list.

:rtype: QAction

"""

# Create the dialog (after translation) and keep reference

icon = QIcon(icon_path)

action = QAction(icon, text, parent)

action.triggered.connect(callback)

action.setEnabled(enabled_flag)

if status_tip is not None:

action.setStatusTip(status_tip)

if whats_this is not None:

action.setWhatsThis(whats_this)

if add_to_toolbar:

self.toolbar.addAction(action)

if add_to_menu:

self.iface.addPluginToMenu(

self.menu,

action)

self.actions.append(action)

return action

def GetModelForCombobox(self, list):

model = QtGui.QStandardItemModel(1, 1)

n = 0

for l in list:

item_str = l.rpartition('\\')[2]

item = QtGui.QStandardItem(item_str)

model.setItem(n, 0, item)

n = n + 1

return model

def initGui(self):

"""Create the menu entries and toolbar icons inside the QGIS GUI."""

icon_path = ':/plugins/ViewShpForRscCode/icon.png'

self.add_action(

icon_path,

text=self.tr('ViewShpForRscCode'),

callback=self.run,

parent=self.iface.mainWindow())

def unload(self):

"""Removes the plugin menu item and icon from QGIS GUI."""

for action in self.actions:

self.iface.removePluginMenu(

self.tr(u'&MViewShpForRscCodePlugin'),

action)

self.iface.removeToolBarIcon(action)

# remove the toolbar

del self.toolbar

QgsExpression.unregisterFunction("fontProperty")

def run(self):

"""Run method that performs all the real work"""

# show the dialog

#self.dlg.show()

# Run the dialog event loop

self.fillCombos()

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

def fillCombos(self):

xml = glob.glob(self.plugin_dir + '/svg/*.xml')

tab = glob.glob(self.plugin_dir + '/tab/*.csv')

self.dlg.scaleComboBox.setModel(self.GetModelForCombobox(xml))

self.dlg.tabComboBox.setModel(self.GetModelForCombobox(tab))

def reorderLayers(self, layer, node):

lyr = self.root.findLayer(layer.id())

lyrClone = lyr.clone()

parent = lyr.parent()

parent.insertChildNode(node, lyrClone)

parent.removeChildNode(lyr)

def setMapStyle(self):

layers = self.iface.legendInterface().layers()

self.scrc = SymbolCategoryRenderClass(self.makeokdict(self.getTabs(csv=True)))

self.labProp = LabelProperties(self.okdict)

for layer in layers:

layerType = layer.type()

if layerType == QgsMapLayer.VectorLayer:

node = len(layers)

if layer.name() == u'LAYER3-polygon':

self.reorderLayers(layer, node)

if layer.name() == u'LAYER7-polygon':

self.reorderLayers(layer, node-1)

if layer.name() == u'LAYER12-polygon':

self.reorderLayers(layer, node-2)

if layer.name() == u'LAYER19-polygon':

self.reorderLayers(layer, node-3)

if layer.name() == u'LAYER17-label':

self.labProp.makeLabels(layer)

self.scrc.CategoryRender(self.style,layer)

layer.triggerRepaint()

def getTabs(self, csv=None, lbl=None):

if csv:

tabName = self.dlg.tabComboBox.currentText()

csv_path = os.path.join(

self.plugin_dir, 'tab/' + tabName)

return csv_path

def makeokdict(self,csv_path):

with open(csv_path) as f:

reader = csv.reader(f, delimiter =';')

for row in reader:

if 'T' in row[0]:

self.okdict[row[0]] = [row[1].decode('utf-8'), row[2].decode('utf-8').split(',')]

else:

self.okdict[row[0]] = [row[1].decode('utf-8'),row[2].decode('utf-8')]

return self.okdict

def importXML(self):

srcStyle = QgsStyleV2()

dstStyle = self.style

symLib = self.dlg.scaleComboBox.currentText()

svg_dst = QgsApplication.svgPaths()[0]

spath = self.plugin_dir + '/svg/'

copylist = []

if not os.path.isfile(spath):

for path, dirs, filenames in os.walk(spath):

for directory in dirs:

dir_path = os.path.join(svg_dst, directory)

if not os.path.exists(dir_path):

os.makedirs(os.path.join(dir_path))

for sfile in filenames:

s_file = os.path.join(path, sfile)

d_file = os.path.join(path.replace(spath, svg_dst), sfile)

copylist.append([s_file, d_file])

for c in copylist:

shutil.copy(c[0], c[1])

xml_path = spath + symLib

srcStyle.importXML(xml_path)

groupName = symLib.replace(".xml", "")

if groupName not in dstStyle.groupNames():

dstStyle.addGroup(groupName)

groupid = dstStyle.groupId(groupName)

for sym in srcStyle.symbolNames():

symbol = srcStyle.symbol(sym)

dstStyle.addSymbol(sym, symbol)

dstStyle.saveSymbol(sym, symbol, groupid, [])

QtGui.QMessageBox.information(QWidget(), u'Информация',

u'Импорт завершен', QtGui.QMessageBox.Ok

)

return

def editRender(self):

tab = self.dlg.tabComboBox.currentText()

self.editRend.listWgt.setIconSize(QSize(32, 32))

self.editRend.listWgt.setEditTriggers(QtGui.QAbstractItemView.SelectedClicked)

self.model = QStandardItemModel()

self.model.setColumnCount(3)

self.model.setHorizontalHeaderItem(0, QStandardItem(u'Код'))

self.model.setHorizontalHeaderItem(1, QStandardItem(u'Название'))

self.model.setHorizontalHeaderItem(2, QStandardItem(u'Знак'))

self.editRend.listWgt.setModel(self.model)

self.editRend.listWgt.setSelectionBehavior(QTableView.SelectRows)

self.codename = {}

with open(self.plugin_dir + '/tab/' + tab) as f:

reader = csv.reader(f, delimiter=';')

for row in reader:

itemCode = QStandardItem(row[0])

itemName = QStandardItem(row[1].decode('utf-8'))

symName = row[2].decode('utf-8')

if symName == 'None':

if 'L' in itemCode.text():

symName = 'NoneStyleLine'

if 'P' in itemCode.text():

symName = 'NoneStyleMark'

if 'S' in itemCode.text():

symName = 'NoneStylePoly'

if 'V' in itemCode.text():

symName = 'NoneStyleLine'

itemSymb = QStandardItem(symName)

symbol = self.style.symbol(symName)

if symbol:

itemSymb = QStandardItem(symName)

icon = QgsSymbolLayerV2Utils.symbolPreviewIcon(symbol, self.editRend.listWgt.iconSize())

itemSymb.setIcon(icon)

self.model.appendRow([itemCode, itemName, itemSymb])

self.codename[itemCode.text()] = symName

self.editRend.listWgt.resizeRowsToContents()

self.editRend.listWgt.setColumnWidth(2, 125)

self.editRend.listWgt.setColumnWidth(1, 250)

self.editRend.exec_()

def editSymbol(self):

row = self.editRend.listWgt.selectionModel().currentIndex().row()

code = self.model.item(row, 0).text()

name = self.model.item(row, 2).text()

if name == 'None':

t = QInputDialog.getItem(QWidget(), u'Создание условного знака',

u'Укажите тип объекта',

[u'Точка', u'Линия', u'Полигон', u'Текст'],editable = False)

if t[1]:

if t[0] == u'Точка':

name = 'NoneStyleMark'

if t[0] == u'Линия':

name = 'NoneStyleLine'

if t[0] == u'Полигон':

name = 'NoneStylePoly'

if t[0] == u'Текст':

name = None

else:

return

if 'T' in code:

fontProps = self.fontDialog(name)

if fontProps:

item = QStandardItem(','.join(fontProps))

self.model.setItem(row, 2, item)

return

symbol = self.style.symbol(name)

if not symbol:

return

d = QgsSymbolV2SelectorDialog(symbol, self.style, None)

if d.exec_() == 0:

return

if 'None' in name:

name = ''

newName = QInputDialog.getText(QWidget(),u"Имя условного знака", u"Введите имя условного знака:",text=name)

if newName[1]:

name = newName[0]

if not newName[1]:

return

self.style.addSymbol(name, symbol, True)

itemSymb = QStandardItem(name)

icon = QgsSymbolLayerV2Utils.symbolPreviewIcon(symbol, self.editRend.listWgt.iconSize())

itemSymb.setIcon(icon)

self.model.setItem(row, 2, itemSymb)

def fontDialog(self,name=False):

fsc = None

if name:

fsc = name.split(',')

self.fed = fontEditDialog(fsc)

return self.fed.showDial()

def removeTab(self):

tab = self.dlg.tabComboBox.currentText()

if tab == 'codeNameTab_100k.csv':

QtGui.QMessageBox.information(QWidget(), u'Информация',

u'Невозможно удалить этот файл', QtGui.QMessageBox.Ok

)

return

tpath = self.plugin_dir + '/tab/' + tab

os.remove(tpath)

self.fillCombos()

def closeDial(self):

self.dlg.close()

def closeEditRend(self):

self.editRend.close()

self.fillCombos()

def expXML(self):

QgsStyleV2ManagerDialog(self.style).exportItems()

def expCSV(self,save=False):

if not save:

nameInput = QInputDialog.getText(QWidget(), u"Название таблицы", u"Введите название таблицы")

if nameInput[1]:

name = nameInput[0]

else:

return

if save:

name = self.dlg.tabComboBox.currentText().replace('.csv','')

with open(self.plugin_dir + '/tab/' + name + '.csv', 'wb') as fout:

writer = UnicodeWriter(fout, delimiter=';')

for r in range(self.model.rowCount()):

itemCode = self.model.item(r, 0).text()

itemName = self.model.item(r, 1).text()

itemSymb = self.model.item(r, 2).text()

writer.writerow([itemCode, itemName, itemSymb])

def removeCodeItem(self):

indexList = []

for r in self.editRend.listWgt.selectionModel().selectedRows():

row = r.row()

indexList.append(row)

for i in indexList[::-1]:

self.model.removeRows(i, 1)

def addCodeItem(self):

itemCode = QStandardItem('None')

itemName = QStandardItem('None')

itemSymb = QStandardItem('None')

itemList =[itemCode,itemName,itemSymb]

selection = self.editRend.listWgt.selectionModel().selectedRows()

if selection:

self.model.insertRow(selection[-1::][0].row()+1,itemList)

else:

self.model.appendRow(itemList)