INSERT INTO и конкатенация строк с Python - PullRequest
Новая
       19

INSERT INTO и конкатенация строк с Python

11 голосов
/ 25 мая 2011

Я столкнулся с серьезным ударом по скорости вставки данных в мою БД. Из приведенного ниже кода видно, что я просто создаю инструкцию SQL для передачи в команду execute. Значения верны, и все хорошо, но интерпретатор python, кажется, добавляет и удаляет кавычки из параметров во время выполнения.

Это правильный способ вставки пространственных данных в БД. 

INSERT INTO my_table(
            name, url, id, point_geom, poly_geom)
    VALUES ('test', 'http://myurl', '26971012', 
            ST_GeomFromText('POINT(52.147400 19.050780)',4326), 
            ST_GeomFromText('POLYGON(( 52.146542 19.050557, bleh, bleh, bleh))',4326));

Это проверяется в редакторе запросов Postgres ... Теперь, когда я запускаю приведенный ниже код Python, он добавляет двойные кавычки вокруг функции ST_GeomFromText и затем удаляет кавычки из столбца id. 

INSERT INTO my_table(
            name, url, id, point_geom, poly_geom)
    VALUES ('test', 'http://myurl', 26971012, 
     "ST_GeomFromText('POINT(52.147400 19.050780)',4326)", 
     "ST_GeomFromText('POLYGON(( 52.146542 19.050557, 52.148430 19.045527, 52.149525 19.045831, 52.147400 19.050780, 52.147400 19.050780, 52.146542 19.050557))',4326)");

Это приводит к сбою вставки, и PostGIS утверждает, что это неправильная геометрия. Когда я печатаю каждый из параметров для просмотра на экране, в кавычках нет ничего смешного, поэтому я думаю, что проблема должна заключаться в команде execute. Я использую Python 2.7 ... Кто-нибудь может помочь, чтобы предотвратить продолжение этого безумия? 

    conn = psycopg2.connect('dbname=mydb user=postgres password=password')
    cur = conn.cursor()
    SQL = 'INSERT INTO my_table (name, url, id, point_geom, poly_geom) VALUES (%s,%s,%s,%s,%s);'
    Data = name, url, id, point, polygon
    #print Data
    cur.execute(SQL, Data)
    conn.commit()
    cur.close()
    conn.close()

Ответы [ 2 ]

18 голосов
/ 25 мая 2011

Вы передаете строку ST_GeomFromText('POINT(..)') в качестве параметра, а psycopg2 избегает ее.Однако ST_GeomFromText(..) - это функция PostGIS, а не данные, которые вы вставляете.

Чтобы исправить это, вам нужно переместить функцию ST_GeomFromText в статический SQL.Например: 

sql = '''
   INSERT INTO foo (point_geom, poly_geom)
   VALUES (ST_PointFromText(%s, 4326), ST_GeomFromText(%s, 4326))'''
params = ['POINT( 20 20 )', 'POLYGON(( 0 0, 0 10, 10 10, 10 0, 0 0 ))']
cur.execute(sql, params)
0 голосов
/ 25 мая 2011

Лучше (отредактируйте: чем то, что вы делаете, чем первым ответом) сделать что-то вроде этого: 

select st_asgeojson(the_geom_ls) from atlas where the_geom_ls &&\
st_geomfromtext('POLYGON((%s %s, %s %s, %s %s, %s %s, %s %s))', 4326)

И передайте это вашему адаптеру с координатами в кортеже, они будут проанализированы правильно. Это, безусловно, наименее болезненный способ сделать это. Если вам действительно нужно создавать строки отдельно, вы можете очистить экранирование, посмотрев вывод cur.mogrify: 

cur.mogrify('INSERT INTO my_table (name, url, id, point_geom, poly_geom) VALUES (%s,%s,%s,%s,%s);', (1,2,3,4,5))
>>>'INSERT INTO my_table (name, url, id, point_geom, poly_geom) VALUES (1,2,3,4,5);'

query = "ST_GeomFromText('POLYGON(( 52.146542 19.050557, 52.148430 19.045527, 52.149525 19.045831, 52.147400 19.050780, 52.147400 19.050780, 52.146542 19.050557))',4326)"

cur.mogrify('INSERT INTO my_table (name, url, id, point_geom, poly_geom) VALUES (%s,%s,%s,%s,%s);', (1,2,3,4,query))
>>> "INSERT INTO my_table (name, url, id, point_geom, poly_geom) VALUES (1,2,3,4,E'ST_GeomFromText(''POLYGON(( 52.146542 19.050557, 52.148430 19.045527, 52.149525 19.045831, 52.147400 19.050780, 52.147400 19.050780, 52.146542 19.050557))'',4326)');"

Не поддавайтесь искушению использовать метод форматирования строки в строке запроса sql, даже если это может разочаровать.

...