Как создать сценарий обработки QGIS, который добавляет последовательность к столбцу уникального идентификатора в PostGIS?

Кто-нибудь может мне помочь создать сценарий обработки QGIS, который добавляет последовательность к существующему столбцу уникального идентификатора (тип: целое число) в PostGIS?

Это было бы весьма полезно, например, в качестве обходного пути для ошибки № 6798 . К сожалению, у меня нет опыта работы с Python.

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);

Стоит отметить, что модуль python psycopg2не кажется автоматически COMMITтранзакцией (как это делают другие клиенты, такие как QGIS DB Manager или pgAdmin), поэтому COMMITоператор должен быть частью sqlстроки в скрипте.

Это не имеет значения для SELECTутверждений, потому что в этих случаях a COMMIT, очевидно, выполняется при получении результатов через cur.fetchall().

Это переработанная версия скрипта из моего ответа выше:

#!/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.')

Если ваш оператор SQL дает правильные результаты, приведенные ниже сценарии должны делать то, что вам нужно. К сожалению, у меня нет ничего под рукой, чтобы проверить это, но вы можете попробовать и дать отзыв.

Я попытался прокомментировать это для удобства, в основном скрипт выполняет три шага:

• получить параметры подключения к базе данных для выбранного слоя (должен быть postgres)
• заполните параметры подключения в строке оператора sql
• выполнить оператор SQL

Обратите внимание на вывод протокола скрипта.

#!/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.')

Похоже, что подобный плагин уже есть (хотя он создает для вас новое уникальное поле идентификатора, а не последовательность).

Предполагается, что у вас уже есть уникальное поле идентификатора (оно не обязательно должно быть числовым), но вместо этого вам нужен простой числовой идентификатор (1,2,3 ..)

В панели инструментов Обработка перейдите в Сценарии> Инструменты> Получить сценарии из Интернета ...

Разверните «Не установлено» и выберите «EquivalentNumField». Не забудьте установить флажок перед нажатием кнопки ОК. Это застало меня врасплох ... ;-)

Чтобы быстро найти его, введите «Эквив» в строке поиска обработки, и вы сможете дважды щелкнуть по нему.

Вот пример. Эти леса имели уникальное поле (osm_id), но плагин добавил NUM_FIELD с простыми числовыми значениями