| 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 |
8881
|
aaronmk
|
echo_func
|
| 11 |
8273
|
aaronmk
|
before_import_cmds
|
| 12 |
9681
|
aaronmk
|
self_make 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 |
12797
|
aaronmk
|
prep_try; if ! remaking && pg_table_exists; then rethrow; return 0; fi
|
| 35 |
|
|
rethrow
|
| 36 |
11769
|
aaronmk
|
local install_log="${install_log-$top_dir/$install_log_rel}"
|
| 37 |
9934
|
aaronmk
|
local verbosity_min= # install logs require default verbosity
|
| 38 |
11770
|
aaronmk
|
if ! remaking && test -e "$install_log"; then local_export noclobber=1; fi
|
| 39 |
|
|
table_make "$@"
|
| 40 |
9934
|
aaronmk
|
}
|
| 41 |
|
|
|
| 42 |
9936
|
aaronmk
|
datasrc_make_install()
|
| 43 |
|
|
{
|
| 44 |
|
|
echo_func
|
| 45 |
|
|
subdir=. install_log="$top_dir/../$install_log_rel" table_make_install "$@"
|
| 46 |
|
|
}
|
| 47 |
|
|
|
| 48 |
10248
|
aaronmk
|
postprocess_sql="$top_dir"/postprocess.sql
|
| 49 |
|
|
|
| 50 |
10249
|
aaronmk
|
is_view="$(test -e "$postprocess_sql" \
|
| 51 |
|
|
&& grep -F force_update_view "$postprocess_sql" >/dev/null; exit2bool)"
|
| 52 |
|
|
|
| 53 |
10368
|
aaronmk
|
map_table="\"~$table.map\"" # sort after other tables to avoid clutter
|
| 54 |
|
|
|
| 55 |
11602
|
aaronmk
|
##### targets
|
| 56 |
|
|
|
| 57 |
|
|
load_data()
|
| 58 |
|
|
{
|
| 59 |
12779
|
aaronmk
|
begin_target
|
| 60 |
11602
|
aaronmk
|
if remaking; then in_top_dir rm -f header.csv; fi # remake it
|
| 61 |
|
|
datasrc_make_install schema
|
| 62 |
|
|
is_view="$is_view" table_make_install ${_remake:+re}install # just the table
|
| 63 |
|
|
}
|
| 64 |
|
|
|
| 65 |
|
|
trim_table()
|
| 66 |
|
|
{
|
| 67 |
12779
|
aaronmk
|
begin_target
|
| 68 |
11602
|
aaronmk
|
if test "$is_view"; then return 0; fi # do not modify view columns
|
| 69 |
|
|
psql <<EOF
|
| 70 |
|
|
SELECT util.trim('"$table"', '$map_table')
|
| 71 |
|
|
EOF
|
| 72 |
|
|
}
|
| 73 |
|
|
|
| 74 |
10369
|
aaronmk
|
reset_col_names()
|
| 75 |
|
|
{
|
| 76 |
12779
|
aaronmk
|
begin_target
|
| 77 |
10369
|
aaronmk
|
if test "$is_view"; then return 0; fi # do not rename view columns
|
| 78 |
|
|
psql <<EOF
|
| 79 |
|
|
SELECT util.reset_col_names('"$table"', '$map_table')
|
| 80 |
|
|
EOF
|
| 81 |
|
|
}
|
| 82 |
|
|
|
| 83 |
9074
|
aaronmk
|
map_table()
|
| 84 |
10210
|
aaronmk
|
# note that collisions may prevent all renames from being made at once.
|
| 85 |
10211
|
aaronmk
|
# if this is the case, run map_table repeatedly until no more renames are made:
|
| 86 |
|
|
# $ while true; do .../run map_table; done
|
| 87 |
10210
|
aaronmk
|
# collisions may result if the staging table gets messed up (e.g. due to missing
|
| 88 |
|
|
# input columns in map.csv).
|
| 89 |
8161
|
aaronmk
|
{
|
| 90 |
12779
|
aaronmk
|
begin_target
|
| 91 |
10250
|
aaronmk
|
if test "$is_view"; then return 0; fi # do not rename view columns
|
| 92 |
|
|
|
| 93 |
10370
|
aaronmk
|
if remaking; then reset_col_names; fi
|
| 94 |
9386
|
aaronmk
|
table_make map.csv
|
| 95 |
8214
|
aaronmk
|
psql <<EOF
|
| 96 |
10156
|
aaronmk
|
SELECT util.reset_map_table('$map_table');
|
| 97 |
|
|
ALTER TABLE $map_table DISABLE TRIGGER map_filter_insert;
|
| 98 |
|
|
\copy $map_table FROM 'map.csv' CSV HEADER;
|
| 99 |
10366
|
aaronmk
|
SELECT util.set_col_names_with_metadata('"$table"', '$map_table');
|
| 100 |
8161
|
aaronmk
|
EOF
|
| 101 |
|
|
}
|
| 102 |
8198
|
aaronmk
|
|
| 103 |
12779
|
aaronmk
|
test_() { begin_target; piped_cmd yes|table_make test; }
|
| 104 |
11913
|
aaronmk
|
|
| 105 |
12691
|
aaronmk
|
mappings()
|
| 106 |
8245
|
aaronmk
|
{
|
| 107 |
12779
|
aaronmk
|
begin_target
|
| 108 |
12773
|
aaronmk
|
prep_try; public_schema_exists || { rethrow; return 0; }; rethrow
|
| 109 |
11000
|
aaronmk
|
if remaking; then in_top_dir rm -f map.csv; fi # remake it
|
| 110 |
11913
|
aaronmk
|
test_
|
| 111 |
8245
|
aaronmk
|
}
|
| 112 |
8273
|
aaronmk
|
|
| 113 |
10273
|
aaronmk
|
custom_postprocess() # overridable
|
| 114 |
8273
|
aaronmk
|
{
|
| 115 |
12779
|
aaronmk
|
begin_target
|
| 116 |
10248
|
aaronmk
|
local file="$postprocess_sql"
|
| 117 |
12743
|
aaronmk
|
if test -e "$file"; then
|
| 118 |
|
|
table_make "$(basename "$file")" # apply renames to SQL statements
|
| 119 |
|
|
psql "$@"
|
| 120 |
|
|
fi
|
| 121 |
8273
|
aaronmk
|
}
|
| 122 |
|
|
|
| 123 |
11602
|
aaronmk
|
mk_derived()
|
| 124 |
12779
|
aaronmk
|
{ begin_target; "$root_dir"/schemas/VegCore/mk_derived; }
|
| 125 |
9392
|
aaronmk
|
|
| 126 |
10274
|
aaronmk
|
postprocess()
|
| 127 |
8273
|
aaronmk
|
{
|
| 128 |
12779
|
aaronmk
|
begin_target
|
| 129 |
10275
|
aaronmk
|
kw_params can_test; local can_test="${can_test-1}"
|
| 130 |
10372
|
aaronmk
|
if remaking; then trim_table; fi
|
| 131 |
8282
|
aaronmk
|
map_table
|
| 132 |
10273
|
aaronmk
|
custom_postprocess
|
| 133 |
10171
|
aaronmk
|
mk_derived
|
| 134 |
12691
|
aaronmk
|
if test "$can_test"; then mappings; fi
|
| 135 |
8273
|
aaronmk
|
}
|
| 136 |
8705
|
aaronmk
|
|
| 137 |
10274
|
aaronmk
|
import()
|
| 138 |
|
|
{
|
| 139 |
12779
|
aaronmk
|
begin_target
|
| 140 |
10274
|
aaronmk
|
self_make load_data
|
| 141 |
|
|
postprocess
|
| 142 |
|
|
}
|
| 143 |
|
|
|
| 144 |
8705
|
aaronmk
|
fi
|
