Le portail francophone de la géomatique

# To change this license header, choose License Headers in Project Properties.
# To change this template file, choose Tools | Templates
# and open the template in the editor.

__author__ = "guyon_000"
__date__ = "$16 mars 2015 15:18:04$"

# -*- coding: utf-8 -*-
"""
/***************************************************************************
 Class_remi
                                 A QGIS plugin
 Description_remi
                              -------------------
        begin                : 2015-03-05
        git sha              : $Format:%H$
        copyright            : (C) 2015 by author_remi
        email                : igor.belot@orange.fr
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   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 the PyQt and QGIS libraries
from PyQt4.QtCore import *
from PyQt4.QtGui import *
from qgis.core import *
from qgis.gui import *
from qgis.utils import *
# Initialize Qt resources from file resources.py
import resources_rc
# Import the code for the dialog
from Module_name_remi_dialog import Class_remiDialog
import os.path


class Class_remi:
    """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','Class_remi_{}.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)

        # Create the dialog (after translation) and keep reference
        self.dlg = Class_remiDialog(self.iface)

    def initGui(self):
        
        # Create action that will start plugin configuration
        self.action = QAction(QIcon(":/plugins/Class_remi/icon.png"), "Class_remi", self.iface.mainWindow())
        
        # connect the action to the run method
        self.action.triggered.connect(self.run)
         
        # Add toolbar button and menu item
        self.iface.addToolBarIcon(self.action)
        self.iface.addPluginToVectorMenu("&Class_remi", self.action)
        

    def unload(self):
        # Remove the plugin menu item and icon
        self.iface.removePluginVectorMenu("&Class_remi", self.action)
        self.iface.removeToolBarIcon(self.action)

    def run(self):
        """Run method that performs all the real work"""
        # show the dialog
        self.dlg.init_before_show()
        self.dlg.show()
        # Run the dialog event loop
        result = self.dlg.exec_()
        # See if OK was pressed
        if result == 1:
            # Do something useful here - delete the line containing pass and
            # substitute with your code.
            self.load_line()
            self.load_arrets()
            self.test()
            
       
    def load_line(self):
        uri = QgsDataSourceURI()
        # set host name, port, database name, username and password
        uri.setConnection("localhost", "5432", "postgres", "postgres", "210893")
        # set database schema, table name, geometry column and optionally
        # subset (WHERE clause)
        uri.setDataSource("public", "ligne", "geom")
        vlayer1 = QgsVectorLayer(uri.uri(), "ligne_tan", "postgres")
    
    
    def load_arrets(self):
        uri = QgsDataSourceURI()
        # set host name, port, database name, username and password
        uri.setConnection("localhost", "5432", "postgres", "postgres", "210893")
        # set database schema, table name, geometry column and optionally
        # subset (WHERE clause)
        uri.setDataSource("public", "arrettan", "geom")
        vlayer2 = QgsVectorLayer(uri.uri(), "arrets_tan", "postgres")
        
    def test(self):
        ligne_tan = self.dlg.ui.Ligne_choix.currentText()
        arrets_tan = self.dlg.ui.arrets_choix.currentText()

NameError: global name 'ui_Class_remi' is not defined

# To change this license header, choose License Headers in Project Properties.
# To change this template file, choose Tools | Templates
# and open the template in the editor.

__author__ = "guyon_000"
__date__ = "$16 mars 2015 15:18:04$"

