Support reindexing only one by-law

Author: longhotsummer

bin/reindex_bylaws.rb

@@ -13,16 +13,27 @@
 require 'steno/search'
 
 unless ENV['BONSAI_URL']
-    puts "WARNING: The env variable BONSAI_URL is not set, indexing to local Elasticsearch."
-    puts "         You probably want to set BONSAI_URL, check heroku config | fgrep BONSAI"
+  puts "WARNING: The env variable BONSAI_URL is not set, indexing to local Elasticsearch."
+  puts "         You probably want to set BONSAI_URL, check heroku config | fgrep BONSAI"
 end
 
-bylaws = Slaw::DocumentCollection.new
-bylaws.discover('../za-by-laws/by-laws/', Slaw::ByLaw)
-
 regions = File.open('../za-by-laws/regions/regions.json') { |f| JSON.load(f) }
+searcher = Steno::Search.searcher
+
+if ARGV.empty?
+  bylaws = Slaw::DocumentCollection.new
+  bylaws.discover('../za-by-laws/by-laws/', Slaw::ByLaw)
+  remap = true
+else
+  bylaws = ARGV.map { |fname| b = Slaw::ByLaw.new; b.load(fname); b }
+  remap = false
+end
+
+# only do this if we're doing the whole set, since it cleans out the
+# existing index
+searcher.define_mapping! if remap
 
-Steno::Search.searcher.reindex!(bylaws) do |doc, hash|
+searcher.reindex!(bylaws) do |doc, hash|
   hash['region_name'] = regions[doc.region]['name']
 end
 

lib/steno/elasticsearch.rb

@@ -32,7 +32,6 @@ def create_client(client_params)
     end
 
     def reindex!(docs, &block)
-      define_mapping!
       index_documents!(docs, &block)
     end