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)