def __init__(self, iface):
"""Constructor."""
QtGui.QDialog.__init__(self)
self.ui = Ui_bdhabnat_dialog()
self.ui.setupUi(self)
# Référencement de iface dans l'interface (iface = interface de QGIS)
self.iface = iface
self.canvas = self.iface.mapCanvas()
# Connexion à la base de données. DB type, host, user, password...
self.db = QtSql.QSqlDatabase.addDatabase("QPSQL") # QPSQL = nom du pilote postgreSQL
self.db.setHostName("192.168.0.10")
self.db.setPort(5432)
self.db.setDatabaseName("sitescsn")
self.db.setUserName("postgres")
self.db.setPassword("postgres")
ok = self.db.open()
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'La connexion est échouée'+self.db.hostName())
# Remplir la combobox "cbx_codesite" avec les codes et noms de sites issus de la table "t_sitescen"
query_codesite = QtSql.QSqlQuery(self.db)
if query_codesite.exec_('select idchamp, codesite, nomsite from sites_cen.t_sitescen order by codesite'):
while query_codesite.next():
self.ui.cbx_codesite.addItem(query_codesite.value(1) + " " + query_codesite.value(2), query_codesite.value(1) )
# Remplir la combobox "cbx_auteurs" avec les noms et prénoms issus de la table "t_liste_obsv"
query_obsv = QtSql.QSqlQuery(self.db)
if query_obsv.exec_('select id_obsv, nom_obsv, prenom_obsv, initiale_obsv from bd_habnat.t_liste_obsv order by nom_obsv, prenom_obsv'):
while query_obsv.next():
self.ui.cbx_auteur.addItem(query_obsv.value(1) + " " + query_obsv.value(2), query_obsv.value(3) )
# Combobox cbx_annee : Affecter l'année en cours par défaut en fonction de la date (changement d'année au 28/2/N)
if datetime.now().strftime('%m/%d') < '04/02' :
self.ui.cbx_annee.setCurrentIndex(self.ui.cbx_annee.findText((str(int(datetime.now().strftime('%Y'))-1)), QtCore.Qt.MatchStartsWith))
else:
self.ui.cbx_annee.setCurrentIndex(self.ui.cbx_annee.findText((datetime.now().strftime('%Y')), QtCore.Qt.MatchStartsWith))
# Remplir la combobox "cbx_habref" avec les noms de référentiels issus de la table "t_liste_ref"
query_ref = QtSql.QSqlQuery(self.db)
if query_ref.exec_('select id_ref, nom_ref, code_ref from bd_habnat.t_liste_ref order by id_ref'):
while query_ref.next():
self.ui.cbx_habref.addItem(query_ref.value(1), query_ref.value(2) )
self.ui.cbx_habref.model().item(0).setEnabled(False)
# Remplir la combobox "cbx_eur27_mep" avec le champs hab_cod de la table "t_liste_ref_eur27"
query_eur27 = QtSql.QSqlQuery(self.db)
if query_eur27.exec_('select hab_cod from bd_habnat.t_liste_ref_eur27 order by hab_cod'):
while query_eur27.next():
self.ui.cbx_eur27_mep.addItem(query_eur27.value(0), query_eur27.value(0) )
# Désactiver le bouton "OK" tant qu'on n'a pas choisi au moins un référentiel.
self.ui.buttonBox.setEnabled(False)
# Connexions signaux - slots
self.ui.cbx_habref.currentIndexChanged.connect(self.listesref)
self.ui.cbx_habfr.currentIndexChanged.connect(self.coreur27)
self.ui.buttonBox.accepted.connect(self.sauvSaisie)
self.ui.buttonBox.rejected.connect(self.close)
self.ui.chx_plantation.stateChanged.connect(self.plantation)
self.ui.cbx_hablat.currentIndexChanged.connect(self.listefran)
self.ui.cbx_habref.currentIndexChanged.connect(self.activButton)
def __init__(self, iface):
"""Constructor."""
QtGui.QDialog.__init__(self)
self.ui = Ui_bdhabnat_dialog()
self.ui.setupUi(self)
# Référencement de iface dans l'interface (iface = interface de QGIS)
self.iface = iface
self.canvas = self.iface.mapCanvas()
# Connexion à la base de données. DB type, host, user, password...
self.db = QtSql.QSqlDatabase.addDatabase("QPSQL") # QPSQL = nom du pilote postgreSQL
self.db.setHostName("192.168.0.10")
self.db.setDatabaseName("sitescsn")
self.db.setUserName("postgres")
self.db.setPassword("postgres")
ok = self.db.open()
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'La connexion est échouée'+self.db.hostName())
# Remplir la combobox "cbx_codesite" avec les codes et noms de sites issus de la table "t_sitescen"
query_codesite = QtSql.QSqlQuery(self.db)
if query_codesite.exec_('select idchamp, codesite, nomsite from sites_cen.t_sitescen order by codesite'):
while query_codesite.next():
self.ui.cbx_codesite.addItem(query_codesite.value(1) + " " + query_codesite.value(2), query_codesite.value(1) )
# Remplir la combobox "cbx_auteurs" avec les noms et prénoms issus de la table "t_liste_obsv"
query_obsv = QtSql.QSqlQuery(self.db)
if query_obsv.exec_('select id_obsv, nom_obsv, prenom_obsv, initiale_obsv from bd_habnat.t_liste_obsv order by nom_obsv, prenom_obsv'):
while query_obsv.next():
self.ui.cbx_auteur.addItem(query_obsv.value(1) + " " + query_obsv.value(2), query_obsv.value(3) )
# Combobox cbx_annee : Affecter l'année en cours par défaut en fonction de la date (changement d'année au 28/2/N)
if datetime.now().strftime('%m/%d') < '04/02' :
self.ui.cbx_annee.setCurrentIndex(self.ui.cbx_annee.findText((str(int(datetime.now().strftime('%Y'))-1)), QtCore.Qt.MatchStartsWith))
else:
self.ui.cbx_annee.setCurrentIndex(self.ui.cbx_annee.findText((datetime.now().strftime('%Y')), QtCore.Qt.MatchStartsWith))
# Remplir la combobox "cbx_habref" avec les noms de référentiels issus de la table "t_liste_ref"
query_ref = QtSql.QSqlQuery(self.db)
if query_ref.exec_('select id_ref, nom_ref, code_ref from bd_habnat.t_liste_ref order by id_ref'):
while query_ref.next():
self.ui.cbx_habref.addItem(query_ref.value(1), query_ref.value(2) )
self.ui.cbx_habref.model().item(0).setEnabled(False)
# Remplir la combobox "cbx_eur27_mep" avec le champs hab_cod de la table "t_liste_ref_eur27"
query_eur27 = QtSql.QSqlQuery(self.db)
if query_eur27.exec_('select hab_cod from bd_habnat.t_liste_ref_eur27 order by hab_cod'):
while query_eur27.next():
self.ui.cbx_eur27_mep.addItem(query_eur27.value(0), query_eur27.value(0) )
# Désactiver le bouton "OK" tant qu'on n'a pas choisi au moins un référentiel.
self.ui.buttonBox.setEnabled(False)
# Connexions signaux - slots
self.ui.cbx_habref.currentIndexChanged.connect(self.listesref)
self.ui.cbx_habfr.currentIndexChanged.connect(self.coreur27)
self.ui.buttonBox.accepted.connect(self.sauvSaisie)
self.ui.buttonBox.rejected.connect(self.close)
self.ui.chx_plantation.stateChanged.connect(self.plantation)
self.ui.cbx_hablat.currentIndexChanged.connect(self.listefran)
self.ui.cbx_habref.currentIndexChanged.connect(self.activButton)
class bdhabnatDialog(QtGui.QDialog):
def __init__(self, iface):
"""Constructor."""
QtGui.QDialog.__init__(self)
self.ui = Ui_bdhabnat_dialog()
self.ui.setupUi(self)
# Référencement de iface dans l'interface (iface = interface de QGIS)
self.iface = iface
self.canvas = self.iface.mapCanvas()
# Connexion à la base de données. DB type, host, user, password...
self.db = QtSql.QSqlDatabase.addDatabase("QPSQL") # QPSQL = nom du pilote postgreSQL
self.db.setHostName("192.168.0.10")
self.db.setPort(5432)
self.db.setDatabaseName("sitescsn")
self.db.setUserName("postgres")
self.db.setPassword("postgres")
ok = self.db.open()
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'La connexion est échouée'+self.db.hostName())
# Remplir la combobox "cbx_codesite" avec les codes et noms de sites issus de la table "t_sitescen"
query_codesite = QtSql.QSqlQuery(self.db)
if query_codesite.exec_('select idchamp, codesite, nomsite from sites_cen.t_sitescen order by codesite'):
while query_codesite.next():
self.ui.cbx_codesite.addItem(query_codesite.value(1) + " " + query_codesite.value(2), query_codesite.value(1) )
# Remplir la combobox "cbx_auteurs" avec les noms et prénoms issus de la table "t_liste_obsv"
query_obsv = QtSql.QSqlQuery(self.db)
if query_obsv.exec_('select id_obsv, nom_obsv, prenom_obsv, initiale_obsv from bd_habnat.t_liste_obsv order by nom_obsv, prenom_obsv'):
while query_obsv.next():
self.ui.cbx_auteur.addItem(query_obsv.value(1) + " " + query_obsv.value(2), query_obsv.value(3) )
# Combobox cbx_annee : Affecter l'année en cours par défaut en fonction de la date (changement d'année au 28/2/N)
if datetime.now().strftime('%m/%d') < '04/02' :
self.ui.cbx_annee.setCurrentIndex(self.ui.cbx_annee.findText((str(int(datetime.now().strftime('%Y'))-1)), QtCore.Qt.MatchStartsWith))
else:
self.ui.cbx_annee.setCurrentIndex(self.ui.cbx_annee.findText((datetime.now().strftime('%Y')), QtCore.Qt.MatchStartsWith))
# Remplir la combobox "cbx_habref" avec les noms de référentiels issus de la table "t_liste_ref"
query_ref = QtSql.QSqlQuery(self.db)
if query_ref.exec_('select id_ref, nom_ref, code_ref from bd_habnat.t_liste_ref order by id_ref'):
while query_ref.next():
self.ui.cbx_habref.addItem(query_ref.value(1), query_ref.value(2) )
self.ui.cbx_habref.model().item(0).setEnabled(False)
# Remplir la combobox "cbx_eur27_mep" avec le champs hab_cod de la table "t_liste_ref_eur27"
query_eur27 = QtSql.QSqlQuery(self.db)
if query_eur27.exec_('select hab_cod from bd_habnat.t_liste_ref_eur27 order by hab_cod'):
while query_eur27.next():
self.ui.cbx_eur27_mep.addItem(query_eur27.value(0), query_eur27.value(0) )
# Désactiver le bouton "OK" tant qu'on n'a pas choisi au moins un référentiel.
self.ui.buttonBox.setEnabled(False)
# Connexions signaux - slots
self.ui.cbx_habref.currentIndexChanged.connect(self.listesref)
self.ui.cbx_habfr.currentIndexChanged.connect(self.coreur27)
self.ui.buttonBox.accepted.connect(self.sauvSaisie)
self.ui.buttonBox.rejected.connect(self.close)
self.ui.chx_plantation.stateChanged.connect(self.plantation)
self.ui.cbx_hablat.currentIndexChanged.connect(self.listefran)
self.ui.cbx_habref.currentIndexChanged.connect(self.activButton)
def listesref(self):
# Réinitialisations.
# Attention de bien récupérer la nouvelle valeur de habref avant de vider hablat, pour ne pas lancer inopinément la fonction listefran.
self.habref = self.ui.cbx_habref.itemData(self.ui.cbx_habref.currentIndex())
self.ui.cbx_hablat.clear()
self.ui.cbx_habfr.clear()
self.ui.txt_codeeur27.clear()
self.ui.cbx_eur27_mep.setEnabled(True)
#print 'listesref'
if self.habref == 'cbnbl':
# Si le référentiel "Digitale" du CBNBL est sélectionné, alors remplir la combobox "cbx_hablat"
query_cbnbl = QtSql.QSqlQuery(self.db)
qcbnbl = u'select hab_cod, hab_lat from (select * from bd_habnat.t_liste_ref_cbnbl_v12 UNION select * from bd_habnat.t_liste_ref_newlat_cbnbl) ref order by hab_lat'
ok = query_cbnbl.exec_(qcbnbl)
while query_cbnbl.next():
self.ui.cbx_hablat.addItem(query_cbnbl.value(1), query_cbnbl.value(0))
#print str(qcbnbl)
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête CBNBL ratée')
#print qcbnbl
self.listefran()
else :
#Si un autre référentiel est sélectionné, alors remplir la combobox "cbx_habfr" avec les valeurs de la table correspondant au référentiel.
query_autrref = QtSql.QSqlQuery(self.db)
qautrref = u"""select hab_cod, hab_fr from bd_habnat.t_liste_ref_{zr_table} order by hab_fr""".format(zr_table = str(self.habref))
ok = query_autrref.exec_(qautrref)
#print 'autre'
while query_autrref.next():
self.ui.cbx_habfr.addItem(query_autrref.value(1), query_autrref.value(0))
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête Autre ref ratée')
#print unicode(qautrref)
if self.habref == 'eur27':
self.ui.cbx_eur27_mep.setEnabled(False)
def listefran(self):
#Quand le référentiel de Bailleul ou un taxon scientifique est sélectionné, afficher le taxon en français s'il est disponible
self.ui.cbx_habfr.clear()
if self.habref == 'cbnbl':
query_fr = QtSql.QSqlQuery(self.db)
qfr = u"""select hab_cod, hab_fr from bd_habnat.t_liste_ref_cbnbl_v12 cbnbl where cbnbl.hab_cod = {zr_codlat} order by hab_fr""".format (zr_codlat = self.ui.cbx_hablat.itemData(self.ui.cbx_hablat.currentIndex()))
ok = query_fr.exec_(qfr)
#print 'listefran'
if not ok :
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête Hab Fr ratée')
while query_fr.next():
self.ui.cbx_habfr.addItem(query_fr.value(1), query_fr.value(0) )
def activButton(self):
# Activation du bouton "OK et annuler" si un habitat a été sélectionné
cbxhabrefindex = self.ui.cbx_habref.currentIndex()
if cbxhabrefindex!=0:
self.ui.buttonBox.setEnabled(1)
def coreur27(self):
# Remplissage éventuel de la zone de texte "txt_codeeur27"
self.ui.txt_codeeur27.clear()
self.codecoreu27 = self.ui.cbx_habfr.itemData(self.ui.cbx_habfr.currentIndex())
if self.habref == 'eur27':
self.ui.txt_codeeur27.setText(unicode(self.codecoreu27))
def plantation(self):
# Si l'habitat saisi est une plantation, il ne peut pas faire partie d'une mosaïque. => il prend 100 % de la surface.
# cbx_pourcent est désactivé est sa valeur fixée à 100.
if self.ui.chx_plantation.isChecked() == True:
self.ui.cbx_pourcent.setEnabled(0)
self.ui.cbx_pourcent.setCurrentIndex(self.ui.cbx_pourcent.findText('100', QtCore.Qt.MatchStartsWith))
else :
self.ui.cbx_pourcent.setEnabled(1)
def sauvSaisie(self):
if self.ui.cbx_habref.itemData(self.ui.cbx_habref.currentIndex())=='0':
return
self.erreurSaisieBase = '0'
#copie des entités sélectionnées dans une couche "memory". Evite les problèmes avec les types de couches "non éditables" (comme les GPX).
coucheactive=self.iface.activeLayer()
# On vérifie que les entités sélectionnées dans la couche active sont bien des polygones (et non des lignes ou des points)
entselect=[QgsGeometry(feature.geometry()) for feature in coucheactive.selectedFeatures()]
if entselect[0].type() == QGis.Polygon:
typegeom='Polygon'
# geom = self.transfoPoly(entselect)
# print "geom=" + str(geom.exportToWkt())
else:
QtGui.QMessageBox.warning(self, 'Alerte', u'Les entités sélectionnées ne sont pas des polygones')
self.close
return
# Création de la couche en mémoire "memlayer" et début de la session d'édition
if coucheactive.crs().authid() == u'EPSG:4326':
memlayer=QgsVectorLayer("{zr_typegeom}?crs=epsg:4326".format(zr_typegeom = typegeom), "memlayer", "memory")
if coucheactive.crs().authid() == u'EPSG:2154':
memlayer=QgsVectorLayer("{zr_typegeom}?crs=epsg:2154".format(zr_typegeom = typegeom), "memlayer", "memory")
else :
QtGui.QMessageBox.warning(self, 'Alerte', u'Les seules projections supportées sont le WGS84 et le Lambert93. Merci de vérifier la projection de votre couche SIG')
return
QgsMapLayerRegistry.instance().addMapLayer(memlayer, False)
root = QgsProject.instance().layerTreeRoot()
memlayerNode = QgsLayerTreeLayer(memlayer)
root.insertChildNode(0, memlayerNode)
self.iface.setActiveLayer(memlayer)
memlayer.startEditing()
# Pour chaque entité sélectionnée, si elle est multipartie, on ajoute chacune de ses parties individuellement à la couche memlayer.
# Sinon, on l'ajoute directement à "memlayer". Puis, on clot la session d'édition.
for feature in coucheactive.selectedFeatures() :
geom = feature.geometry()
temp_feature = QgsFeature(feature)
# check if feature geometry is multipart
if geom.isMultipart():
# if feature is multipart creates a new feature using the geometry of each part
for part in geom.asGeometryCollection ():
temp_feature.setGeometry(part)
memlayer.dataProvider().addFeatures([temp_feature])
memlayer.updateExtents()
# if feature is singlepart, simply adds it to the layer memory
else :
temp_feature.setGeometry(geom)
memlayer.dataProvider().addFeatures([temp_feature])
memlayer.updateExtents()
memlayer.commitChanges()
#print "memlayercount="+str(memlayer.featureCount())
#on sélectionne toutes les entités de memlayer pour en faire une liste de géométries, qui sera saisie en base.
memlayer.selectAll()
geomlis = [QgsGeometry(feature.geometry()) for feature in memlayer.selectedFeatures()]
#geomlist = QgsGeometry.fromMultiPolygon([poly.asPolygon() for poly in geomlis]) --- transformation en multipolygone
for entite in geomlis :
#export de la géométrie en WKT et transformation de la projection si les données ne sont pas saisies en Lambert 93
if memlayer.crs().authid() == u'EPSG:2154':
thegeom='st_setsrid(st_geometryfromtext (\'{zr_geom}\'), 2154)'.format(zr_geom=entite.exportToWkt())
elif memlayer.crs().authid() == u'EPSG:4326':
thegeom='st_transform(st_setsrid(st_geometryfromtext (\'{zr_geom}\'),4326), 2154)'.format(zr_geom=entite.exportToWkt())
else :
print u'La projection de la couche active n\'est pas supportée'
#gestion de l'identifiant id_mosaik, servant à regrouper les enregistrements appartenant à une même mosaïque d'habitats
pourcent = int(self.ui.cbx_pourcent.itemText(self.ui.cbx_pourcent.currentIndex()))
if self.ui.chx_plantation.isChecked() == True : #cas de la plantation
id_mosaik = 0
print 'zorglub'
else :
print 'pas le bon'
querymosaik = QtSql.QSqlQuery(self.db)
qmosaik = u"""SELECT id_hab_ce, annee, pourcent,the_geom, plantation, id_mosaik FROM bd_habnat.t_ce_saisie WHERE the_geom = {zr_thegeom} AND annee = '{zr_annee}' AND plantation = 'f' """.format (\
zr_thegeom = thegeom,\
zr_annee = self.ui.cbx_annee.itemText(self.ui.cbx_annee.currentIndex()))
ok = querymosaik.exec_(qmosaik)
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête Mosaik ratée')
if querymosaik.size() > 0 :
#si la req retourne une valeur => soit 1 erreur (100%), soit une mosaik
#print "on est deja dans la mosaique ou dans l'erreur"
sumprct = 0
while querymosaik.next() :
sumprct += int(querymosaik.value(2))
sumprct += pourcent
if sumprct > 100 :
QtGui.QMessageBox.warning(self, 'Alerte', u'La somme des pourcentages des habitats dépasse 100 % dans la mosaïque')
return
querymosaik.first()
id_mosaik = querymosaik.value(5)
else :
if pourcent > 99 :
#print ">99"
id_mosaik = 0
else :
querybiggestid = QtSql.QSqlQuery(self.db)
qbiggestid = u"""SELECT id_mosaik FROM bd_habnat.t_ce_saisie ORDER BY id_mosaik DESC LIMIT 1"""
ok = querybiggestid.exec_(qbiggestid)
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête PlusGrandIdMosaik ratée')
querybiggestid.next()
#print "debut de la mosaique"
id_mosaik = int(querybiggestid.value(0))+1
self.habref = self.ui.cbx_habref.itemData(self.ui.cbx_habref.currentIndex())
# si ref = cbnbl : recup rareté, menace et intérêt patri en fonction du taxon sélectionné par l'utilisateur
if self.habref == 'cbnbl':
queryrarmen = QtSql.QSqlQuery(self.db)
qrarmen = u"""SELECT rarete, menace, interetpatr FROM bd_habnat.t_liste_ref_cbnbl_v12 WHERE hab_lat = '{zr_hablat}'""".format (\
zr_hablat= self.ui.cbx_hablat.itemText(self.ui.cbx_hablat.currentIndex()).replace("\'","\'\'"))
okrarmen=queryrarmen.exec_(qrarmen)
if not okrarmen :
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête RarMen ratée')
queryrarmen.next()
self.rarete = queryrarmen.value(0)
self.menace = queryrarmen.value(1)
self.interetpatr = queryrarmen.value(2)
# construction de pat_cen en fonction de intérêt patri
querypatcen = QtSql.QSqlQuery(self.db)
qpatcen = u""" SELECT CASE WHEN cbnbl.interetpatr IN ('Oui','#') THEN True ELSE False END FROM bd_habnat.t_liste_ref_cbnbl_v12 cbnbl WHERE hab_lat = '{zr_hablat}'""".format (\
zr_hablat= self.ui.cbx_hablat.itemText(self.ui.cbx_hablat.currentIndex()).replace("\'","\'\'"))
okpatcen = querypatcen.exec_(qpatcen)
if not okpatcen :
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête PatCen ratée')
querypatcen.next()
self.pat_cen = querypatcen.value(0)
# si ref != cbnbl : rareté, menace, intérêt patri = '/' et pat_cen = False
else :
self.rarete = '/'
self.menace = '/'
self.interetpatr = '/'
self.pat_cen = 'false'
# aller chercher hab_cod dans la cbx "lat" si self.habref = cbnbl, sinon dans cbx "fr".
# aller chercher code_syntax, code_cahab, code_eunis et code_corine dans itemData cbx_hablat ou cbx_habfr, selon le référentiel choisi
if self.habref == 'cbnbl':
self.habcod = self.ui.cbx_hablat.itemData(self.ui.cbx_hablat.currentIndex())
v_syntax = self.habcod
v_cahab = ''
v_eunis =''
v_corine =''
else :
self.habcod = self.ui.cbx_habfr.itemData(self.ui.cbx_habfr.currentIndex())
if self.habref == 'eur27' :
v_syntax = ''
v_cahab = self.habcod
v_eunis = ''
v_corine = ''
elif self.habref == 'eunis' :
v_syntax = ''
v_cahab = ''
v_eunis = self.habcod
v_corine = ''
elif self.habref == 'corine' :
v_syntax = ''
v_cahab = ''
v_eunis = ''
v_corine = self.habcod
# récupération du code N2000 par txt_codeeur27 s'il est renseigné, sinon cbx_eur27_mep
if self.ui.txt_codeeur27.text() != '':
#print 'txt_codeeur27 renseigne'
self.codeeur27 = self.ui.txt_codeeur27.text()
else :
if self.ui.cbx_eur27_mep.currentIndex() == 0 :
self.codeeur27 = ''
else :
self.codeeur27 = self.ui.cbx_eur27_mep.itemText(self.ui.cbx_eur27_mep.currentIndex())
#lancement de la requête SQL qui introduit les données géographiques et du formulaire dans la base de données.
querysauvhab = QtSql.QSqlQuery(self.db)
query = u"""INSERT INTO bd_habnat.t_ce_saisie(codesite, auteur, annee, hab_ref, hab_cod, hab_lat, hab_fr, hab_comment, code_syntax, code_cahab, code_eur27, code_eunis, code_corine, pourcent, rarete, menace, patrimoine, pat_cen, surf_tot, the_geom, plantation, id_mosaik, hab_etat, faciesa, date_crea_poly) VALUES ('{zr_codesite}', '{zr_auteur}', '{zr_annee}', '{zr_habref}','{zr_habcod}', '{zr_hablat}', '{zr_habfr}', '{zr_habcomment}', '{zr_codesyntax}', '{zr_codecahab}', '{zr_codeeur27}', '{zr_codeeunis}', '{zr_codecorine}', '{zr_pourcent}', '{zr_rarete}', '{zr_menace}', '{zr_interetpatr}', {zr_pat_cen}, st_area({zr_thegeom}), {zr_thegeom}, {zr_plantation}, {zr_idmosaik}, '{zr_habetat}', '{zr_faciesa}', current_date)""".format (\
zr_codesite = self.ui.cbx_codesite.itemData(self.ui.cbx_codesite.currentIndex()),\
zr_auteur = self.ui.cbx_auteur.itemData(self.ui.cbx_auteur.currentIndex()),\
zr_annee = self.ui.cbx_annee.itemText(self.ui.cbx_annee.currentIndex()),\
zr_habref = self.ui.cbx_habref.itemData(self.ui.cbx_habref.currentIndex()),\
zr_habcod = self.habcod,\
zr_hablat = self.ui.cbx_hablat.itemText(self.ui.cbx_hablat.currentIndex()).replace("\'","\'\'"),\
zr_habfr = self.ui.cbx_habfr.itemText(self.ui.cbx_habfr.currentIndex()).replace("\'","\'\'"),\
zr_habcomment = self.ui.txt_comment.toPlainText().replace("\'","\'\'"),\
zr_codesyntax = v_syntax,\
zr_codecahab = v_cahab,\
zr_codeeur27 = self.codeeur27,\
zr_codeeunis = v_eunis,\
zr_codecorine = v_corine,\
zr_pourcent = self.ui.cbx_pourcent.itemText(self.ui.cbx_pourcent.currentIndex()),\
zr_rarete = self.rarete,\
zr_menace = self.menace,\
zr_interetpatr = self.interetpatr,\
zr_pat_cen = self.pat_cen,\
zr_thegeom = thegeom,\
zr_plantation = str(self.ui.chx_plantation.isChecked()).lower(),\
zr_idmosaik = id_mosaik,\
zr_habetat = ";".join([unicode(x.text()) for x in self.ui.lst_evol.selectedItems()]),\
zr_faciesa = self.ui.txt_faciesa.text().replace("\'","\'\'"))
ok = querysauvhab.exec_(query)
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête sauver Ope ratée')
self.erreurSaisieBase = '1'
self.iface.setActiveLayer(coucheactive)
QgsMapLayerRegistry.instance().removeMapLayer(memlayer.id())
# Réinitialiser certains contrôles
self.ui.lst_evol.clearSelection()
self.ui.txt_faciesa.setText('')
self.ui.txt_comment.setText('')
self.ui.cbx_eur27_mep.setCurrentIndex(0)
if self.erreurSaisieBase == '0':
QtGui.QMessageBox.information(self, 'Information', u'Données correctement saisies dans la base')
else :
QtGui.QMessageBox.warning(self, 'Alerte', u'Il y a eu une erreur lors de la saisie. Données non saisies en base.')
self.close()
class bdhabnatDialog(QtGui.QDialog):
def __init__(self, iface):
"""Constructor."""
QtGui.QDialog.__init__(self)
self.ui = Ui_bdhabnat_dialog()
self.ui.setupUi(self)
# Référencement de iface dans l'interface (iface = interface de QGIS)
self.iface = iface
self.canvas = self.iface.mapCanvas()
# Connexion à la base de données. DB type, host, user, password...
self.db = QtSql.QSqlDatabase.addDatabase("QPSQL") # QPSQL = nom du pilote postgreSQL
self.db.setHostName("192.168.0.10")
self.db.setDatabaseName("sitescsn")
self.db.setUserName("postgres")
self.db.setPassword("postgres")
ok = self.db.open()
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'La connexion est échouée'+self.db.hostName())
# Remplir la combobox "cbx_codesite" avec les codes et noms de sites issus de la table "t_sitescen"
query_codesite = QtSql.QSqlQuery(self.db)
if query_codesite.exec_('select idchamp, codesite, nomsite from sites_cen.t_sitescen order by codesite'):
while query_codesite.next():
self.ui.cbx_codesite.addItem(query_codesite.value(1) + " " + query_codesite.value(2), query_codesite.value(1) )
# Remplir la combobox "cbx_auteurs" avec les noms et prénoms issus de la table "t_liste_obsv"
query_obsv = QtSql.QSqlQuery(self.db)
if query_obsv.exec_('select id_obsv, nom_obsv, prenom_obsv, initiale_obsv from bd_habnat.t_liste_obsv order by nom_obsv, prenom_obsv'):
while query_obsv.next():
self.ui.cbx_auteur.addItem(query_obsv.value(1) + " " + query_obsv.value(2), query_obsv.value(3) )
# Combobox cbx_annee : Affecter l'année en cours par défaut en fonction de la date (changement d'année au 28/2/N)
if datetime.now().strftime('%m/%d') < '04/02' :
self.ui.cbx_annee.setCurrentIndex(self.ui.cbx_annee.findText((str(int(datetime.now().strftime('%Y'))-1)), QtCore.Qt.MatchStartsWith))
else:
self.ui.cbx_annee.setCurrentIndex(self.ui.cbx_annee.findText((datetime.now().strftime('%Y')), QtCore.Qt.MatchStartsWith))
# Remplir la combobox "cbx_habref" avec les noms de référentiels issus de la table "t_liste_ref"
query_ref = QtSql.QSqlQuery(self.db)
if query_ref.exec_('select id_ref, nom_ref, code_ref from bd_habnat.t_liste_ref order by id_ref'):
while query_ref.next():
self.ui.cbx_habref.addItem(query_ref.value(1), query_ref.value(2) )
self.ui.cbx_habref.model().item(0).setEnabled(False)
# Remplir la combobox "cbx_eur27_mep" avec le champs hab_cod de la table "t_liste_ref_eur27"
query_eur27 = QtSql.QSqlQuery(self.db)
if query_eur27.exec_('select hab_cod from bd_habnat.t_liste_ref_eur27 order by hab_cod'):
while query_eur27.next():
self.ui.cbx_eur27_mep.addItem(query_eur27.value(0), query_eur27.value(0) )
# Désactiver le bouton "OK" tant qu'on n'a pas choisi au moins un référentiel.
self.ui.buttonBox.setEnabled(False)
# Connexions signaux - slots
self.ui.cbx_habref.currentIndexChanged.connect(self.listesref)
self.ui.cbx_habfr.currentIndexChanged.connect(self.coreur27)
self.ui.buttonBox.accepted.connect(self.sauvSaisie)
self.ui.buttonBox.rejected.connect(self.close)
self.ui.chx_plantation.stateChanged.connect(self.plantation)
self.ui.cbx_hablat.currentIndexChanged.connect(self.listefran)
self.ui.cbx_habref.currentIndexChanged.connect(self.activButton)
def listesref(self):
# Réinitialisations.
# Attention de bien récupérer la nouvelle valeur de habref avant de vider hablat, pour ne pas lancer inopinément la fonction listefran.
self.habref = self.ui.cbx_habref.itemData(self.ui.cbx_habref.currentIndex())
self.ui.cbx_hablat.clear()
self.ui.cbx_habfr.clear()
self.ui.txt_codeeur27.clear()
self.ui.cbx_eur27_mep.setEnabled(True)
print 'listesref'
if self.habref == 'cbnbl':
# Si le référentiel "Digitale" du CBNBL est sélectionné, alors remplir la combobox "cbx_hablat"
query_cbnbl = QtSql.QSqlQuery(self.db)
qcbnbl = u'select hab_cod, hab_lat from (select * from bd_habnat.t_liste_ref_cbnbl_v12 UNION select * from bd_habnat.t_liste_ref_newlat_cbnbl) ref order by hab_lat'
ok = query_cbnbl.exec_(qcbnbl)
while query_cbnbl.next():
self.ui.cbx_hablat.addItem(query_cbnbl.value(1), query_cbnbl.value(0))
print 'cbnbl'
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête CBNBL ratée')
print qcbnbl
self.listefran()
else :
#Si un autre référentiel est sélectionné, alors remplir la combobox "cbx_habfr" avec les valeurs de la table correspondant au référentiel.
query_autrref = QtSql.QSqlQuery(self.db)
qautrref = u"""select hab_cod, hab_fr from bd_habnat.t_liste_ref_{zr_table} order by hab_fr""".format(zr_table = str(self.habref))
ok = query_autrref.exec_(qautrref)
print 'autre'
while query_autrref.next():
self.ui.cbx_habfr.addItem(query_autrref.value(1), query_autrref.value(0))
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête Autre ref ratée')
print unicode(qautrref)
if self.habref == 'eur27':
self.ui.cbx_eur27_mep.setEnabled(False)
def listefran(self):
#Quand le référentiel de Bailleul ou un taxon scientifique est sélectionné, afficher le taxon en français s'il est disponible
self.ui.cbx_habfr.clear()
if self.habref == 'cbnbl':
query_fr = QtSql.QSqlQuery(self.db)
qfr = u"""select hab_cod, hab_fr from bd_habnat.t_liste_ref_cbnbl_v12 cbnbl where cbnbl.hab_cod = {zr_codlat} order by hab_fr""".format (zr_codlat = self.ui.cbx_hablat.itemData(self.ui.cbx_hablat.currentIndex()))
ok = query_fr.exec_(qfr)
print 'listefran'
if not ok :
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête Hab Fr ratée')
while query_fr.next():
self.ui.cbx_habfr.addItem(query_fr.value(1), query_fr.value(0) )
def activButton(self):
# Activation du bouton "OK et annuler" si un habitat a été sélectionné
cbxhabrefindex = self.ui.cbx_habref.currentIndex()
if cbxhabrefindex!=0:
self.ui.buttonBox.setEnabled(1)
def coreur27(self):
# Remplissage éventuel de la zone de texte "txt_codeeur27"
self.ui.txt_codeeur27.clear()
self.codecoreu27 = self.ui.cbx_habfr.itemData(self.ui.cbx_habfr.currentIndex())
if self.habref == 'eur27':
self.ui.txt_codeeur27.setText(unicode(self.codecoreu27))
def plantation(self):
# Si l'habitat saisi est une plantation, il ne peut pas faire partie d'une mosaïque. => il prend 100 % de la surface.
# cbx_pourcent est désactivé est sa valeur fixée à 100.
if self.ui.chx_plantation.isChecked() == True:
self.ui.cbx_pourcent.setEnabled(0)
self.ui.cbx_pourcent.setCurrentIndex(self.ui.cbx_pourcent.findText('100', QtCore.Qt.MatchStartsWith))
else :
self.ui.cbx_pourcent.setEnabled(1)
def sauvSaisie(self):
if self.ui.cbx_habref.itemData(self.ui.cbx_habref.currentIndex())=='0':
return
self.erreurSaisieBase = '0'
#copie des entités sélectionnées dans une couche "memory". Evite les problèmes avec les types de couches "non éditables" (comme les GPX).
coucheactive=self.iface.activeLayer()
# On vérifie que les entités sélectionnées dans la couche active sont bien des polygones (et non des lignes ou des points)
entselect=[QgsGeometry(feature.geometry()) for feature in coucheactive.selectedFeatures()]
if entselect[0].type() == QGis.Polygon:
typegeom='Polygon'
# geom = self.transfoPoly(entselect)
# print "geom=" + str(geom.exportToWkt())
else:
QtGui.QMessageBox.warning(self, 'Alerte', u'Les entités sélectionnées ne sont pas des polygones')
self.close
return
# Création de la couche en mémoire "memlayer" et début de la session d'édition
if coucheactive.crs().authid() == u'EPSG:4326':
memlayer=QgsVectorLayer("{zr_typegeom}?crs=epsg:4326".format(zr_typegeom = typegeom), "memlayer", "memory")
if coucheactive.crs().authid() == u'EPSG:2154':
memlayer=QgsVectorLayer("{zr_typegeom}?crs=epsg:2154".format(zr_typegeom = typegeom), "memlayer", "memory")
else :
QtGui.QMessageBox.warning(self, 'Alerte', u'Les seules projections supportées sont le WGS84 et le Lambert93. Merci de vérifier la projection de votre couche SIG')
return
QgsMapLayerRegistry.instance().addMapLayer(memlayer, False)
root = QgsProject.instance().layerTreeRoot()
memlayerNode = QgsLayerTreeLayer(memlayer)
root.insertChildNode(0, memlayerNode)
self.iface.setActiveLayer(memlayer)
memlayer.startEditing()
# Pour chaque entité sélectionnée, si elle est multipartie, on ajoute chacune de ses parties individuellement à la couche memlayer.
# Sinon, on l'ajoute directement à "memlayer". Puis, on clot la session d'édition.
for feature in coucheactive.selectedFeatures() :
geom = feature.geometry()
temp_feature = QgsFeature(feature)
# check if feature geometry is multipart
if geom.isMultipart():
# if feature is multipart creates a new feature using the geometry of each part
for part in geom.asGeometryCollection ():
temp_feature.setGeometry(part)
memlayer.dataProvider().addFeatures([temp_feature])
memlayer.updateExtents()
# if feature is singlepart, simply adds it to the layer memory
else :
temp_feature.setGeometry(geom)
memlayer.dataProvider().addFeatures([temp_feature])
memlayer.updateExtents()
memlayer.commitChanges()
print "memlayercount="+str(memlayer.featureCount())
#on sélectionne toutes les entités de memlayer pour en faire une liste de géométries, qui sera saisie en base.
memlayer.selectAll()
geomlis = [QgsGeometry(feature.geometry()) for feature in memlayer.selectedFeatures()]
#geomlist = QgsGeometry.fromMultiPolygon([poly.asPolygon() for poly in geomlis]) --- transformation en multipolygone
for entite in geomlis :
#export de la géométrie en WKT et transformation de la projection si les données ne sont pas saisies en Lambert 93
if memlayer.crs().authid() == u'EPSG:2154':
thegeom='st_setsrid(st_geometryfromtext (\'{zr_geom}\'), 2154)'.format(zr_geom=entite.exportToWkt())
elif memlayer.crs().authid() == u'EPSG:4326':
thegeom='st_transform(st_setsrid(st_geometryfromtext (\'{zr_geom}\'),4326), 2154)'.format(zr_geom=entite.exportToWkt())
else :
print u'La projection de la couche active n\'est pas supportée'
#gestion de l'identifiant id_mosaik, servant à regrouper les enregistrements appartenant à une même mosaïque d'habitats
pourcent = int(self.ui.cbx_pourcent.itemText(self.ui.cbx_pourcent.currentIndex()))
if pourcent > 99 :
print ">99"
id_mosaik = 0
else :
querymosaik = QtSql.QSqlQuery(self.db)
qmosaik = u"""SELECT id_hab_ce, annee, pourcent,the_geom, plantation, id_mosaik FROM bd_habnat.t_ce_saisie WHERE the_geom = {zr_thegeom} AND annee = '{zr_annee}' AND plantation = 'f' """.format (\
zr_thegeom = thegeom,\
zr_annee = self.ui.cbx_annee.itemText(self.ui.cbx_annee.currentIndex()))
ok = querymosaik.exec_(qmosaik)
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête Mosaik ratée')
if querymosaik.size() > 0 :
print "on est deja dans la mosaique"
sumprct = 0
while querymosaik.next() :
sumprct += int(querymosaik.value(2))
sumprct += pourcent
if sumprct > 100 :
QtGui.QMessageBox.warning(self, 'Alerte', u'La somme des pourcentages des habitats dépasse 100 % dans la mosaïque')
return
querymosaik.first()
id_mosaik = querymosaik.value(5)
else :
querybiggestid = QtSql.QSqlQuery(self.db)
qbiggestid = u"""SELECT id_mosaik FROM bd_habnat.t_ce_saisie ORDER BY id_mosaik DESC LIMIT 1"""
ok = querybiggestid.exec_(qbiggestid)
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête PlusGrandIdMosaik ratée')
querybiggestid.next()
print "debut de la mosaique"
if self.ui.chx_plantation.isChecked == True :
id_mosaik = 0
else :
id_mosaik = int(querybiggestid.value(0))+1
self.habref = self.ui.cbx_habref.itemData(self.ui.cbx_habref.currentIndex())
# si ref = cbnbl : recup rareté, menace et intérêt patri en fonction du taxon sélectionné par l'utilisateur
if self.habref == 'cbnbl':
queryrarmen = QtSql.QSqlQuery(self.db)
qrarmen = u"""SELECT rarete, menace, interetpatr FROM bd_habnat.t_liste_ref_cbnbl_v12 WHERE hab_lat = '{zr_hablat}'""".format (\
zr_hablat= self.ui.cbx_hablat.itemText(self.ui.cbx_hablat.currentIndex()).replace("\'","\'\'"))
okrarmen=queryrarmen.exec_(qrarmen)
if not okrarmen :
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête RarMen ratée')
queryrarmen.next()
self.rarete = queryrarmen.value(0)
self.menace = queryrarmen.value(1)
self.interetpatr = queryrarmen.value(2)
# construction de pat_cen en fonction de intérêt patri
querypatcen = QtSql.QSqlQuery(self.db)
qpatcen = u""" SELECT CASE WHEN cbnbl.interetpatr IN ('Oui','#') THEN True ELSE False END FROM bd_habnat.t_liste_ref_cbnbl_v12 cbnbl WHERE hab_lat = '{zr_hablat}'""".format (\
zr_hablat= self.ui.cbx_hablat.itemText(self.ui.cbx_hablat.currentIndex()).replace("\'","\'\'"))
okpatcen = querypatcen.exec_(qpatcen)
if not okpatcen :
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête PatCen ratée')
querypatcen.next()
self.pat_cen = querypatcen.value(0)
# si ref != cbnbl : rareté, menace, intérêt patri = '/' et pat_cen = False
else :
self.rarete = '/'
self.menace = '/'
self.interetpatr = '/'
self.pat_cen = 'false'
# aller chercher hab_cod dans la cbx "lat" si self.habref = cbnbl, sinon dans cbx "fr".
# aller chercher code_syntax, code_cahab, code_eunis et code_corine dans itemData cbx_hablat ou cbx_habfr, selon le référentiel choisi
if self.habref == 'cbnbl':
self.habcod = self.ui.cbx_hablat.itemData(self.ui.cbx_hablat.currentIndex())
v_syntax = self.habcod
v_cahab = ''
v_eunis =''
v_corine =''
else :
self.habcod = self.ui.cbx_habfr.itemData(self.ui.cbx_habfr.currentIndex())
if self.habref == 'eur27' :
v_syntax = ''
v_cahab = self.habcod
v_eunis = ''
v_corine = ''
elif self.habref == 'eunis' :
v_syntax = ''
v_cahab = ''
v_eunis = self.habcod
v_corine = ''
elif self.habref == 'corine' :
v_syntax = ''
v_cahab = ''
v_eunis = ''
v_corine = self.habcod
# récupération du code N2000 par txt_codeeur27 s'il est renseigné, sinon cbx_eur27_mep
if self.ui.txt_codeeur27.text() != '':
print 'txt_codeeur27 renseigne'
self.codeeur27 = self.ui.txt_codeeur27.text()
else :
if self.ui.cbx_eur27_mep.currentIndex() == 0 :
self.codeeur27 = ''
else :
self.codeeur27 = self.ui.cbx_eur27_mep.itemText(self.ui.cbx_eur27_mep.currentIndex())
#lancement de la requête SQL qui introduit les données géographiques et du formulaire dans la base de données.
querysauvhab = QtSql.QSqlQuery(self.db)
query = u"""INSERT INTO bd_habnat.t_ce_saisie(codesite, auteur, annee, hab_ref, hab_cod, hab_lat, hab_fr, hab_comment, code_syntax, code_cahab, code_eur27, code_eunis, code_corine, pourcent, rarete, menace, patrimoine, pat_cen, surf_tot, the_geom, plantation, id_mosaik, hab_etat, faciesa, date_crea_poly) VALUES ('{zr_codesite}', '{zr_auteur}', '{zr_annee}', '{zr_habref}','{zr_habcod}', '{zr_hablat}', '{zr_habfr}', '{zr_habcomment}', '{zr_codesyntax}', '{zr_codecahab}', '{zr_codeeur27}', '{zr_codeeunis}', '{zr_codecorine}', '{zr_pourcent}', '{zr_rarete}', '{zr_menace}', '{zr_interetpatr}', {zr_pat_cen}, st_area({zr_thegeom}), {zr_thegeom}, {zr_plantation}, {zr_idmosaik}, '{zr_habetat}', '{zr_faciesa}', current_date)""".format (\
zr_codesite = self.ui.cbx_codesite.itemData(self.ui.cbx_codesite.currentIndex()),\
zr_auteur = self.ui.cbx_auteur.itemData(self.ui.cbx_auteur.currentIndex()),\
zr_annee = self.ui.cbx_annee.itemText(self.ui.cbx_annee.currentIndex()),\
zr_habref = self.ui.cbx_habref.itemData(self.ui.cbx_habref.currentIndex()),\
zr_habcod = self.habcod,\
zr_hablat = self.ui.cbx_hablat.itemText(self.ui.cbx_hablat.currentIndex()).replace("\'","\'\'"),\
zr_habfr = self.ui.cbx_habfr.itemText(self.ui.cbx_habfr.currentIndex()).replace("\'","\'\'"),\
zr_habcomment = self.ui.txt_comment.toPlainText().replace("\'","\'\'"),\
zr_codesyntax = v_syntax,\
zr_codecahab = v_cahab,\
zr_codeeur27 = self.codeeur27,\
zr_codeeunis = v_eunis,\
zr_codecorine = v_corine,\
zr_pourcent = self.ui.cbx_pourcent.itemText(self.ui.cbx_pourcent.currentIndex()),\
zr_rarete = self.rarete,\
zr_menace = self.menace,\
zr_interetpatr = self.interetpatr,\
zr_pat_cen = self.pat_cen,\
zr_thegeom = thegeom,\
zr_plantation = str(self.ui.chx_plantation.isChecked()).lower(),\
zr_idmosaik = id_mosaik,\
zr_habetat = ";".join([unicode(x.text()) for x in self.ui.lst_evol.selectedItems()]),\
zr_faciesa = self.ui.txt_faciesa.text().replace("\'","\'\'"))
ok = querysauvhab.exec_(query)
if not ok:
QtGui.QMessageBox.warning(self, 'Alerte', u'Requête sauver Ope ratée')
self.erreurSaisieBase = '1'
self.iface.setActiveLayer(coucheactive)
QgsMapLayerRegistry.instance().removeMapLayer(memlayer.id())
# Réinitialiser certains contrôles
self.ui.lst_evol.clearSelection()
self.ui.txt_faciesa.setText('')
self.ui.txt_comment.setText('')
self.ui.cbx_eur27_mep.setCurrentIndex(0)
if self.erreurSaisieBase == '0':
QtGui.QMessageBox.information(self, 'Information', u'Données correctement saisies dans la base')
else :
QtGui.QMessageBox.warning(self, 'Alerte', u'Il y a eu une erreur lors de la saisie. Données non saisies en base.')
self.close()
