使用Python INSERT INTO和String Concatenation

时间:2011-05-24 22:24:46

标签: python postgis psycopg2

在将数据插入我的数据库时,我遇到了一个重大的速度障碍。您可以从下面的代码中看到,我只是构建SQL语句以传递给execute命令。值是正确的,一切都很好,但python解释器似乎是在运行时添加和删除params中的引号。

这是将空间数据插入数据库的正确方法。

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声称它不是正确的几何图形。当我打印每个参数在屏幕上查看时,引号没有什么好玩的,所以我认为问题必须在执行命令中。我正在使用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 个答案:

答案 0 :(得分:18)

您正在传递字符串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)

答案 1 :(得分:0)

更好(编辑:比你正在做的,而不是第一个答案)做这样的事情:

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)

使用元组中的coords将它传递给适配器,它们将被正确解析。这是迄今为止最不痛苦的方法。如果你真的必须单独构造字符串,你可以通过查看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查询字符串上使用字符串格式化方法的诱惑,即使这些事情令人沮丧。