Project

General

Profile

1 3761 aaronmk
selfDir_uZPPqC := $(dir $(lastword $(MAKEFILE_LIST)))
2
root := $(selfDir_uZPPqC)..
3
include $(root)/lib/common.Makefile
4
5
6 783 aaronmk
##### Configuration
7
8 1921 aaronmk
# Command line
9 1190 aaronmk
log ?= $(if $(test),,1)
10 1257 aaronmk
profile ?=
11 1983 aaronmk
quiet ?=
12 1620 aaronmk
reverify ?= 1
13 1990 aaronmk
use_staged ?= $(by_col)
14 1921 aaronmk
15
# Makefile
16 1386 aaronmk
exts ?= csv tsv txt xml
17 783 aaronmk
test_n ?= 2
18
19
##### Vars/functions
20
21 1821 aaronmk
# Paths
22
datasrc := $(notdir $(realpath .))
23
bin := $(root)/bin
24
mappings := $(root)/mappings
25
26 368 aaronmk
# Make
27 640 aaronmk
SHELL := /bin/bash
28 1509 aaronmk
selfMake = $(MAKE) --makefile=../input.Makefile
29 1821 aaronmk
subMake = $(MAKE) $(@:$(root)/%=%) --directory=$(root)
30 404 aaronmk
+_ = $(+:_%=)
31 368 aaronmk
addBeforeExt = $(basename $(2))$(1)$(suffix $(2))
32
33 1742 aaronmk
# OS
34
os := $(shell uname)
35
forOs = $(patsubst %,%-$(filter Linux Darwin,$(os)),$(1))
36
37
# Formatting
38
SED = sed -$(if $(filter Darwin,$(os)),E,r)
39
40 790 aaronmk
# System
41
date = $(shell date +"%Y-%m-%d-%H-%M-%S")
42
43 640 aaronmk
# Terminal
44 1184 aaronmk
termCols := $(shell tput cols)
45 640 aaronmk
esc := '['
46
reset := $(esc)'0m'
47
emph := $(esc)'7m '
48
endEmph := ' '$(reset)
49
50 368 aaronmk
# Commands
51 1245 aaronmk
MKDIR = mkdir -p
52
mkdir = $(MKDIR) $(@D)
53 395 aaronmk
CP = cp -p
54 1184 aaronmk
diff = diff --unified=2
55 3721 aaronmk
diffIgnoreSpace = $(diff) --ignore-space-change
56 1184 aaronmk
diffVerbose = $(if $(verbose),diff --side-by-side --left-column\
57
--width=$(termCols),$(diff))
58 368 aaronmk
59 1918 aaronmk
# BIEN commands
60 4137 aaronmk
sortFilenames = $(shell $(bin)/sort_filenames $(1))
61 1524 aaronmk
selfMap = $(bin)/cols 0 0
62 1081 aaronmk
psqlOpts := --set ON_ERROR_STOP=1 --quiet
63 1289 aaronmk
psqlAsBien := $(bin)/psql_vegbien $(psqlOpts)
64 4408 aaronmk
# Usage: ($(inDatasrc); cat $(file))|$(psqlCmd)
65
inDatasrc = echo 'SET search_path TO "$(datasrc)";'
66 353 aaronmk
67 1594 aaronmk
# SVN
68
addDir = $(if $(wildcard $(1)/),svn add --depth=empty $(1),svn mkdir $(1))
69 1771 aaronmk
setSvnIgnore = svn propset svn:ignore $(2) $(1)
70 1594 aaronmk
define addDirWithIgnore
71
$(addDir)
72 1771 aaronmk
$(setSvnIgnore)
73 1594 aaronmk
endef
74
75 783 aaronmk
##### General targets
76 250 aaronmk
77 418 aaronmk
all: _always maps ;
78 247 aaronmk
79 383 aaronmk
clean: _always
80 256 aaronmk
	$(RM) $(all)
81
82 1742 aaronmk
remake: _always clean
83
	+$(selfMake)
