Project

General

Profile

1 8291 aaronmk
#!/bin/bash -e
2 8273 aaronmk
3
if false; then #### run script template:
4 8291 aaronmk
#!/bin/bash -e
5 8273 aaronmk
. "$(dirname "${BASH_SOURCE[0]}")"/path/to/table.run
6 8161 aaronmk
7 9039 aaronmk
func_override import__table_run
8 9074 aaronmk
import()
9 8273 aaronmk
{
10 12968 aaronmk
	begin_target
11 8273 aaronmk
	before_import_cmds
12 12967 aaronmk
	with_rm import__table_run "$@"
13 8273 aaronmk
	after_import_cmds
14
}
15
fi ####
16 8213 aaronmk
17 8296 aaronmk
. "$(dirname "${BASH_SOURCE[0]}")"/import.run
18 9854 aaronmk
.rel subdir.run
19
.rel ../sh/db_make.sh
20 8273 aaronmk
21 8705 aaronmk
if self_not_included; then
22
23 11602 aaronmk
##### utils
24
25 9585 aaronmk
: "${table=$subdir}"; export table
26
27
table_make() { subdir_make "$@"; }
28
29 9935 aaronmk
install_log_rel=logs/install.log.sql
30
31 9934 aaronmk
table_make_install()
32
{
33 11769 aaronmk
	echo_func
34 12864 aaronmk
	if ! remaking && pg_table_exists; then skip_table; return 0; fi
35 11769 aaronmk
	local install_log="${install_log-$top_dir/$install_log_rel}"
36 9934 aaronmk
	local verbosity_min= # install logs require default verbosity
37 11770 aaronmk
	if ! remaking && test -e "$install_log"; then local_export noclobber=1; fi
38
	table_make "$@"
39 9934 aaronmk
}
40
41 10248 aaronmk
postprocess_sql="$top_dir"/postprocess.sql
42
43 10249 aaronmk
is_view="$(test -e "$postprocess_sql" \
44
&& grep -F force_update_view "$postprocess_sql" >/dev/null; exit2bool)"
45
46 10368 aaronmk
map_table="\"~$table.map\"" # sort after other tables to avoid clutter
47
48 11602 aaronmk
##### targets
49
50
load_data()
51
{
52 12779 aaronmk
	begin_target
53 11602 aaronmk
	if remaking; then in_top_dir rm -f header.csv; fi # remake it
54 12878 aaronmk
	make --directory="$top_dir/.." schema # works w/ old-style srcs w/o ../run
55
	#"$top_dir/../run" schema/make # uncomment once everything is new-style
56 11602 aaronmk
	is_view="$is_view" table_make_install ${_remake:+re}install # just the table
57
}
58
59
trim_table()
60
{
61 12779 aaronmk
	begin_target
62 11602 aaronmk
	if test "$is_view"; then return 0; fi # do not modify view columns
63
	psql <<EOF
64 12847 aaronmk
SELECT util.trim('"$table"', '$map_table');
65 11602 aaronmk
EOF
66
}
67
68 10369 aaronmk
reset_col_names()
69
{
70 12779 aaronmk
	begin_target
71 10369 aaronmk
	if test "$is_view"; then return 0; fi # do not rename view columns
72
	psql <<EOF
73 12847 aaronmk
SELECT util.reset_col_names('"$table"', '$map_table');
74 10369 aaronmk
EOF
75
}
76
77 9074 aaronmk
map_table()
78 10210 aaronmk
# note that collisions may prevent all renames from being made at once.
79 10211 aaronmk
# if this is the case, run map_table repeatedly until no more renames are made:
80
# $ while true; do .../run map_table; done
81 10210 aaronmk
# collisions may result if the staging table gets messed up (e.g. due to missing
82
# input columns in map.csv).
83 8161 aaronmk
{
84 12779 aaronmk
	begin_target
85 10250 aaronmk
	if test "$is_view"; then return 0; fi # do not rename view columns
86
87 10370 aaronmk
	if remaking; then reset_col_names; fi
88 9386 aaronmk
	table_make map.csv
89 8214 aaronmk
	psql <<EOF
90 10156 aaronmk
SELECT util.reset_map_table('$map_table');
91
ALTER TABLE $map_table DISABLE TRIGGER map_filter_insert;
92
\copy $map_table FROM 'map.csv' CSV HEADER;
93 10366 aaronmk
SELECT util.set_col_names_with_metadata('"$table"', '$map_table');
94 8161 aaronmk
EOF
95
}
96 8198 aaronmk
97 12779 aaronmk
test_() { begin_target; piped_cmd yes|table_make test; }
98 11913 aaronmk
99 12691 aaronmk
mappings()
100 8245 aaronmk
{
101 12779 aaronmk
	begin_target
102 12804 aaronmk
	public_schema_exists || return 0
103 11000 aaronmk
	if remaking; then in_top_dir rm -f map.csv; fi # remake it
104 11913 aaronmk
	test_
105 8245 aaronmk
}
106 8273 aaronmk
107 10273 aaronmk
custom_postprocess() # overridable
108 8273 aaronmk
{
109 12779 aaronmk
	begin_target
110 10248 aaronmk
	local file="$postprocess_sql"
111 12743 aaronmk
	if test -e "$file"; then
112
		table_make "$(basename "$file")" # apply renames to SQL statements
113
		psql "$@"
114
	fi
115 8273 aaronmk
}
116
117 11602 aaronmk
mk_derived()
118 12779 aaronmk
{ begin_target; "$root_dir"/schemas/VegCore/mk_derived; }
119 9392 aaronmk
120 10274 aaronmk
postprocess()
121 8273 aaronmk
{
122 12779 aaronmk
	begin_target
123 10275 aaronmk
	kw_params can_test; local can_test="${can_test-1}"
124 10372 aaronmk
	if remaking; then trim_table; fi
125 8282 aaronmk
	map_table
126 10273 aaronmk
	custom_postprocess
127 10171 aaronmk
	mk_derived
128 12691 aaronmk
	if test "$can_test"; then mappings; fi
129 8273 aaronmk
}
130 8705 aaronmk
131 10274 aaronmk
import()
132
{
133 12779 aaronmk
	begin_target
134 12967 aaronmk
	with_rm load_data
135 10274 aaronmk
	postprocess
136
}
137
138 8705 aaronmk
fi