Class: Geonames::CLI
- Inherits:
-
Object
- Object
- Geonames::CLI
- Defined in:
- lib/geonames_local/cli.rb
Overview
Command Line Interface for Geonames Local
Class Method Summary collapse
- .load_adapter(name) ⇒ Object
- .load_config ⇒ Object
- .stop! ⇒ Object
- .trap_signals ⇒ Object
- .unify!(dump, zip) ⇒ Object
-
.work(argv) ⇒ Object
Ugly but works?.
- .work_nations ⇒ Object
- .work_spots ⇒ Object
- .wrapper ⇒ Object
Class Method Details
.load_adapter(name) ⇒ Object
163 164 165 166 167 168 |
# File 'lib/geonames_local/cli.rb', line 163 def load_adapter(name) require_relative "models/#{name}" rescue LoadError => e info "Can't find adapter for #{name} #{e}" stop! end |
.load_config ⇒ Object
58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
# File 'lib/geonames_local/cli.rb', line 58 def load_config info 'Loading config file...' if Opt[:config] Opt.merge! YAML.load(File.read(Opt[:config])) else # Load config/geonames.yml if there's one if File.exist?(cfg = File.join('config', 'geonames.yml')) Opt.merge! YAML.load(File.read(cfg)) else fail end end rescue info "Can't find config file" exit end |
.stop! ⇒ Object
181 182 183 184 |
# File 'lib/geonames_local/cli.rb', line 181 def stop! info 'Closing Geonames...' exit end |
.trap_signals ⇒ Object
75 76 77 78 79 |
# File 'lib/geonames_local/cli.rb', line 75 def trap_signals puts 'Geopolitical Local Start!' trap(:INT) { stop! } trap(:TERM) { stop! } end |
.unify!(dump, zip) ⇒ Object
170 171 172 173 174 175 176 177 178 179 |
# File 'lib/geonames_local/cli.rb', line 170 def unify!(dump, zip) start = Time.now dump.map! do |spot| next spot unless (other = zip.find { |z| z.code == spot.code }) spot.zip = other.zip spot end info "Done. #{(Time.now - start).to_i}s" dump end |
.work(argv) ⇒ Object
Ugly but works?
108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 |
# File 'lib/geonames_local/cli.rb', line 108 def work(argv) start = Time.now trap_signals Opt.merge! (argv) if Opt[:locales].nil? || Opt[:locales].empty? Opt[:locales] = ['en'] end if shp = Opt[:shp] SHP.import(shp) exit end # # Return Codes and Exit # if argv[0] =~ /list|codes/ Codes.each do |key, val| str = [val.values, key.to_s].join(' ').downcase if s = argv[1] next unless str =~ /#{s.downcase}/ end puts "#{val[:en_us]}: #{key}" end exit end # # If arguments scaffold, config, write down yml. # if argv[0] =~ /scaff|conf|init/ fname = (argv[1] || 'geonames') + '.yml' if File.exist?(fname) puts "File exists: #{fname}" else puts "Writing to #{fname}" `cp #{File.join(File.dirname(__FILE__), 'config', 'geonames.yml')} #{fname}` end exit end # Load config if we got til here load_config # Export Data as CSV or JSON return Geonames::Export.new(Nation.all).to_csv if argv[0] =~ /csv|json/ # Do the magic! Import Geonames Data load_adapter(Opt[:store]) info "Using adapter #{Opt[:store]}.." wrapper.clean if Opt[:clean] puts Benchmark.measure { work_nations } unless wrapper.nations_populated? puts Benchmark.measure { work_spots } end |
.work_nations ⇒ Object
85 86 87 88 89 90 91 92 |
# File 'lib/geonames_local/cli.rb', line 85 def work_nations info "\nPopulating 'nations' database..." dump = Geonames::Dump.new(:all, :dump) info "\n---\nTotal #{dump.data.length} parsed." info 'Writing to nations DB' wrapper.nations dump.data end |
.work_spots ⇒ Object
94 95 96 97 98 99 100 101 102 103 104 105 |
# File 'lib/geonames_local/cli.rb', line 94 def work_spots info "\nPopulating 'regions' and 'cities' database..." zip = Geonames::Dump.new(Opt[:nations], :zip).data dump = Geonames::Dump.new(Opt[:nations], :dump).data info "\n---\nTotal #{dump.size} parsed. #{zip.size} postal codes." info 'Join dump << zip' dump = unify!(dump, zip).group_by(&:kind) info 'Writing to DB...' wrapper.batch dump end |
.wrapper ⇒ Object
81 82 83 |
# File 'lib/geonames_local/cli.rb', line 81 def wrapper Geonames::Models::MongoWrapper end |