84
# re-run make so that cache of existing files is reset
85
86 1627 aaronmk
make_script = ./$< >$@
87 368 aaronmk
88 1604 aaronmk
%/: % _always ;
89
90 4379 aaronmk
%: %.make
91 4411 aaronmk
	$(if $(wildcard $@),,(set -x; "time" $(make_script)) 2>>$<.log)
92
# Only remake if doesn't exist. This prevents unintentional remaking when the
93
# make script is newly checked out from svn (which sets the mod time to now) but
94
# the output is synced externally.
95
# Can't remove prereq to do this, because it determines when the rule applies.
96 4182 aaronmk
.PRECIOUS: % # save partial outputs of aborted src make scripts
97 1627 aaronmk
98 783 aaronmk
##### SVN
99 250 aaronmk
100 4221 aaronmk
add: _always
101 4182 aaronmk
	$(call setSvnIgnore,.,'*')
102 4124 aaronmk
	$(call addDirWithIgnore,verify,'*.out')
103 4223 aaronmk
	$(call addFile,import_order.txt)
104 766 aaronmk
105 4217 aaronmk
# Adds a new table subdir
106
%/add: _always
107
	$(call addDirWithIgnore,$*,'*')
108
	$(call addDirWithIgnore,$*/logs,$$'*.log.sql\n*.trace')
109
110 1969 aaronmk
##### Existing maps discovery
111 1955 aaronmk
112 4182 aaronmk
sortFile := import_order.txt
113 4181 aaronmk
114 4253 aaronmk
tables := $(if $(wildcard $(sortFile)),$(shell cat $(sortFile)))
115
    # $(shell) replaces "\n" with " "
