Як створити скрипт обробки 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 дає справжні результати, нижче наведені сценарії повинні робити те, що ви хочете. На жаль, у мене немає нічого для того, щоб перевірити це, але ви можете спробувати і дати відгук.

Я намагався прокоментувати це для зручності, в основному сценарій виконує три дії:

• отримання параметрів підключення до бази даних для вибраного шару (має бути постгрес)
• заповніть параметри з'єднання в рядку заяви 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 ..)

У панелі інструментів "Обробка" перейдіть до Сценарії> Інструменти> Отримати сценарії з Інтернету ...

Розгорніть "Не встановлено" та виберіть "ЕквівалентнеNumField". Не забудьте встановити прапорець перед тим, як натиснути кнопку ОК. Це мене зловило ... ;-)

Щоб швидко знайти його, введіть "Equiv" на панелі пошуку обробки, і ви повинні мати можливість двічі клацнути по ньому.

Ось приклад. Ці ліси мали унікальне поле (osm_id), але плагін замість цього додав NUM_FIELD з простими числовими значеннями