PostgreSQL
 sql >> Database >  >> RDS >> PostgreSQL

Script Python per assimilare uno shapefile in un database PostgreSQL/PostGIS utilizzando shp2pgsql.exe su Windows

Ecco alcune modifiche che dovrebbero far funzionare la cosa. Si noti che sarebbe necessaria un'ulteriore modifica se è necessario essere avvisati se uno qualsiasi dei comandi fallisce. Nota che fallirà per più di uno shapefile, poiché un new_shp_table la tabella esisterà già fino a quando non avrai ulteriore logica per spostare o rinominare quella tabella altrove o per caricarla con un nome univoco.

Inoltre, tieni presente che PostgreSQL 8.4 raggiungerà la fine del suo ciclo di vita entro la fine dell'anno, quindi potresti voler pianificare l'aggiornamento a una versione più recente prima che sia troppo tardi.

import os, subprocess

# Choose your PostgreSQL version here
os.environ['PATH'] += r';C:\Program Files (x86)\PostgreSQL\8.4\bin'
# http://www.postgresql.org/docs/current/static/libpq-envars.html
os.environ['PGHOST'] = 'localhost'
os.environ['PGPORT'] = '5432'
os.environ['PGUSER'] = 'someuser'
os.environ['PGPASSWORD'] = 'clever password'
os.environ['PGDATABASE'] = 'geometry_database'

base_dir = r"c:\shape_file_repository"
full_dir = os.walk(base_dir)
shapefile_list = []
for source, dirs, files in full_dir:
    for file_ in files:
        if file_[-3:] == 'shp':
            shapefile_path = os.path.join(base_dir, file_)
            shapefile_list.append(shapefile_path)
for shape_path in shapefile_list:
    cmds = 'shp2pgsql "' + shape_path + '" new_shp_table | psql '
    subprocess.call(cmds, shell=True)