Project

General

Profile

1
selfDir_uZPPqC := $(dir $(lastword $(MAKEFILE_LIST)))
2
root := $(selfDir_uZPPqC)..
3
include $(root)/lib/common.Makefile
4

    
5

    
6
##### Configuration
7

    
8
# Command line
9
log ?= $(if $(test),,1)
10
profile ?=
11
quiet ?=
12
reverify ?= 1
13
use_staged ?= $(by_col)
14

    
15
# Makefile
16
exts ?= csv tsv txt xml
17
test_n ?= 2
18

    
19
##### Vars/functions
20

    
21
# Paths
22
datasrc := $(notdir $(realpath .))
23
bin := $(root)/bin
24
mappings := $(root)/mappings
25

    
26
# Make
27
SHELL := /bin/bash
28
selfMake = $(MAKE) --makefile=../input.Makefile
29
subMake = $(MAKE) $(@:$(root)/%=%) --directory=$(root)
30
+_ = $(+:_%=)
31
addBeforeExt = $(basename $(2))$(1)$(suffix $(2))
32

    
33
# 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
# System
41
date = $(shell date +"%Y-%m-%d-%H-%M-%S")
42

    
43
# Terminal
44
termCols := $(shell tput cols)
45
esc := '['
46
reset := $(esc)'0m'
47
emph := $(esc)'7m '
48
endEmph := ' '$(reset)
49

    
50
# Commands
51
MKDIR = mkdir -p
52
mkdir = $(MKDIR) $(@D)
53
CP = cp -p
54
diff = diff --unified=2
55
diffIgnoreSpace = $(diff) --ignore-space-change
56
diffVerbose = $(if $(verbose),diff --side-by-side --left-column\
57
--width=$(termCols),$(diff))
58

    
59
# BIEN commands
60
sortFilenames = $(shell $(bin)/sort_filenames $(1))
61
selfMap = $(bin)/cols 0 0
62
psqlOpts := --set ON_ERROR_STOP=1 --quiet
63
psqlAsBien := $(bin)/psql_vegbien $(psqlOpts)
64

    
65
# SVN
66
addDir = $(if $(wildcard $(1)/),svn add --depth=empty $(1),svn mkdir $(1))
67
setSvnIgnore = svn propset svn:ignore $(2) $(1)
68
define addDirWithIgnore
69
$(addDir)
70
$(setSvnIgnore)
71
endef
72

    
73
##### General targets
74

    
75
all: _always maps ;
76

    
77
clean: _always
78
	$(RM) $(all)
79

    
80
remake: _always clean
81
	+$(selfMake)
82
# re-run make so that cache of existing files is reset
83

    
84
make_script = ./$< >$@
85

    
86
%/: % _always ;
87

    
88
# Must come before `%: %.make` to override it
89
%: %.make _always
90
	(set -x; $(make_script)) 2>>$<.log.sql
91
.PRECIOUS: % # save partial outputs of aborted src make scripts
92

    
93
%: %.make _always
94
	$(make_script)
95

    
96
##### SVN
97

    
98
add: _always
99
	$(call setSvnIgnore,.,'*')
100
	$(call addDirWithIgnore,verify,'*.out')
101
	$(call addFile,import_order.txt)
102

    
103
# Adds a new table subdir
104
%/add: _always
105
	$(call addDirWithIgnore,$*,'*')
106
	$(call addDirWithIgnore,$*/logs,$$'*.log.sql\n*.trace')
107

    
108
##### Existing maps discovery
109

    
110
sortFile := import_order.txt
111

    
112
tables := $(if $(wildcard $(sortFile)),$(shell cat $(sortFile)))
113
    # $(shell) replaces "\n" with " "
