Inserir um ponto no PostGIS usando Python

11

Qual é a maneira correta de inserir um ponto em um banco de dados PostGIS usando Python?

Adam Matan
fonte

Respostas:

21

Primeiro, instale o pacote psycopg2 , uma interface Pythonic para o PostgreSQL.

Então, use ST_MakePoint:

>>> import psycopg2
>>> conn = psycopg2.connect(dbname=..., port=..., user=...,
                            password=..., host=...)
>>> cur = conn.cursor()
>>> x, y, z, = 32, 34, 0
>>> cur.execute("SELECT ST_SetSRID(ST_MakePoint(%s, %s, %s),4326);", (x, y, z))
>>> cur.fetchall()
[('01010000A0E6100000000000000000404000000000000041400000000000000000',)]

ST_AsText pode ser usado para validar os valores:

>>> cur.execute("SELECT ST_AsText(ST_SetSRID(ST_MakePoint(%s, %s, %s),4326));", (x, y, z))
>>> cur.fetchall()
[('POINT Z (32 34 0)',)]

Notas

  • Lembre-se que (lat, lon)é (y, x), não (x, y).
  • Sempre use parâmetros, em vez de manipulações de string, para impedir a injeção de SQL . Neste exemplo, fizemos uma tupla (x, y, z)no final para que psycopg2possamos lidar com a substituição.
Adam Matan
fonte
20

Para geometrias mais complicadas, como as geometrias LineString e Polygon, você pode manipulá-las com Shapely e passá-las pelo psycopg2 como WKB codificado em hexadecimal. Observe que Shapely 1.3 ou posterior é necessário para manipular a exportação de geometrias 3D com a wkb_hexpropriedade

import psycopg2
from shapely.geometry import LineString
from shapely import wkb

conn = psycopg2.connect('...')
curs = conn.cursor()

# Make a Shapely geometry
ls = LineString([(2.2, 4.4, 10.2), (3.3, 5.5, 8.4)])
ls.wkt  # LINESTRING Z (2.2 4.4 10.2, 3.3 5.5 8.4)
ls.wkb_hex  # 0102000080020000009A999999999901409A999999999911406666666666662440666666...

# Send it to PostGIS
curs.execute('CREATE TEMP TABLE my_lines(geom geometry, name text)')
curs.execute(
    'INSERT INTO my_lines(geom, name)'
    'VALUES (ST_SetSRID(%(geom)s::geometry, %(srid)s), %(name)s)',
    {'geom': ls.wkb_hex, 'srid': 4326, 'name': 'First Line'})

conn.commit()  # save data

# Fetch the data from PostGIS, reading hex-encoded WKB into a Shapely geometry
curs.execute('SELECT name, geom FROM my_lines')
for name, geom_wkb in curs:
    geom = wkb.loads(geom_wkb, hex=True)
    print('{0}: {1}'.format(name, geom.wkt))
# First Line: LINESTRING Z (2.2 4.4 10.2, 3.3 5.5 8.4)

Observe também que similar pode ser conseguido enviando o WKT da geometria, no entanto, uma vez que é convertido em texto, é com perdas e pode reduzir angstroms de precisão. A transferência de geometrias como WKB codificado em hexadecimal é sem perdas e preserva a precisão exata de cada coordenada.

Mike T
fonte
Excelente, obrigado! Gostaria de saber se existe uma diferença de desempenho entre essas duas abordagens.
Adam Matan
Há uma vantagem de 10% no desempenho de construção de pontos ST_MakePoint, o que é ótimo para geometrias de pontos. No entanto, criar tipos de geometria mais complexos geralmente será mais simples com o Shapely.
Mike T