spatial_filter.py

"""
/***************************************************************************
 SpatialFilter
                                 A QGIS plugin
 This plugin applies spatial filters to layers for performance and efficiency.

 Base was generated by Plugin Builder: http://g-sherman.github.io/Qgis-Plugin-Builder/
                              -------------------
        begin                : 2022-10-16
        copyright            : (C) 2022 Wheregroup GmbH
        email                : info@wheregroup.com
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   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 qgis.PyQt.QtCore import QLocale, QSettings, QTranslator, QCoreApplication

import os.path


class SpatialFilter:

    def __init__(self, iface):
        self.iface = iface
        self.plugin_dir = os.path.dirname(__file__)
        locale = QSettings().value('locale/userLocale', QLocale().name())[0:2]
        locale_path = os.path.join(self.plugin_dir, 'i18n', f'spatial_filter_{locale}.qm')

        if os.path.exists(locale_path):
            self.translator = QTranslator()
            self.translator.load(locale_path)
            QCoreApplication.installTranslator(self.translator)

    def initGui(self):
        # we import just now because we need the QCoreApplication translator
        # to be installed already (e.g. for the LOCALIZED_PLUGIN_NAME)
        from .controller import FilterController
        from .helpers import warnAboutQgisBugProjectSaving
        from .widgets import FilterToolbar

        self.toolbar = FilterToolbar(FilterController(), self.iface.mainWindow())
        self.iface.mainWindow().addToolBar(self.toolbar)

        warnAboutQgisBugProjectSaving()


    def unload(self):
        self.toolbar.hideFilterGeom()
        self.toolbar.controller.removeFilter()
        self.toolbar.controller.disconnectSignals()
        self.toolbar.deleteLater()