HitchBuildPg
A small, self contained python library for building postgres database locally that, on first run will:
-
Downloads and compiles a specified version of postgres.
-
Builds a database from a series of SQL commands and by restoring a database dump file.
-
Takes a snapshot of the newly built database files by taking a copy of the folder.
On subsequent builds, it will skip the long, expensive steps of downloading, compiling postgres and restoring a database from a .sql file and just overwrite the data files.
import hitchbuildpg
postgres_version = "10.10"
pgapp_dir = "{}/postgres-{}".format(share, postgres_version)
pgapp = hitchbuildpg.PostgresApp(pgapp_dir, postgres_version)
class DataBuild(hitchbuildpg.DataBuild):
def run(self):
self.run_sql_as_root("create user myuser with password 'mypassword';")
self.run_sql_as_root("create database mydb with owner myuser;")
self.load_database_dump(
database="mydb",
username="myuser",
password="mypassword",
filename="dump.sql"
)
pgdata = hitchbuildpg.PostgresDatafiles(
"./myappdata",
pgapp,
DataBuild(),
)
pgdata.ensure_built()
db_service = pgdata.server()
db_service.start()
psql = db_service.psql(
"-U", "myuser", "-p", "15432", "-d", "mydb",
).with_env(PG_PASSWORD="mypassword")
psql("-c", "select name from cities where location = 'GB';").run()
# Prints output of SQL statement
psql("-c", "delete from cities where location = 'GB';").run()
db_service.stop()
