Ето някои модификации, които трябва да накарат нещата да работят. Имайте предвид, че ще се нуждае от допълнителна модификация, ако трябва да бъдете уведомени, ако някоя от командите е неуспешна. Обърнете внимание, че няма да успее за повече от един шейпфайл, тъй като new_shp_table
таблицата вече ще съществува, докато нямате допълнителна логика да преместите или преименувате тази таблица другаде или да я заредите с уникално име.
Също така имайте предвид, че PostgreSQL 8.4 ще достигне края на живота си по-късно тази година, така че може да искате да планирате надстройка до по-нова версия, преди да е станало твърде късно.
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)