Changeset 818 for buildout/pcl.buildout/trunk/init_postgresql.py

Timestamp:

08/09/07 06:49:21 (3 years ago)

Author:

dokai

Message:

Added support for setting the PostgreSQL port by running:
./bin/python init_postgresql.py --port=[port_number]

Files:

• buildout/pcl.buildout/trunk/init_postgresql.py (modified) (3 diffs)

buildout/pcl.buildout/trunk/init_postgresql.py

@@ -9 +9 @@
 import shutil
 
+from optparse import OptionParser
+
 container = os.getcwd()
 PGDATA = os.path.join(container, 'var', 'postgresql')
 PGROOT = os.path.join(container, 'parts', 'postgresql')
-PGPORT = 5432
 PGDBNAME = 'pcl_test'
 POSTGIS = os.path.join(container, 'parts', 'postgis')
@@ -18 +19 @@
 
 wrapper_script = os.path.join(container, 'bin', 'pg_ctl')
+wrapper_script_tmpl = """\
+#!/bin/sh
+PGDATA=%s %s/bin/pg_ctl $@
+""" 
+
+test_env_tmpl = """\
+[test-environment]
+PCLCORE_DSN = host=localhost dbname=pcl_test port=%(port)s
+"""
 
 def system(cmd):
@@ -23 +33 @@
         raise RuntimeError('Error running command: %s' % cmd)
 
-PIDFILE = '%s/postmaster.pid' % PGDATA
-if os.path.exists(wrapper_script) and os.path.exists(PIDFILE):
-    print 'Shutting down PostgreSQL server...'
-    system('%s stop' % wrapper_script)
-    while os.path.exists(PIDFILE):
-        time.sleep(1)
+def initpg(port):
+    """Initializes a test PostgreSQL server."""
+    # Shut down the server if it is running
+    PIDFILE = '%s/postmaster.pid' % PGDATA
+    if os.path.exists(wrapper_script) and os.path.exists(PIDFILE):
+        print 'Shutting down PostgreSQL server...'
+        system('%s stop' % wrapper_script)
+        while os.path.exists(PIDFILE):
+            time.sleep(1)
 
-# Clean up existing parts
-if os.path.exists(PGDATA):
-    shutil.rmtree(PGDATA)
+    # Clean up existing parts
+    if os.path.exists(PGDATA):
+        shutil.rmtree(PGDATA)
 
-assert not os.path.exists(PGDATA)
-os.makedirs(PGDATA)
+    assert not os.path.exists(PGDATA)
+    os.makedirs(PGDATA)
 
-# Initialize the database
-system('%s/bin/initdb --auth=trust -D %s' % (PGROOT, PGDATA))
+    # Initialize the database
+    system('%s/bin/initdb --auth=trust -D %s' % (PGROOT, PGDATA))
 
-# Create a wrapper script for pg_ctl
-wrapper = open(wrapper_script, 'w')
-wrapper.write("""\
-#!/bin/sh
-PGDATA=%s %s/bin/pg_ctl $@
-""" % (PGDATA, PGROOT))
-wrapper.close()
-os.chmod(wrapper_script, stat.S_IRWXU)
+    # Create a wrapper script for pg_ctl
+    wrapper = open(wrapper_script, 'w')
+    wrapper.write(wrapper_script_tmpl % (PGDATA, PGROOT))
+    wrapper.close()
+    os.chmod(wrapper_script, stat.S_IRWXU)
 
