mirror of
https://github.com/ARUP-CAS/aiscr-qgis-amcr-viewer.git
synced 2026-06-17 11:22:53 +02:00
126 lines
4.4 KiB
Python
126 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
from qgis.PyQt.QtCore import QSettings, QTranslator, QCoreApplication
|
|
from qgis.PyQt.QtGui import QIcon
|
|
from qgis.PyQt.QtWidgets import QMenu, QAction, QToolButton
|
|
|
|
from .amcr_tools import load_amcr_data
|
|
from .amcr_dialog import AmcrFilterDialog
|
|
from .resources import *
|
|
import os.path
|
|
|
|
class AmcrViewer:
|
|
|
|
def __init__(self, iface):
|
|
self.iface = iface
|
|
self.plugin_dir = os.path.dirname(__file__)
|
|
locale = QSettings().value('locale/userLocale')[0:2]
|
|
locale_path = os.path.join(
|
|
self.plugin_dir,
|
|
'i18n',
|
|
'AmcrViewer_{}.qm'.format(locale))
|
|
|
|
if os.path.exists(locale_path):
|
|
self.translator = QTranslator()
|
|
self.translator.load(locale_path)
|
|
QCoreApplication.installTranslator(self.translator)
|
|
|
|
self.actions = []
|
|
self.menu = self.tr(u'&AMČR Viewer')
|
|
self.first_start = None
|
|
|
|
def tr(self, message):
|
|
return QCoreApplication.translate('AmcrViewer', 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):
|
|
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.iface.addToolBarIcon(action)
|
|
|
|
if add_to_menu:
|
|
self.iface.addPluginToMenu(self.menu, action)
|
|
|
|
self.actions.append(action)
|
|
return action
|
|
|
|
def initGui(self):
|
|
|
|
import os
|
|
plugin_dir = os.path.dirname(__file__)
|
|
icon_akce_path = os.path.join(plugin_dir, 'akce.png')
|
|
icon_lokality_path = os.path.join(plugin_dir, 'lokality.png')
|
|
|
|
# 1. Vytvoření společného menu
|
|
self.plugin_menu = QMenu()
|
|
|
|
# 2. Vytvoření akcí (bez automatického přidání do lišty a menu)
|
|
self.action_download_akce = self.add_action(
|
|
icon_path=icon_akce_path,
|
|
text=self.tr(u'Stáhnout data akcí | AMČR Viewer'),
|
|
callback=lambda checked=False: self.run_download('akce'),
|
|
parent=self.iface.mainWindow(),
|
|
add_to_menu=False,
|
|
add_to_toolbar=False
|
|
)
|
|
self.plugin_menu.addAction(self.action_download_akce)
|
|
|
|
self.action_download_lokality = self.add_action(
|
|
icon_path=icon_lokality_path,
|
|
text=self.tr(u'Stáhnout data lokalit | AMČR Viewer'),
|
|
callback=lambda checked=False: self.run_download('lokalita'),
|
|
parent=self.iface.mainWindow(),
|
|
add_to_menu=False,
|
|
add_to_toolbar=False
|
|
)
|
|
self.plugin_menu.addAction(self.action_download_lokality)
|
|
|
|
# 3. Přidání rozbalovacího menu do hlavního menu QGIS
|
|
main_icon = QIcon(icon_akce_path)
|
|
self.main_action = QAction(main_icon, 'AMČR Viewer', self.iface.mainWindow())
|
|
self.main_action.setMenu(self.plugin_menu)
|
|
self.iface.addPluginToMenu(self.menu, self.main_action)
|
|
|
|
# 4. Přidání rozevíracího tlačítka do nástrojové lišty (Toolbar)
|
|
self.tool_button = QToolButton()
|
|
self.tool_button.setMenu(self.plugin_menu)
|
|
self.tool_button.setDefaultAction(self.action_download_akce)
|
|
self.tool_button.setPopupMode(QToolButton.MenuButtonPopup)
|
|
|
|
# Vložení vytvořeného tlačítka do QGIS rozhraní
|
|
self.iface.addToolBarWidget(self.tool_button)
|
|
|
|
self.first_start = True
|
|
|
|
def unload(self):
|
|
for action in self.actions:
|
|
self.iface.removePluginMenu(self.tr(u'&AMČR Viewer'), action)
|
|
self.iface.removeToolBarIcon(action)
|
|
|
|
if hasattr(self, 'tool'):
|
|
self.iface.mapCanvas().unsetMapTool(self.tool)
|
|
|
|
# --- Data downloading ---
|
|
def run_download(self, typ_dat):
|
|
|
|
dlg = AmcrFilterDialog(typ_dat)
|
|
result = dlg.exec_()
|
|
|
|
if result == 1:
|
|
filters = dlg.get_filters()
|
|
bbox = dlg.get_bbox()
|
|
komponenty = dlg.get_komponenty()
|
|
|
|
canvas = self.iface.mapCanvas()
|
|
load_amcr_data(canvas, bbox, filters, typ_dat, komponenty)
|