Oto kilka modyfikacji, które powinny sprawić, że wszystko będzie działać. Zwróć uwagę, że wymagałoby to dalszych modyfikacji, jeśli chcesz zostać powiadomiony, jeśli którekolwiek z poleceń nie powiedzie się. Zauważ, że nie powiedzie się dla więcej niż jednego pliku kształtu, ponieważ new_shp_table
tabela będzie już istnieć, dopóki nie będziesz mieć dalszej logiki, aby przenieść lub zmienić nazwę tej tabeli w inne miejsce lub załadować ją z unikalną nazwą.
Należy również pamiętać, że PostgreSQL 8.4 dobiegnie końca w tym roku, więc warto zaplanować aktualizację do nowszej wersji, zanim będzie za późno.
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)