dougal-software/sbin/upgrade_database.py

#!/usr/bin/python3

import sys
import os
# Because Python 🙄
bin_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../bin"))
sys.path.insert(1, bin_path)
import configuration

print(
"""
\u001b[31;4mYou are about to upgrade the database to the latest schema in /etc/db/schema-template.sql.\u001b[0m

\u001b[33mThis will cause all existing surveys to be wiped out and re-created.
Data which is captured by the database itself, e.g., event data will be exported¹
and re-imported. The rest should be brought back in at the next bin/runner.sh
run.

¹ As of this version, not all data will be exported. The following information
will be lost:

 - Sequence NTBA status
 - Shot NTBA status
 - Shot NTBP status
 - Raw sequence remarks
 - Raw shot remarks
 - Preplot point remarks
 \u001b[33;1m- All real-time data will also be lost\u001b[33m

\u001b[0m

If this is what you want to do, press ENTER, else press Ctrl+C to cancel.
""")

input()

dbname=os.environ["PGDATABASE"] if "PGDATABASE" in os.environ else "dougal"
dbuser=os.environ["PGUSER"] if "PGUSER" in os.environ else "postgres"
dbtemplate=os.path.abspath(os.path.join(bin_path, "..", "etc", "db", "database-template.sql"))

system_exports = os.path.join(bin_path, "system_exports.py")
create_survey = os.path.join(bin_path, "create_survey.sh")
runner = "RUNNER_NOEXPORT=true RUNNER_IMPORT=true " + os.path.join(bin_path, "runner.sh")

print(system_exports)
#os.system(system_exports)

cmd = f"psql -U {dbuser} -d template1 -c 'DROP DATABASE {dbname};'"
print(cmd)
res = os.system(cmd)
if res != 0:
	print("Cannot proceed. Upgrade aborted.")
	exit(res)

cmd = f"psql -U {dbuser} -d template1 <{dbtemplate}"
print(cmd)
os.system(cmd)

#for survey in configuration.surveys():
	#schema = survey["schema"]
	#cmd = f"psql dougal -c 'DROP schema {schema} CASCADE;'"
	#print(cmd)

for survey in configuration.surveys():
	id = survey["id"].lower()
	name = survey["name"]
	epsg = survey["epsg"]
	schema = survey["schema"]

	cmd = f"{create_survey} \"{id}\" \"{name}\" {epsg}"
	print(cmd)
	os.system(cmd)

print(runner)
os.system(runner)

print("Done")