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
# Usage: ($(inDatasrc); cat $(file))|$(psqlCmd)
65
inDatasrc = echo 'SET search_path TO "$(datasrc)";'
66

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

    
75
##### Environment
76

    
77
export PATH := $(bin):$(PATH)
78

    
79
##### General targets
80

    
81
all: _always maps ;
82

    
83
clean: _always
84
	$(RM) $(all)
85

    
86
remake: _always clean
87
	+$(selfMake)
88
# re-run make so that cache of existing files is reset
89

    
90
make_script = ./$< >$@
91

    
92
%/: % _always ;
93

    
94
%: %.make
95
	$(if $(wildcard $@),,"time" $(make_script))
96
# Only remake if doesn't exist. This prevents unintentional remaking when the
97
# make script is newly checked out from svn (which sets the mod time to now) but
98
# the output is synced externally.
99
# Can't remove prereq to do this, because it determines when the rule applies.
100
.PRECIOUS: % # save partial outputs of aborted src make scripts
101

    
102
##### SVN
103

    
104
add: _always
105
	$(call setSvnIgnore,.,'*')
106
	$(call addDirWithIgnore,logs,$$'*.log.sql\n*.trace')
107
	$(call addDirWithIgnore,verify,'*.out')
108
	$(call addFile,import_order.txt)
109

    
110
# Adds a new table subdir
111
%/add: _always
112
	$(call addDirWithIgnore,$*,'*')
113
	$(call addDirWithIgnore,$*/logs,$$'*.log.sql\n*.trace')
114

    
115
##### Existing maps discovery
116

    
117
sortFile := import_order.txt
118

    
119
tables := $(if $(wildcard $(sortFile)),$(shell cat $(sortFile)))
120
    # $(shell) replaces "\n" with " "