-system('%s start' % wrapper_script)
-time.sleep(2)
+    # Update the port setting and start up the server
+    conffile = '%s/postgresql.conf' % PGDATA
+    f = open(conffile)
+    conf = ('port = %s' % port).join(f.read().split('#port = 5432'))
+    f.close()
+    open(conffile, 'w').write(conf)
+    system('%s start' % wrapper_script)
+    time.sleep(2)
+
+    # Convert the shapefile data to SQL insert statements
+    system('%s/bin/shp2pgsql -I -s 4269 %s postgis.world_borders > /tmp/wb.sql' % (POSTGIS, DATASET))
+
+    # Create the test database
+    system('%s/bin/createdb --port=%s %s' % (PGROOT, port, PGDBNAME))
+    system('%s/bin/createlang --port=%s plpgsql %s' % (PGROOT, port, PGDBNAME))
+    system('%s/bin/psql -p %s -d %s -c "CREATE SCHEMA postgis"' % (PGROOT, port, PGDBNAME))
+    system('%s/bin/psql -p %s -d %s -f %s/share/lwpostgis.sql' % (PGROOT, port, PGDBNAME, POSTGIS))
+    system('%s/bin/psql -p %s -d %s -f %s/share/spatial_ref_sys.sql' % (PGROOT, port, PGDBNAME, POSTGIS))
+
+    # Import the data
+    system('%s/bin/psql -p %s -d %s -f /tmp/wb.sql' % (PGROOT, port, PGDBNAME))
+    system('%s/bin/psql -p %s -d %s -c "CREATE VIEW postgis.uk_borders AS SELECT * FROM postgis.world_borders WHERE fips_cntry = \'UK\'"' % (PGROOT, port, PGDBNAME))
+    system('%s/bin/psql -p %s -d %s -c "INSERT INTO geometry_columns (f_table_catalog, f_table_schema, f_table_name, f_geometry_column, coord_dimension, srid, type) VALUES (\'\', \'postgis\', \'uk_borders\', \'the_geom\', 2, 4269, \'MULTIPOLYGON\')"' % (PGROOT, port, PGDBNAME))
+    system('%s/bin/psql -p %s -d %s -c "CREATE TABLE postgis.us_borders AS SELECT * FROM postgis.world_borders WHERE fips_cntry = \'US\'"' % (PGROOT, port, PGDBNAME))
+    system('%s/bin/psql -p %s -d %s -c "INSERT INTO geometry_columns (f_table_catalog, f_table_schema, f_table_name, f_geometry_column, coord_dimension, srid, type) VALUES (\'\', \'postgis\', \'us_borders\', \'the_geom\', 2, 4269, \'MULTIPOLYGON\')"' % (PGROOT, port, PGDBNAME))
+
+    os.unlink('/tmp/wb.sql')
+
+    if port != '5432':
+        # If we're not using the default port, we need to update the
+        # test configuration and run the buildout again so that the
+        # test runner gets updated.
+        open('test-environment.cfg', 'w').write(test_env_tmpl % dict(port=port))
+        system('./bin/buildout')
+
+    
+
+def main():
+    parser = OptionParser()
+    parser.add_option('--port', dest='port', default='5432',
+                      help='PostgreSQL server port number')
+
+    options, args = parser.parse_args()
+
+    initpg(port=options.port)
+
+    print '\n\n'
+    print 'PostgreSQL test server initialized'
+    print '----------------------------------'
+    print 'The server is listening on localhost:%s' % options.port
+    print 'You can control the server process by using the ./bin/pg_ctl script' 
 
 
-# Create a database and import the world borders data set into the db
-# for testing
-system('%s/bin/shp2pgsql -I -s 4269 %s postgis.world_borders > /tmp/wb.sql' % (POSTGIS, DATASET))
-system('%s/bin/createdb --port=%s %s' % (PGROOT, PGPORT, PGDBNAME))
-system('%s/bin/createlang --port=%s plpgsql %s' % (PGROOT, PGPORT, PGDBNAME))
-system('%s/bin/psql -p %s -d %s -c "CREATE SCHEMA postgis"' % (PGROOT, PGPORT, PGDBNAME))
-system('%s/bin/psql -p %s -d %s -f %s/share/lwpostgis.sql' % (PGROOT, PGPORT, PGDBNAME, POSTGIS))
-system('%s/bin/psql -p %s -d %s -f %s/share/spatial_ref_sys.sql' % (PGROOT, PGPORT, PGDBNAME, POSTGIS))
-system('%s/bin/psql -p %s -d %s -f /tmp/wb.sql' % (PGROOT, PGPORT, PGDBNAME))
-
-system('%s/bin/psql -p %s -d %s -c "CREATE VIEW postgis.uk_borders AS SELECT * FROM postgis.world_borders WHERE fips_cntry = \'UK\'"' % (PGROOT, PGPORT, PGDBNAME))
-system('%s/bin/psql -p %s -d %s -c "INSERT INTO geometry_columns (f_table_catalog, f_table_schema, f_table_name, f_geometry_column, coord_dimension, srid, type) VALUES (\'\', \'postgis\', \'uk_borders\', \'the_geom\', 2, 4269, \'MULTIPOLYGON\')"' % (PGROOT, PGPORT, PGDBNAME))
-system('%s/bin/psql -p %s -d %s -c "CREATE TABLE postgis.us_borders AS SELECT * FROM postgis.world_borders WHERE fips_cntry = \'US\'"' % (PGROOT, PGPORT, PGDBNAME))
-system('%s/bin/psql -p %s -d %s -c "INSERT INTO geometry_columns (f_table_catalog, f_table_schema, f_table_name, f_geometry_column, coord_dimension, srid, type) VALUES (\'\', \'postgis\', \'us_borders\', \'the_geom\', 2, 4269, \'MULTIPOLYGON\')"' % (PGROOT, PGPORT, PGDBNAME))
+if __name__ == '__main__':
+    main()
 
 
+