Jak utworzyć skrypt przetwarzania QGIS, który dodaje sekwencję do unikalnej kolumny identyfikatora w PostGIS?

Czy ktoś może mi pomóc w utworzeniu skryptu przetwarzania QGIS, który dodaje sekwencję do istniejącej unikalnej kolumny identyfikatora (typ: liczba całkowita) w PostGIS?

Byłoby to bardzo pomocne, np. Jako obejście błędu 6798 . Niestety nie mam doświadczenia w Pythonie.

CREATE SEQUENCE /*input_schema*/./*input_table*/_/*uic*/_seq OWNED BY /*input_schema*/./*input_table*/./*uic*/;
SELECT SETVAL('/*input_schema*/./*input_table*/_/*uic*/_seq', (SELECT MAX(/*uic*/) FROM /*input_schema*/./*input_table*/));
ALTER TABLE /*input_schema*/./*input_table*/
ALTER COLUMN /*uic*/ SET DEFAULT nextval('/*input_schema*/./*input_table*/_/*uic*/_seq'::regclass);

Warto zauważyć, że moduł Pythona psycopg2nie wydaje się automatycznie COMMITtransakcją (jak robią to inni klienci, tacy jak QGIS DB Manager lub pgAdmin), dlatego COMMITinstrukcja musi być częścią sqlciągu w skrypcie.

Nie ma to znaczenia w przypadku SELECTinstrukcji, ponieważ w tych przypadkach a COMMITjest oczywiście przeprowadzane przy uzyskiwaniu wyników cur.fetchall().

To jest przerobiona wersja skryptu z mojej powyższej odpowiedzi:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

#--------- define Interface
##[my_scripts]=group
##Add Serial to PostgreSQL Table=name
##Postgres_Table=vector
##Unique_identifier_column=field Postgres_Table

from PyQt4.QtCore import *
from PyQt4.QtGui import *
from PyQt4 import *

from qgis.core import *
from qgis.gui import *
from qgis.utils import *

import psycopg2

#get the parameters for the tpg table into a directory
#get the table
pg_table = processing.getObject(Postgres_Table)
#create empty dictionary for key/value pairs of the tables connection parameters
db_params = {}
db_params['uic'] = Unique_identifier_column
#iterate over connection string
progress.setInfo(20*'-' + '  Connection parameters')
for param in pg_table.dataProvider().dataSourceUri().split(' '):
    key_val = param.split('=')
    progress.setInfo(str(key_val))
    try:
        #set key/value pair
        db_params[key_val[0]] = key_val[1]
    except:
        pass

#generate the sql statement string
#the text in round brackets are the keys from the db_params dictionary created above
#the values belonging to the keys are inserted into the string
progress.setInfo(20*'-' + '  SQL statement')
sql = """CREATE SEQUENCE %(table)s_%(uic)s_seq OWNED BY %(table)s.%(uic)s;
SELECT SETVAL('%(table)s_%(uic)s_seq', (SELECT MAX(%(uic)s) FROM %(table)s)); 
ALTER TABLE %(table)s ALTER COLUMN %(uic)s SET DEFAULT nextval('%(table)s_%(uic)s_seq'::regclass);
COMMIT;""" % db_params
#remove double quotes
sql = sql.replace('"','') 
progress.setInfo(sql)

#make connection string
constr = """dbname=%(dbname)s host=%(host)s port=%(port)s user=%(user)s     password=%(password)s""" % db_params
progress.setInfo(20*'-' + '  DB Connection string')
progress.setInfo(constr)
#make db connection
con = psycopg2.connect(constr)
cur = con.cursor()
#execute the above created sql statement
progress.setInfo(20*'-' + '  Executing SQL statement ...')
cur.execute(sql)
progress.setInfo(20*'-' + '  ... done.')

Pod warunkiem, że twoja instrukcja SQL generuje prawidłowe wyniki, poniższe skrypty powinny robić to, czego szukasz. Niestety nie mam nic pod ręką, aby to przetestować, ale możesz spróbować wyrazić opinię.

Próbowałem to skomentować dla wygody, w zasadzie skrypt wykonuje trzy kroki:

• uzyskaj parametry połączenia z bazą danych dla wybranej warstwy (powinno być postgres)
• wypełnij parametry połączenia w ciągu instrukcji sql
• wykonać instrukcję sql

Zanotuj wynik protokołu skryptu.

#!/usr/bin/env python
# -*- coding: utf-8 -*-

#--------- define Interface
##[my_scripts]=group
##Add Serial to PostgreSQL Table=name
##Postgres_Table=vector
##Unique_identifier_column=string replace_this_with_your_uic

from PyQt4.QtCore import *
from PyQt4.QtGui import *
from PyQt4 import *

from qgis.core import *
from qgis.gui import *
from qgis.utils import *

import psycopg2

#get the parameters for the tpg table into a directory
#get the table
pg_table = processing.getObject(Postgres_Table)
#create empty dictionary for key/value pairs of the tables connection parameters
db_params = {}
db_params['uic'] = Unique_identifier_column
#iterate over connection string
progress.setInfo(20*'-' + '  Connection parameters')
for param in pg_table.dataProvider().dataSourceUri().split(' '):
    key_val = param.split('=')
    progress.setInfo(str(key_val))
    try:
        #set key/value pair
        db_params[key_val[0]] = key_val[1]
    except:
        pass

#generate the sql statement string
#the text in round brackets are the keys from the db_params dictionary created above
#the values belonging to the keys are inserted into the string
progress.setInfo(20*'-' + '  SQL statement')
sql = """CREATE SEQUENCE %(table)s_%(uic)s_seq OWNED BY %(table)s.%(uic)s;
            SELECT SETVAL(%(table)s_%(uic)s_seq, (SELECT MAX(%(uic)s) FROM %(table)s));
            ALTER TABLE %(table)s
            ALTER COLUMN %(uic)s SET DEFAULT nextval(%(table)s_%(uic)s_seq::regclass);""" % db_params
#remove double quotes
sql = sql.replace('"','') 
progress.setInfo(sql)

#make connection string
constr = """dbname=%(dbname)s host=%(host)s port=%(port)s user=%(user)s     password=%(password)s""" % db_params
progress.setInfo(20*'-' + '  DB Connection string')
progress.setInfo(constr)
#make db connection
con = psycopg2.connect(constr)
cur = con.cursor()
#execute the above created sql statement
progress.setInfo(20*'-' + '  Executing SQL statement ...')
cur.execute(sql)
progress.setInfo(20*'-' + '  ... done.')

Wydaje się, że już istnieje podobna wtyczka (chociaż tworzy dla ciebie nowe unikalne pole identyfikatora, zamiast tworzyć sekwencję).

Zakłada się, że masz już unikalne pole identyfikatora (nie musi to być wartość liczbowa), ale zamiast tego potrzebujesz prostego identyfikatora numerycznego (1,2,3 ..)

W przyborniku Przetwarzanie przejdź do Skrypty> Narzędzia> Pobierz skrypty z Internetu ...

Rozwiń „Niezainstalowany” i wybierz „EquivalentNumField”. Pamiętaj, aby kliknąć pole wyboru przed kliknięciem OK. To mnie złapało ... ;-)

Aby szybko go znaleźć, wpisz „Equiv” na pasku wyszukiwania przetwarzania i powinieneś być w stanie kliknąć go dwukrotnie.

Oto przykład. Te lasy miały unikalne pole (osm_id), ale wtyczka dodała NUM_FIELD z prostymi wartościami liczbowymi