121
allSubdirs := $(call wildcard/,*/)
122
allTables := $(call sortFilenames,$(filter-out _% verify logs,$(allSubdirs:%/=%)))
123
joinedTables := $(filter-out $(tables),$(allTables))
124
allTables := $(joinedTables) $(tables)# move joined tables to beginning
125
ifeq ($(tables),)# none specified in sort file
126
tables := $(allTables)
127
endif
128

    
129
srcMaps := $(wildcard */src.csv)
130

    
131
srcMap := %/src.csv
132
viaMap := %/map.csv
133
fullViaMap := %/map.full.csv
134
directMap := %/VegBIEN.csv
135
anyMap := $(srcMap) $(viaMap) $(fullViaMap) $(directMap)
136

    
137
extsFilter := $(addprefix %.,$(exts))
138
dataOnly = $(filter $(extsFilter),$(1))
139

    
140
anyTest = $*/test.%
141
srcsOnly = $(filter-out $(anyMap) $(anyTest) %/logs,$(call dataOnly,$(1)))
142

    
143
via := Veg+
144

    
145
coreMap := $(mappings)/$(via)-VegBIEN.csv
146
coreSelfMap := $(mappings)/$(via).self.csv
147
vocab := $(mappings)/$(via).vocab.csv
148
dict := $(mappings)/$(via)-VegCore.csv
149

    
150
viaMaps := $(wildcard $(tables:%=%/map.csv))
151
viaMaps += $(filter-out $(viaMaps),$(srcMaps:%/src.csv=%/map.csv))
152
viaMaps += $(filter-out $(viaMaps) $(fullViaMap),$(wildcard */map.csv))
153

    
154
autogenMaps := $(subst map.,VegBIEN.,$(viaMaps))
155
directMaps := $(autogenMaps) $(filter-out $(autogenMaps),\
156
$(wildcard */VegBIEN.csv))
157

    
158
##### Sources
159

    
160
srcs = $(call sortFilenames,$(call srcsOnly,$(wildcard $*/*)))
161
nonHeaderSrcs = $(filter-out %/header.csv,$(srcs))
162
isRef = $(if $(nonHeaderSrcs),,1)
163
    # empty subdir, so references an already-installed staging table
164
isXml = $(filter %.xml,$(nonHeaderSrcs))
165
nonXml = $(if $(isXml),,1)
166
isCsv = $(if $(nonHeaderSrcs),$(if $(isXml),,1))
167
    # true if $(srcs) non-empty and contains no *.xml
168
catSrcs = $(bin)/cat$(if $(nonXml),_csv) $(srcs)
169
withCatSrcs = $(catSrcs:$(bin)/%=$(bin)/with_%) --
170

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

    
174
%/cat: _always
175
	$(catSrcs)
176

    
177
##### Staging tables installation
178

    
179
dbExports := $(sort $(wildcard *.schema.sql))# schemas first
180
dbExports += $(sort $(filter-out $(dbExports),$(wildcard *.sql)))# all others
181
dbExports := $(strip $(dbExports))# += adds extra whitespace
182
allInstalls := $(if $(dbExports),sql) $(allTables)
183

    
184
install: _always schema $(allInstalls:%=%/install) ;
185

    
186
uninstall: _always rm_schema ;
187
# rm_schema will also drop all staging tables
188

    
189
reinstall: _always uninstall install ;
190

    
191
schema: _always
192
	-echo 'CREATE SCHEMA "$(datasrc)";'|$(psqlAsBien)
193
# ignore errors if schema exists
194

    
195
rm_schema: _always
196
	echo 'DROP SCHEMA IF EXISTS "$(datasrc)" CASCADE;'|$(psqlAsBien)
197

    
198
logInstall = $(if $(log),$(if $(quiet),&>,2>&1|tee $(2)\
199
)$(1)logs/install.log.sql)
200
logInstall* = $(call logInstall,$*/)
201
logInstall*Add = $(call logInstall,$*/,-a)# append to log
202

    
203
# Must come before %/install to override it
204
sql/install: $(dbExports)
205
	($(inDatasrc); cat $+|grep -vF 'SET search_path')|"time" $(psqlAsBien) \
206
$(logInstall)
207

    
208
# The export must be created with:
209
# `--compatible=postgresql --add-locks=false --set-charset --no-create-info`
210
# Must come before `%.sql: _MySQL/%.sql` to override it
211
%.data.sql: _MySQL/%.data.sql
212
	$(bin)/my2pg.data <$< >$@
213

    
214
# The export must be created with:
215
# `--compatible=postgresql --add-locks=false --set-charset`
216
# Add `--no-data` to create a schema-only export.
217
%.sql: _MySQL/%.sql
218
	$(bin)/my2pg <$< >$@
219

    
220
cleanup = (prefix=; . $(bin)/vegbien_dest; env schema=$(datasrc) table=$*\
221
$(bin)/csv2db) $(logInstall*Add)
222

    
223
define exportHeader
224
$(cleanup)
225
echo 'SELECT * FROM "$(datasrc)"."$*" LIMIT 0;'|$(psqlAsBien) \
226
--no-align --field-separator=, --pset=footer=off >$*/header.csv
227
endef
228

    
229
# For staging tables which are derived by joining together other staging tables.
230
%/install %/header.csv: %/create.sql _always
231
	($(inDatasrc); echo 'CREATE TABLE "$*" AS'; cat $<; echo ';')|"time" \
232
$(psqlAsBien) --echo-all --set=table='"$*"' $(logInstall*)
233
	-$(addRowNum)
234
	$(exportHeader)
235
# ignore errors if create.sql already added a primary key
236
addRowNum = echo 'ALTER TABLE "$(datasrc)"."$*"\
237
ADD COLUMN row_num serial NOT NULL PRIMARY KEY;'|$(psqlAsBien) $(logInstall*Add)
238

    
239
# The joined tables must be suffixed with ".src" to prevent the creation of a
240
# row_num column, which collides during joins.
241
isJoinedTable = $(filter %.src,$*)
242
hasRowNum = $(if $(isJoinedTable),,1)
243

    
244
%/install: _always
245
	$(if $(isRef),$(exportHeader),$(if $(nonXml),$(import_install_)))
246
import_install_ = (prefix=; . $(bin)/vegbien_dest; "time" nice -n +5\
247
env schema=$(datasrc) table=$* has_row_num=$(hasRowNum) $(bin)/csv2db\
248
$(catSrcs) $(logInstall*))
249

    
250
##### Maps building
251

    
252
# WARNING: You CANNOT make a subdir using `make inputs/<datasrc>/<subdir>/`.
253
# You must instead make the entire datasource dir: `make inputs/<datasrc>/`
254

    
255
# Maps to (try to) build are added to one of these
256
maps :=
257
createOnlyMaps :=
258

    
259
srcRoot = $(mappings)/root.sh
260
mkSrcMap = $(catSrcs)|(. $(srcRoot); env datasrc=$(datasrc) $(bin)/src_map >$@)
261

    
262
# Src maps cleanup
263
ifneq ($(filter %/.src.csv.last_cleanup,$(MAKECMDGOALS)),)
264
%/.src.csv.last_cleanup: %/src.csv
265
	$(bin)/in_place $< $(bin)/cols +
266
	touch $@
267
else
268
# Autogen src maps with known table names
269
%/src.csv: _always
270
	$(if $(wildcard $@),,$(if $(nonXml),$(mkSrcMap)))
271
	+$(if $(nonXml),$(selfMake) $(@:%/src.csv=%/.src.csv.last_cleanup))
272
# only build if doesn't exist
273
# only build if CSV srcs exist for that table name
274
endif
275

    
276
createOnlyMaps += $(tables:%=%/src.csv)
277

    
278
# Must come before $(root)/% to override it
279
$(coreSelfMap): _always
280
	-+$(subMake)
281
# ignore errors if $(coreSelfMap) does not exist
282

    
283
# Via maps cleanup
284
ifneq ($(filter %/.map.csv.last_cleanup,$(MAKECMDGOALS)),)
285
%/.map.csv.last_cleanup %/unmapped_terms.csv %/new_terms.csv: %/map.csv\
286
$(coreMap) $(vocab) $(dict)
287
	$(bin)/in_place $< $(bin)/canon 1 $(word 3,$+)
288
	$(bin)/in_place $< $(bin)/translate 1 $(word 4,$+)
289
	$(bin)/in_place $< env ignore=1 $(bin)/subtract $(word 2,$+) 0 1 2
290
	$(call newTerms,1) >$*/unmapped_terms.csv
291
	$(call newTerms,0) >$*/new_terms.csv
292
	touch $@
293
# Include comment column so commented mappings are never removed
294

    
295
newTerms = $(bin)/filter_out_ci $(1) $(word 3,$+) <$<|$(bin)/cols $(1)|tail +2
296
else
297
%/map.csv: %/src.csv _always
298
	$(if $(wildcard $@),,$(if $(wildcard $<),$(CP) $< $@))
299
	+$(selfMake) $(@:%/map.csv=%/.map.csv.last_cleanup)
300
endif
301

    
302
srcMap* = $*/src.csv
303
joinSrcMap = $(if $(hasSrc),$(bin)/in_place $@ $(bin)/intersect $(srcMap*) 0)
304
hasSrc = $(shell test -s $(word 3,$+) && echo t)
305

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

    
308
%/map.full.csv: %/map.csv $(coreSelfMap)
309
	$(makeFullCsv)
310
	$(joinSrcMap)
311
maps += $(patsubst %/map.csv,%/map.full.csv,$(viaMaps))
312

    
313
%/VegBIEN.csv: %/map.full.csv $(coreMap)
314
	$(bin)/join <$+|$(bin)/sort_map >$@
315
maps += $(autogenMaps)
316

    
317
maps: $(createOnlyMaps) $(maps) _always ;
318

    
319
all += $(maps)
320

    
321
##### Maps validation
322

    
323
# `env -i sort`: use system's sort, not bin/sort
324
missingMappingsCmd = +$(selfMake) remake 2>&1\
325
|$(SED) -n 's/^.*No $(*2Space) mapping for (.*)$$/\1/p'|$(SED) 's/\/_.*//'\
326
$(if $(filter non-empty_join,$*),|$(bin)/ucase_first 0)|env -i sort|uniq
327

    
328
include $(root)/lib/mappings.Makefile
329

    
330
##### External dependencies
331

    
332
$(root)/%: _always
333
	+$(subMake)
334
.PRECIOUS: $(root)/% # let ext. dir's Makefile decide whether to delete on error
335

    
336
##### Mapping
337

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

    
345
##### Import to VegBIEN
346

    
347
profileTest = $(if $(profile),$(if $(test),1))
348
profileOnly = -env profile_to=/dev/fd/3 $(map2db) 3>&1 1>&2|\
349
$(bin)/profile_stats /dev/fd/0
350

    
351
log_ = $*/logs/$(if $(n),n=$(n).,)$(date).log.sql
352
trace = $(log_:.log.sql=.trace)
353
import = -$(if $(profileTest),$(profileOnly),(set -x; "time" env commit=1\
354
$(if $(profile),profile_to=$(trace)) $(map2db)) $(if $(log),\
355
$(if $(n),,&>$(log_)))$(if $(log),$(if $(n), 2>&1|tee -a $(log_))))
356
# don't abort on import errors, which often relate to invalid input data
357

    
358
import: $(tables:%=%/import) _always ;
359

    
360
%/import: %/VegBIEN.csv _always
361
	$(import)
362
# default:
363
%/import: _always ;
364

    
365
##### Log files from import
366

    
367
logs := $(wildcard */logs/*.log.sql */logs/*.trace)
368

    
369
rm_logs: _always
370
	$(RM) $(logs)
371

    
372
##### Verification of import
373

    
374
verify: $(tables:%=%/verify) _always ;
375

    
376
%/verify: verify/%.ref verify/%.out _always
377
	-$(diffVerbose) $(+_)
378
# don't abort on verification errors, which are expected during development
379
# default:
380
%/verify: verify/%.out _always
381
	$(if $(shell test -e $< && echo t),cat $<)
382
# don't run if verify/%.out's default do-nothing action was used
383
# can't use $(wildcard) because it won't recheck file after verify/%.out is run
384

    
385
psqlExport := "time" $(psqlAsBien) --no-align --field-separator=$$'\t'\
386
--pset=footer=off --pset=null=NULL
387
verify = $(if $(reverify),$(psqlExport) --set=datasource="'$(datasrc)'" <$< >$@)
388

    
389
verify/%.out: $(mappings)/verify.%.sql _always
390
	$(verify)
391
# default:
392
verify/%.out: _always ;
393

    
394
all += $(wildcard verify/*.out)
395

    
396
%.ref: %.ref.sql
397
	($(inDatasrc); cat $<)|$(psqlExport) >$@
398

    
399
##### Editing import
400

    
401
rotate: _always
402
	echo "UPDATE party SET organizationname = organizationname||'.$(date)'\
403
WHERE organizationname = '$(datasrc)';"|$(psqlAsBien)
404

    
405
rm: _always
406
	echo "DELETE FROM party WHERE organizationname = '$(datasrc)';"|\
407
$(psqlAsBien)
408

    
409
##### Testing
410

    
411
testRefOutput = $(subst .by_col,,$(1))
412
testRef = $(testRefOutput).ref
413
hasOwnRef = $(filter $@,$(call testRefOutput,$@))
414
# filter returns non-empty if they are equal
415

    
416
# `rm $@`: Remove outputs of successful tests to reduce clutter
417
# `$(foreach use_staged...)`: Run with use_staged=1
418
define runTest
419
@echo "Testing $(abspath $@)..."
420
>$@ env test=1 n=$(test_n) $(1) $(foreach use_staged,1,$(map2db))
421
@(set -x; $(diffIgnoreSpace) $(call testRef,$@) $@) 2>&1 && rm $@ || { e=$$?;\
422
$(if $(wildcard $(call testRef,$@)),,cat $@;)\
423
$(if $(hasOwnRef),\
424
{\
425
read -p $(emph)'Accept new test output? (y/n)'$(endEmph) REPLY;\
426
if test "$$REPLY" = y; then\
427
(set -x; $(MAKE) $@-ok --directory=$(realpath .) --makefile=../input.Makefile);\
428
exit 0;\
429
fi;\
430
};,\
431
echo $(emph)"Note: The preceding failed test is compared to another test's\
432
output"$(endEmph);\
433
echo $(emph)"When it fails, this always indicates a bug"$(endEmph);\
434
)\
435
exit $$e;}
436
endef
437

    
438
tests :=
439

    
440
# Requires staging tables. To create them, run `make inputs/<datasrc>/install`.
441
# Non-flat-file inputs fall back to mimicking a successful test
442
%/test.xml: %/VegBIEN.csv _always
443
	$(if $(nonXml),$(call runTest,by_col=))
444
tests += %/test.xml
445

    
446
%/test.by_col.xml: %/VegBIEN.csv _always
447
	-$(if $(nonXml),$(call runTest,by_col=1))
448
# Don't abort tester if only by-column test fails
449

    
450
# Only run column-based tests if column-based mode enabled, because these tests
451
# are much slower than the row-based tests for small numbers of rows
452
ifneq ($(by_col),)
453
tests += %/test.by_col.xml
454
endif
455

    
456
testOutputs := $(foreach test,$(tests),$(tables:%=$(test)))
457

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

    
460
test: _always $(testOutputs) ;
461

    
462
all += $(wildcard %/test*.xml)
463

    
464
# Accepts a test output: make <test_output_path>-ok
465
%-ok: _always
466
	mv $* $(call testRef,$*)
467

    
468
accept-all: _always
469
	+yes|$(selfMake) test
470

    
471
##### Documentation
472

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

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