newmemorylayer.py

# -*- coding: utf-8 -*-
"""
/***************************************************************************
 NewMemoryLayer
                                 A QGIS plugin
 Creates a memory layer
                              -------------------
        begin                : 2011-05-14
        copyright            : by Borys Jurgiel
        email                : info at borysjurgiel dot pl
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   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.                                   *
 *                                                                         *
 ***************************************************************************/
"""

import os
from qgis.PyQt.QtCore import QCoreApplication, QFileInfo, QLocale, QTranslator
from qgis.PyQt.QtGui import QAction, QIcon
from .newmemorylayerdialog import NewMemoryLayerDialog


class NewMemoryLayer:

    def __init__(self, iface):
        self.iface = iface
        self.action = None
        self.dlg = None
        # i18n
        pluginPath = QFileInfo(os.path.realpath(__file__)).path()
        localeName = QLocale.system().name()
        if QFileInfo(pluginPath).exists():
            self.localePath = pluginPath + "/i18n/newmemorylayer_" + localeName + ".qm"
        if QFileInfo(self.localePath).exists():
            self.translator = QTranslator()
            self.translator.load(self.localePath)
            QCoreApplication.installTranslator(self.translator)

    def initGui(self):
        self.action = QAction(
            QIcon(os.path.join(os.path.dirname(__file__), "layer-memory-create.png")),
            QCoreApplication.translate("NewMemoryLayer", "New Memory Layer..."),
            self.iface.mainWindow(),
        )
        self.action.setShortcut("Ctrl+W")
        self.iface.registerMainWindowAction(self.action, "Ctrl+W")
        self.action.triggered.connect(self.run)
        self.iface.newLayerMenu().addAction(self.action)
        self.iface.layerToolBar().addAction(self.action)

    def unload(self):
        if self.action:
            self.action.triggered.disconnect(self.run)
            self.iface.unregisterMainWindowAction(self.action)
            self.iface.newLayerMenu().removeAction(self.action)
            self.iface.layerToolBar().removeAction(self.action)

    def run(self):
        if not self.dlg:
            self.dlg = NewMemoryLayerDialog()
        self.dlg.show()
        self.dlg.activateWindow()