114
allSubdirs := $(call wildcard/,*/)
115
allTables := $(call sortFilenames,$(filter-out _% verify,$(allSubdirs:%/=%)))
116
joinedTables := $(filter-out $(tables),$(allTables))
117
allTables := $(joinedTables) $(tables)# move joined tables to beginning
118
ifeq ($(tables),)# none specified in sort file
119
tables := $(allTables)
120
endif
121

    
122
srcMaps := $(wildcard */src.csv)
123

    
124
srcMap := %/src.csv
125
viaMap := %/map.csv
126
fullViaMap := %/map.full.csv
127
directMap := %/VegBIEN.csv
128
anyMap := $(srcMap) $(viaMap) $(fullViaMap) $(directMap)
129

    
130
extsFilter := $(addprefix %.,$(exts))
131
dataOnly = $(filter $(extsFilter),$(1))
132

    
133
anyTest = $*/test.%
134
srcsOnly = $(filter-out $(anyMap) $(anyTest) %/logs,$(call dataOnly,$(1)))
135

    
136
via := Veg+
137

    
138
coreMap := $(mappings)/$(via)-VegBIEN.csv
139
coreSelfMap := $(mappings)/$(via).self.csv
140

    
141
viaMaps := $(wildcard $(tables:%=%/map.csv))
142
viaMaps += $(filter-out $(viaMaps),$(srcMaps:%/src.csv=%/map.csv))
143
viaMaps += $(filter-out $(viaMaps) $(fullViaMap),$(wildcard */map.csv))
144

    
145
autogenMaps := $(subst map.,VegBIEN.,$(viaMaps))
146
directMaps := $(autogenMaps) $(filter-out $(autogenMaps),\
147
$(wildcard */VegBIEN.csv))
148

    
149
##### Sources
150

    
151
srcs = $(call sortFilenames,$(call srcsOnly,$(wildcard $*/*)))
152
isCsv = $(if $(srcs),$(if $(filter %.xml,$(srcs)),,1))
153
    # true if $(srcs) non-empty and contains no *.xml
154
catSrcs = $(bin)/cat$(if $(isCsv),_csv) $(srcs)
155
withCatSrcs = $(catSrcs:$(bin)/%=$(bin)/with_%) --
156

    
157
# Usage: `make {--silent|-s} inputs/<datasrc>/cat` (don't echo make commands)
158
cat: $(tables:%=%/cat) _always ;
159

    
160
%/cat: _always
161
	$(catSrcs)
162

    
163
##### Staging tables installation
164

    
165
install: _always schema $(allTables:%=%/install) ;
166

    
167
uninstall: _always rm_schema ;
168
# rm_schema will also drop all staging tables
169

    
170
reinstall: _always uninstall install ;
171

    
172
schema: _always
173
	-echo 'CREATE SCHEMA "$(datasrc)";'|$(psqlAsBien)
174
# ignore errors if schema exists
175

    
176
rm_schema: _always
177
	echo 'DROP SCHEMA IF EXISTS "$(datasrc)" CASCADE;'|$(psqlAsBien)
178

    
179
# For staging tables which are derived by joining together other staging tables.
180
%/install %/header.csv: %/create.sql _always
181
	(echo 'SET search_path TO "$(datasrc)";'; echo 'CREATE TABLE "$*" AS';\
182
cat $<; echo ';'; \
183
echo 'ALTER TABLE "$*" ADD COLUMN row_num serial NOT NULL PRIMARY KEY;'; \
184
echo 'VACUUM ANALYZE "$*";')|$(psqlAsBien) --echo-all
185
	(prefix=; . $(bin)/vegbien_dest; env schema=$(datasrc) table=$* \
186
errors_table_only=1 $(bin)/csv2db)
187
	echo 'SELECT * FROM "$(datasrc)"."$*" LIMIT 0;'|$(psqlAsBien) \
188
--no-align --field-separator=, --pset=footer=off >$*/header.csv
189

    
190
# The joined tables must be suffixed with ".src" to prevent the creation of a
191
# row_num column, which collides during joins.
192
isJoinedTable = $(filter %.src,$*)
193
hasRowNum = $(if $(isJoinedTable),,1)
194

    
195
%/install: _always
196
	$(if $(isCsv),$(import_install_))
197
import_install_ = (prefix=; . $(bin)/vegbien_dest; "time" nice -n +5\
198
env schema=$(datasrc) table=$* has_row_num=$(hasRowNum) $(bin)/csv2db\
199
$(catSrcs) $(if $(log),$(if $(quiet),&>,2>&1|tee )$*/logs/install.log.sql))
200

    
201
##### Maps building
202

    
203
# Maps to (try to) build are added to one of these
204
maps :=
205
createOnlyMaps :=
206

    
207
srcRoot = $(mappings)/root.sh
208
mkSrcMap = $(catSrcs)|(. $(srcRoot); env datasrc=$(datasrc) $(bin)/src_map >$@)
209

    
210
# Src maps cleanup
211
ifneq ($(filter %/.src.csv.last_cleanup,$(MAKECMDGOALS)),)
212
%/.src.csv.last_cleanup: %/src.csv
213
	$(bin)/in_place $< $(bin)/cols +
214
	touch $@
215
else
216
# Autogen src maps with known table names
217
%/src.csv: _always
218
	$(if $(wildcard $@),,$(if $(isCsv),$(mkSrcMap)))
219
	+$(if $(isCsv),$(selfMake) $(@:%/src.csv=%/.src.csv.last_cleanup))
220
# only build if doesn't exist
221
# only build if CSV srcs exist for that table name
222
endif
223

    
224
createOnlyMaps += $(tables:%=%/src.csv)
225

    
226
# Must come before $(root)/% to override it
227
$(coreSelfMap): _always
228
	-+$(subMake)
229
# ignore errors if $(coreSelfMap) does not exist
230

    
231
# Via maps cleanup
232
ifneq ($(filter %/.map.csv.last_cleanup,$(MAKECMDGOALS)),)
233
%/.map.csv.last_cleanup: %/map.csv $(coreMap)
234
	$(bin)/in_place $< env ignore=1 $(bin)/subtract $(word 2,$+) 0 1 2
235
	touch $@
236
# Include comment column so commented mappings are never removed
237
else
238
%/map.csv: %/src.csv _always
239
	$(if $(wildcard $@),,$(if $(wildcard $<),$(CP) $< $@))
240
	+$(selfMake) $(@:%/map.csv=%/.map.csv.last_cleanup)
241
endif
242

    
243
srcMap* = $*/src.csv
244
joinSrcMap = $(if $(hasSrc),$(bin)/in_place $@ $(bin)/intersect $(srcMap*) 0)
245
hasSrc = $(shell test -s $(word 3,$+) && echo t)
246

    
247
makeFullCsv = env ignore=1 $(bin)/union <$(wordlist 1,2,$+)|$(bin)/sort_map >$@
248

    
249
%/map.full.csv: %/map.csv $(coreSelfMap)
250
	$(makeFullCsv)
251
	$(joinSrcMap)
252
maps += $(patsubst %/map.csv,%/map.full.csv,$(viaMaps))
253

    
254
%/VegBIEN.csv: %/map.full.csv $(coreMap)
255
	$(bin)/join <$+|$(bin)/sort_map >$@
256
maps += $(autogenMaps)
257

    
258
maps: $(createOnlyMaps) $(maps) _always ;
259

    
260
all += $(maps)
261

    
262
##### Maps validation
263

    
264
missingMappingsCmd = +$(selfMake) remake 2>&1\
265
|$(SED) -n 's/^.*No $(*2Space) mapping for (.*)$$/\1/p'|$(SED) 's/\/_.*//'\
266
$(if $(filter non-empty_join,$*),|$(bin)/ucase_first 0)|sort|uniq
267

    
268
include $(root)/lib/mappings.Makefile
269

    
270
##### External dependencies
271

    
272
$(root)/%: _always
273
	+$(subMake)
274
.PRECIOUS: $(root)/% # let ext. dir's Makefile decide whether to delete on error
275

    
276
##### Mapping
277

    
278
+maps = $(filter %/map.csv %/VegBIEN.csv $(mappings)/%,$(+_))
279
map = $(if $(srcs),$(withCatSrcs) $(root)/map $(+maps),\
280
$(shell echo Warning: No input file $*/*.{$(exts)} >&2)false)
281
# need false to run some command, when prefixed by `env ...` below
282
map2db = $(if $(use_staged),env in_database=vegbien in_schema=$(datasrc)\
283
in_table=$*) env out_database=vegbien $(map)
284

    
285
##### Import to VegBIEN
286

    
287
profileTest = $(if $(profile),$(if $(test),1))
288
profileOnly = -env profile_to=/dev/fd/3 $(map2db) 3>&1 1>&2|\
289
$(bin)/profile_stats /dev/fd/0
290

    
291
log_ = $*/logs/$(if $(n),n=$(n).,)$(date).log.sql
292
trace = $(log_:.log.sql=.trace)
293
import = -$(if $(profileTest),$(profileOnly),(set -x; "time" env commit=1\
294
$(if $(profile),profile_to=$(trace)) $(map2db)) $(if $(log),\
295
$(if $(n),,&>$(log_)))$(if $(log),$(if $(n), 2>&1|tee -a $(log_))))
296
# don't abort on import errors, which often relate to invalid input data
297

    
298
import: $(tables:%=%/import) _always ;
299

    
300
%/import: %/VegBIEN.csv _always
301
	$(import)
302
# default:
303
%/import: _always ;
304

    
305
##### Log files from import
306

    
307
logs := $(wildcard */logs/*.log.sql */logs/*.trace)
308

    
309
rm_logs: _always
310
	$(RM) $(logs)
311

    
312
##### Verification of import
313

    
314
verify: $(tables:%=%/verify) _always ;
315

    
316
%/verify: verify/%.ref verify/%.out _always
317
	-$(diffVerbose) $(+_)
318
# don't abort on verification errors, which are expected during development
319
# default:
320
%/verify: verify/%.out _always
321
	$(if $(shell test -e $< && echo t),cat $<)
322
# don't run if verify/%.out's default do-nothing action was used
323
# can't use $(wildcard) because it won't recheck file after verify/%.out is run
324

    
325
verify = $(if $(reverify),"time" $(psqlAsBien) --set=datasource="'$(datasrc)'"\
326
--no-align --field-separator=$$'\t' --pset=footer=off --pset=null=NULL <$< >$@)
327

    
328
verify/%.out: $(mappings)/verify.%.sql _always
329
	$(verify)
330
# default:
331
verify/%.out: _always ;
332

    
333
all += $(wildcard verify/*.out)
334

    
335
##### Editing import
336

    
337
rotate: _always
338
	echo "UPDATE party SET organizationname = organizationname||'.$(date)'\
339
WHERE organizationname = '$(datasrc)';"|$(psqlAsBien)
340

    
341
rm: _always
342
	echo "DELETE FROM party WHERE organizationname = '$(datasrc)';"|\
343
$(psqlAsBien)
344

    
345
##### Testing
346

    
347
testRefOutput = $(subst .2-step,,$(subst .staging,,$(1)))
348
testRef = $(testRefOutput).ref
349
hasOwnRef = $(filter $@,$(call testRefOutput,$@))
350
# filter returns non-empty if they are equal
351

    
352
# `rm $@`: Remove outputs of successful tests to reduce clutter
353
# `$(foreach use_staged...)`: Run with use_staged=1
354
define runTest
355
@echo "Testing $(abspath $@)..."
356
>$@ env test=1 n=$(test_n) $(1) $(foreach use_staged,1,$(map2db))
357
@(set -x; $(diffIgnoreSpace) $(call testRef,$@) $@) 2>&1 && rm $@ || { e=$$?;\
358
$(if $(wildcard $(call testRef,$@)),,cat $@;)\
359
$(if $(hasOwnRef),\
360
{\
361
read -p $(emph)'Accept new test output? (y/n)'$(endEmph) REPLY;\
362
if test "$$REPLY" = y; then\
363
(set -x; $(MAKE) $@-ok --directory=$(realpath .) --makefile=../input.Makefile);\
364
exit 0;\
365
fi;\
366
};,\
367
echo $(emph)"Note: The preceding failed test is compared to another test's\
368
output"$(endEmph);\
369
echo $(emph)"When it fails, this always indicates a bug"$(endEmph);\
370
)\
371
exit $$e;}
372
endef
373

    
374
test2Db = $(runTest)
375

    
376
tests :=
377

    
378
# Requires staging tables. To create them, run `make inputs/<datasrc>/install`.
379
# Non-flat-file inputs fall back to mimicking a successful test
380
%/test.xml: %/VegBIEN.csv _always
381
	$(if $(isCsv),$(test2Db))
382
tests += %/test.xml
383

    
384
testOutputs := $(foreach test,$(tests),$(tables:%=$(test)))
385

    
386
.PRECIOUS: $(testOutputs) # save outputs of failed tests so they can be accepted
387

    
388
test: _always $(testOutputs) ;
389

    
390
all += $(wildcard %/test*.xml)
391

    
392
# Accepts a test output: make <test_output_path>-ok
393
%-ok: _always
394
	$(CP) $* $(call testRef,$*)
395

    
396
accept-all: _always
397
	+yes|$(selfMake) test
398

    
399
##### Documentation
400

    
401
steps = $(selfMake) -s import-$* test=1 by_col=1 verbosity=2 n=100\
402
2>&1|$(bin)/debug2redmine >$@
403

    
404
%/logs/steps.by_col.log.sql: _always
405
	+$(steps)
(4-4/4)