# -*- coding: utf-8 -*-
"""
/***************************************************************************
 Class_remiDialog
                                 A QGIS plugin
 Description_remi
                             -------------------
        begin                : 2015-03-05
        git sha              : $Format:%H$
        copyright            : (C) 2015 by author_remi
        email                : igor.belot@orange.fr
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   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 PyQt4 import QtGui, uic

FORM_CLASS, _ = uic.loadUiType(os.path.join(
    os.path.dirname(__file__), 'Module_name_remi_dialog_base.ui'))


class Class_remiDialog(QtGui.QDialog, FORM_CLASS):
    def __init__(self, parent=None):
        """Constructor."""
        super(Class_remiDialog, self).__init__(parent)
        self.ui = ui_Class_remi()
        self.setupUi(self)
        self.iface = iface
        self.settings = QSettings()
        self.mc = self.iface.mapCanvas()
        self.legend = self.iface.legendInterface()
        self.loaded_layers = self.legend.layers()
        # selected layer
        self.vlayer = self.mc.currentLayer()
        # Clear form
        self.ui.Ligne_choix.clear()
        # UI CCONNECTORS
        self.ui.button_box.accepted.connect(self.run)
        self.ui.button_box.rejected.connect(self.close)
         #self.ui.comboBox_inputlayer.currentIndexChanged.connect(self.update_fields)
        self.ui.Ligne_choix.activated.connect(self.update_fields)
        self.ui.arrets_choix.activated.connect(self.update_fiels)
        
       
       

    def init_before_show(self):
        # Get layers from legend
        self.mc = self.iface.mapCanvas()
        self.legend = self.iface.legendInterface()
        self.loaded_layers = self.legend.layers()
        self.ui.Ligne_choix.clear()
        
        # Populate comboboxes and fields
        list_of_vlayers = self.get_useful_layers()
        if len(list_of_vlayers) > 0:
            self.ui.Ligne_choix.addItems(list_of_vlayers)
            self.update_working_layer()
            self.update_fields()
        else:
            self.iface.messageBar().pushMessage("Error", "Plugin needs a valid vector layer loaded.", level=QgsMessageBar.CRITICAL, duration=5)

    def update_working_layer(self):
        for layer in self.loaded_layers:
            if layer.name() == self.ui.Ligne_choix.currentText():
                self.vlayer = layer
                break
                
    def get_useful_layers(self):
      self.vector_line_layers = {}
      for layer in self.loaded_layers:
            fields_names = []
            # select line vector layers
            if (layer.type() == layer.VectorLayer) and (layer.geometryType() == QGis.Line):
                layer_info = [layer]
                provider = layer.dataProvider()
                fields = provider.fields()
                # get vector layer fields
                for field in fields:
                    if (field.type() == QVariant.Int or field.type() == QVariant.Double):
                        fields_names.append(field.name())
                layer_info += [fields_names]
                self.vector_line_layers[str(layer.name())] = layer_info
            else:
                pass
            vector_line_layers = list(self.vector_line_layers)
            return vector_line_layers

    def update_fields(self):
        self.update_working_layer()

    def run(self):
        pass
        return

# To change this license header, choose License Headers in Project Properties.
# To change this template file, choose Tools | Templates
# and open the template in the editor.

__author__ = "guyon_000"
__date__ = "$16 mars 2015 15:18:04$"

# -*- coding: utf-8 -*-
"""
/***************************************************************************
 Class_remiDialog
                                 A QGIS plugin
 Description_remi
                             -------------------
        begin                : 2015-03-05
        git sha              : $Format:%H$
        copyright            : (C) 2015 by author_remi
        email                : igor.belot@orange.fr
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   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 PyQt4 import QtGui, uic
from PyQt4 import QtCore
from PyQt4.QtCore import *
from PyQt4.QtGui import *
from qgis.core import *
from qgis.gui import *
from qgis.utils import *

FORM_CLASS, _ = uic.loadUiType(os.path.join(
    os.path.dirname(__file__), 'Module_name_remi_dialog_base.ui'))


class Class_remiDialog(QtGui.QDialog, FORM_CLASS):
    def __init__(self, iface, parent=None):
        """Constructor."""
        super(Class_remiDialog, self).__init__(parent)
        self.setupUi(self)
        self.iface = iface
        self.settings = QSettings()
        self.mc = self.iface.mapCanvas()
        self.legend = self.iface.legendInterface()
        self.loaded_layers = self.legend.layers()
        # selected layer
        self.vlayer = self.mc.currentLayer()
        # Clear form
        self.Ligne_choix.clear()
        # UI CCONNECTORS
        self.button_box.accepted.connect(self.run)
        self.button_box.rejected.connect(self.close)
        
        self.Ligne_choix.activated.connect(self.update_fields)
        self.arrets_choix.activated.connect(self.update_fiels)
        
       
       

    def init_before_show(self):
        # Get layers from legend
        self.mc = self.iface.mapCanvas()
        self.legend = self.iface.legendInterface()
        self.loaded_layers = self.legend.layers()
        self.Ligne_choix.clear()
        
        # Populate comboboxes and fields
        list_of_vlayers = self.get_useful_layers()
        if len(list_of_vlayers) > 0:
            self.Ligne_choix.addItems(list_of_vlayers)
            self.update_working_layer()
            self.update_fields()
        else:
            self.iface.messageBar().pushMessage("Error", "Plugin needs a valid vector layer loaded.", level=QgsMessageBar.CRITICAL, duration=5)

    def update_working_layer(self):
        for layer in self.loaded_layers:
            if layer.name() == self.Ligne_choix.currentText():
                self.vlayer = layer
                break
                
    def get_useful_layers(self):
      self.vector_line_layers = {}
      for layer in self.loaded_layers:
            fields_names = []
            # select line vector layers
            if (layer.type() == layer.VectorLayer) and (layer.geometryType() == QGis.Line):
                layer_info = [layer]
                provider = layer.dataProvider()
                fields = provider.fields()
                # get vector layer fields
                for field in fields:
                    if (field.type() == QVariant.Int or field.type() == QVariant.Double):
                        fields_names.append(field.name())
                layer_info += [fields_names]
                self.vector_line_layers[str(layer.name())] = layer_info
            else:
                pass
            vector_line_layers = list(self.vector_line_layers)
            return vector_line_layers

    def update_fields(self):
        self.update_working_layer()

    def run(self):
        pass
        return

AttributeError: type object 'QDialog' has no attribute 'mapCanvas'