116 4182 aaronmk
allSubdirs := $(call wildcard/,*/)
117 4256 aaronmk
allTables := $(call sortFilenames,$(filter-out _% verify,$(allSubdirs:%/=%)))
118
joinedTables := $(filter-out $(tables),$(allTables))
119
allTables := $(joinedTables) $(tables)# move joined tables to beginning
120 4253 aaronmk
ifeq ($(tables),)# none specified in sort file
121
tables := $(allTables)
122 4181 aaronmk
endif
123 4137 aaronmk
124 4182 aaronmk
srcMaps := $(wildcard */src.csv)
125 1955 aaronmk
126 4182 aaronmk
srcMap := %/src.csv
127
viaMap := %/map.csv
128
fullViaMap := %/map.full.csv
129
directMap := %/VegBIEN.csv
130 4118 aaronmk
anyMap := $(srcMap) $(viaMap) $(fullViaMap) $(directMap)
131 3574 aaronmk
132 4214 aaronmk
extsFilter := $(addprefix %.,$(exts))
133
dataOnly = $(filter $(extsFilter),$(1))
134
135 4182 aaronmk
anyTest = $*/test.%
136 4214 aaronmk
srcsOnly = $(filter-out $(anyMap) $(anyTest) %/logs,$(call dataOnly,$(1)))
137 4120 aaronmk
138 4095 aaronmk
via := Veg+
139 783 aaronmk
140 4098 aaronmk
coreMap := $(mappings)/$(via)-VegBIEN.csv
141
coreSelfMap := $(mappings)/$(via).self.csv
142 783 aaronmk
143 4182 aaronmk
viaMaps := $(wildcard $(tables:%=%/map.csv))
144
viaMaps += $(filter-out $(viaMaps),$(srcMaps:%/src.csv=%/map.csv))
145
viaMaps += $(filter-out $(viaMaps) $(fullViaMap),$(wildcard */map.csv))
146 783 aaronmk
147 4117 aaronmk
autogenMaps := $(subst map.,VegBIEN.,$(viaMaps))
148 728 aaronmk
directMaps := $(autogenMaps) $(filter-out $(autogenMaps),\
149 4182 aaronmk
$(wildcard */VegBIEN.csv))
150 728 aaronmk
151 1969 aaronmk
##### Sources
152
153 4182 aaronmk
srcs = $(call sortFilenames,$(call srcsOnly,$(wildcard $*/*)))
154 1969 aaronmk
isCsv = $(if $(srcs),$(if $(filter %.xml,$(srcs)),,1))
155
    # true if $(srcs) non-empty and contains no *.xml
156
catSrcs = $(bin)/cat$(if $(isCsv),_csv) $(srcs)
157
withCatSrcs = $(catSrcs:$(bin)/%=$(bin)/with_%) --
158
159 1986 aaronmk
# Usage: `make {--silent|-s} inputs/<datasrc>/cat` (don't echo make commands)
160 4252 aaronmk
cat: $(tables:%=%/cat) _always ;
161 1969 aaronmk
162 4252 aaronmk
%/cat: _always
163 1969 aaronmk
	$(catSrcs)
164
165 4121 aaronmk
##### Staging tables installation
166 1921 aaronmk
167 4254 aaronmk
install: _always schema $(allTables:%=%/install) ;
168 1921 aaronmk
169 4121 aaronmk
uninstall: _always rm_schema ;
170
# rm_schema will also drop all staging tables
171 1921 aaronmk
172
reinstall: _always uninstall install ;
173
174 4121 aaronmk
schema: _always
175 1921 aaronmk
	-echo 'CREATE SCHEMA "$(datasrc)";'|$(psqlAsBien)
176
# ignore errors if schema exists
177
178 4121 aaronmk
rm_schema: _always
179 1921 aaronmk
	echo 'DROP SCHEMA IF EXISTS "$(datasrc)" CASCADE;'|$(psqlAsBien)
180
181 4260 aaronmk
# For staging tables which are derived by joining together other staging tables.
182 4263 aaronmk
%/install %/header.csv: %/create.sql _always
183 4409 aaronmk
	($(inDatasrc); echo 'CREATE TABLE "$*" AS'; cat $<; echo ';'; \
184 4262 aaronmk
echo 'ALTER TABLE "$*" ADD COLUMN row_num serial NOT NULL PRIMARY KEY;'; \
185
echo 'VACUUM ANALYZE "$*";')|$(psqlAsBien) --echo-all
186 4260 aaronmk
	(prefix=; . $(bin)/vegbien_dest; env schema=$(datasrc) table=$* \
187
errors_table_only=1 $(bin)/csv2db)
188 4263 aaronmk
	echo 'SELECT * FROM "$(datasrc)"."$*" LIMIT 0;'|$(psqlAsBien) \
189
--no-align --field-separator=, --pset=footer=off >$*/header.csv
190 4260 aaronmk
191 4259 aaronmk
# The joined tables must be suffixed with ".src" to prevent the creation of a
192
# row_num column, which collides during joins.
193
isJoinedTable = $(filter %.src,$*)
194
hasRowNum = $(if $(isJoinedTable),,1)
195
196 4252 aaronmk
%/install: _always
197 1955 aaronmk
	$(if $(isCsv),$(import_install_))
198 3476 aaronmk
import_install_ = (prefix=; . $(bin)/vegbien_dest; "time" nice -n +5\
199 4259 aaronmk
env schema=$(datasrc) table=$* has_row_num=$(hasRowNum) $(bin)/csv2db\
200
$(catSrcs) $(if $(log),$(if $(quiet),&>,2>&1|tee )$*/logs/install.log.sql))
201 1921 aaronmk
202 1777 aaronmk
##### Maps building
203
204 4345 aaronmk
# WARNING: You CANNOT make a subdir using `make inputs/<datasrc>/<subdir>/`.
205
# You must instead make the entire datasource dir: `make inputs/<datasrc>/`
206
207 1789 aaronmk
# Maps to (try to) build are added to one of these
208 1779 aaronmk
maps :=
209 1789 aaronmk
createOnlyMaps :=
210 1779 aaronmk
211 4208 aaronmk
srcRoot = $(mappings)/root.sh
212 3569 aaronmk
mkSrcMap = $(catSrcs)|(. $(srcRoot); env datasrc=$(datasrc) $(bin)/src_map >$@)
213 1779 aaronmk
214 3567 aaronmk
# Src maps cleanup
215 4182 aaronmk
ifneq ($(filter %/.src.csv.last_cleanup,$(MAKECMDGOALS)),)
216
%/.src.csv.last_cleanup: %/src.csv
217 3814 aaronmk
	$(bin)/in_place $< $(bin)/cols +
218 3567 aaronmk
	touch $@
219
else
220 1779 aaronmk
# Autogen src maps with known table names
221 4182 aaronmk
%/src.csv: _always
222 3567 aaronmk
	$(if $(wildcard $@),,$(if $(isCsv),$(mkSrcMap)))
223 4182 aaronmk
	+$(if $(isCsv),$(selfMake) $(@:%/src.csv=%/.src.csv.last_cleanup))
224 3567 aaronmk
# only build if doesn't exist
225 1779 aaronmk
# only build if CSV srcs exist for that table name
226 3567 aaronmk
endif
227
228 4182 aaronmk
createOnlyMaps += $(tables:%=%/src.csv)
229 1779 aaronmk
230 1273 aaronmk
# Must come before $(root)/% to override it
231 1524 aaronmk
$(coreSelfMap): _always
232 1273 aaronmk
	-+$(subMake)
233 1524 aaronmk
# ignore errors if $(coreSelfMap) does not exist
234 1273 aaronmk
235 3572 aaronmk
# Via maps cleanup
236 4182 aaronmk
ifneq ($(filter %/.map.csv.last_cleanup,$(MAKECMDGOALS)),)
237
%/.map.csv.last_cleanup: %/map.csv $(coreMap)
238 3572 aaronmk
	$(bin)/in_place $< env ignore=1 $(bin)/subtract $(word 2,$+) 0 1 2
239
	touch $@
240
# Include comment column so commented mappings are never removed
241
else
242 4182 aaronmk
%/map.csv: %/src.csv _always
243 3574 aaronmk
	$(if $(wildcard $@),,$(if $(wildcard $<),$(CP) $< $@))
244 4182 aaronmk
	+$(selfMake) $(@:%/map.csv=%/.map.csv.last_cleanup)
245 3572 aaronmk
endif
246
247 4182 aaronmk
srcMap* = $*/src.csv
248 4135 aaronmk
joinSrcMap = $(if $(hasSrc),$(bin)/in_place $@ $(bin)/intersect $(srcMap*) 0)
249 4133 aaronmk
hasSrc = $(shell test -s $(word 3,$+) && echo t)
250
251
makeFullCsv = env ignore=1 $(bin)/union <$(wordlist 1,2,$+)|$(bin)/sort_map >$@
252
253 4182 aaronmk
%/map.full.csv: %/map.csv $(coreSelfMap)
254 4133 aaronmk
	$(makeFullCsv)
255
	$(joinSrcMap)
256 4182 aaronmk
maps += $(patsubst %/map.csv,%/map.full.csv,$(viaMaps))
257 4133 aaronmk
258 4182 aaronmk
%/VegBIEN.csv: %/map.full.csv $(coreMap)
259 1530 aaronmk
	$(bin)/join <$+|$(bin)/sort_map >$@
260
maps += $(autogenMaps)
261
262 1789 aaronmk
maps: $(createOnlyMaps) $(maps) _always ;
263 1139 aaronmk
264
all += $(maps)
265
266 1742 aaronmk
##### Maps validation
267
268 3764 aaronmk
missingMappingsCmd = +$(selfMake) remake 2>&1\
269 3821 aaronmk
|$(SED) -n 's/^.*No $(*2Space) mapping for (.*)$$/\1/p'|$(SED) 's/\/_.*//'\
270 3764 aaronmk
$(if $(filter non-empty_join,$*),|$(bin)/ucase_first 0)|sort|uniq
271 1742 aaronmk
272 3764 aaronmk
include $(root)/lib/mappings.Makefile
273 1742 aaronmk
274 1509 aaronmk
##### External dependencies
275
276
$(root)/%: _always
277
	+$(subMake)
278 1628 aaronmk
.PRECIOUS: $(root)/% # let ext. dir's Makefile decide whether to delete on error
279 1509 aaronmk
280 783 aaronmk
##### Mapping
281 368 aaronmk
282 4182 aaronmk
+maps = $(filter %/map.csv %/VegBIEN.csv $(mappings)/%,$(+_))
283 4106 aaronmk
map = $(if $(srcs),$(withCatSrcs) $(root)/map $(+maps),\
284 4182 aaronmk
$(shell echo Warning: No input file $*/*.{$(exts)} >&2)false)
285 4106 aaronmk
# need false to run some command, when prefixed by `env ...` below
286 3387 aaronmk
map2db = $(if $(use_staged),env in_database=vegbien in_schema=$(datasrc)\
287
in_table=$*) env out_database=vegbien $(map)
288 368 aaronmk
289 783 aaronmk
##### Import to VegBIEN
290 772 aaronmk
291 1853 aaronmk
profileTest = $(if $(profile),$(if $(test),1))
292
profileOnly = -env profile_to=/dev/fd/3 $(map2db) 3>&1 1>&2|\
293
$(bin)/profile_stats /dev/fd/0
294
295 4182 aaronmk
log_ = $*/logs/$(if $(n),n=$(n).,)$(date).log.sql
296 3317 aaronmk
trace = $(log_:.log.sql=.trace)
297 1853 aaronmk
import = -$(if $(profileTest),$(profileOnly),(set -x; "time" env commit=1\
298 1190 aaronmk
$(if $(profile),profile_to=$(trace)) $(map2db)) $(if $(log),\
299 3616 aaronmk
$(if $(n),,&>$(log_)))$(if $(log),$(if $(n), 2>&1|tee -a $(log_))))
300 1088 aaronmk
# don't abort on import errors, which often relate to invalid input data
301 624 aaronmk
302 4252 aaronmk
import: $(tables:%=%/import) _always ;
303 718 aaronmk
304 4252 aaronmk
%/import: %/VegBIEN.csv _always
305 718 aaronmk
	$(import)
306 772 aaronmk
# default:
307 4252 aaronmk
%/import: _always ;
308 718 aaronmk
309 790 aaronmk
##### Log files from import
310
311 4182 aaronmk
logs := $(wildcard */logs/*.log.sql */logs/*.trace)
312 339 aaronmk
313 383 aaronmk
rm_logs: _always
314 339 aaronmk
	$(RM) $(logs)
315
316 783 aaronmk
##### Verification of import
317 264 aaronmk
318 4252 aaronmk
verify: $(tables:%=%/verify) _always ;
319 1194 aaronmk
320 4252 aaronmk
%/verify: verify/%.ref verify/%.out _always
321 1184 aaronmk
	-$(diffVerbose) $(+_)
322 1088 aaronmk
# don't abort on verification errors, which are expected during development
323 1194 aaronmk
# default:
324 4252 aaronmk
%/verify: verify/%.out _always
325 1200 aaronmk
	$(if $(shell test -e $< && echo t),cat $<)
326 1199 aaronmk
# don't run if verify/%.out's default do-nothing action was used
327 1200 aaronmk
# can't use $(wildcard) because it won't recheck file after verify/%.out is run
328 368 aaronmk
329 4401 aaronmk
psqlExport := "time" $(psqlAsBien) --no-align --field-separator=$$'\t'\
330
--pset=footer=off --pset=null=NULL
331
verify = $(if $(reverify),$(psqlExport) --set=datasource="'$(datasrc)'" <$< >$@)
332 514 aaronmk
333 1199 aaronmk
verify/%.out: $(mappings)/verify.%.sql _always
334 1192 aaronmk
	$(verify)
335 1194 aaronmk
# default:
336 1199 aaronmk
verify/%.out: _always ;
337 369 aaronmk
338 1199 aaronmk
all += $(wildcard verify/*.out)
339 1192 aaronmk
340 4402 aaronmk
%.ref: %.ref.sql
341 4409 aaronmk
	($(inDatasrc); cat $<)|$(psqlExport) >$@
342 4402 aaronmk
343 1667 aaronmk
##### Editing import
344
345 4121 aaronmk
rotate: _always
346 1668 aaronmk
	echo "UPDATE party SET organizationname = organizationname||'.$(date)'\
347 1854 aaronmk
WHERE organizationname = '$(datasrc)';"|$(psqlAsBien)
348 1667 aaronmk
349 4121 aaronmk
rm: _always
350 1854 aaronmk
	echo "DELETE FROM party WHERE organizationname = '$(datasrc)';"|\
351
$(psqlAsBien)
352 1667 aaronmk
353 783 aaronmk
##### Testing
354 368 aaronmk
355 4274 aaronmk
testRefOutput = $(subst .by_col,,$(1))
356 1986 aaronmk
testRef = $(testRefOutput).ref
357
hasOwnRef = $(filter $@,$(call testRefOutput,$@))
358
# filter returns non-empty if they are equal
359 630 aaronmk
360 4246 aaronmk
# `rm $@`: Remove outputs of successful tests to reduce clutter
361 4272 aaronmk
# `$(foreach use_staged...)`: Run with use_staged=1
362 897 aaronmk
define runTest
363 776 aaronmk
@echo "Testing $(abspath $@)..."
364 4272 aaronmk
>$@ env test=1 n=$(test_n) $(1) $(foreach use_staged,1,$(map2db))
365 4246 aaronmk
@(set -x; $(diffIgnoreSpace) $(call testRef,$@) $@) 2>&1 && rm $@ || { e=$$?;\
366 780 aaronmk
$(if $(wildcard $(call testRef,$@)),,cat $@;)\
367 777 aaronmk
$(if $(hasOwnRef),\
368 3179 aaronmk
{\
369
read -p $(emph)'Accept new test output? (y/n)'$(endEmph) REPLY;\
370
if test "$$REPLY" = y; then\
371
(set -x; $(MAKE) $@-ok --directory=$(realpath .) --makefile=../input.Makefile);\
372
exit 0;\
373
fi;\
374
};,\
375 777 aaronmk
echo $(emph)"Note: The preceding failed test is compared to another test's\
376
output"$(endEmph);\
377
echo $(emph)"When it fails, this always indicates a bug"$(endEmph);\
378
)\
379 652 aaronmk
exit $$e;}
380 628 aaronmk
endef
381
382 876 aaronmk
tests :=
383
384 4269 aaronmk
# Requires staging tables. To create them, run `make inputs/<datasrc>/install`.
385
# Non-flat-file inputs fall back to mimicking a successful test
386 4182 aaronmk
%/test.xml: %/VegBIEN.csv _always
387 4275 aaronmk
	$(if $(isCsv),$(call runTest,by_col=))
388 4182 aaronmk
tests += %/test.xml
389 627 aaronmk
390 4274 aaronmk
%/test.by_col.xml: %/VegBIEN.csv _always
391
	-$(if $(isCsv),$(call runTest,by_col=1))
392
# Don't abort tester if only by-column test fails
393 4275 aaronmk
394
# Only run column-based tests if column-based mode enabled, because these tests
395
# are much slower than the row-based tests for small numbers of rows
396
ifneq ($(by_col),)
397 4274 aaronmk
tests += %/test.by_col.xml
398 4275 aaronmk
endif
399 4274 aaronmk
400 876 aaronmk
testOutputs := $(foreach test,$(tests),$(tables:%=$(test)))
401 724 aaronmk
402 1361 aaronmk
.PRECIOUS: $(testOutputs) # save outputs of failed tests so they can be accepted
403
404 876 aaronmk
test: _always $(testOutputs) ;
405
406 4182 aaronmk
all += $(wildcard %/test*.xml)
407 876 aaronmk
408 634 aaronmk
# Accepts a test output: make <test_output_path>-ok
409 626 aaronmk
%-ok: _always
410 4383 aaronmk
	mv $* $(call testRef,$*)
411 502 aaronmk
412 4120 aaronmk
accept-all: _always
413 3579 aaronmk
	+yes|$(selfMake) test
414
415 3133 aaronmk
##### Documentation
416
417 4122 aaronmk
steps = $(selfMake) -s import-$* test=1 by_col=1 verbosity=2 n=100\
418 3385 aaronmk
2>&1|$(bin)/debug2redmine >$@
419 3188 aaronmk
420 4182 aaronmk
%/logs/steps.by_col.log.sql: _always
421 3188 aaronmk
	+$(steps)