fias 0.0.2 → 1.0.0

data/lib/fias/importer/sqlite.rb

@@ -1,38 +0,0 @@
-module Fias
-  module Importer
-    # Нужно для :memory: баз в первую очередь
-    class Sqlite < Base
-      def import_table(name, table_name, dbf, &block)
-        truncate_table(table_name)
-
-        qmarks = ['?'] * dbf.columns.keys.size
-        qmarks = qmarks.join(', ')
-
-        dbf.each_with_index do |record, index|
-          data = record.attributes
-
-          should_import = yield(name, data, index) if block_given?
-
-          unless should_import === false
-            columns = data.keys.join(', ')
-
-            values = data.values.map do |value|
-              if value.is_a?(Date)
-                value.to_s
-              else
-                value
-              end
-            end
-
-            connection.execute("INSERT INTO #{table_name} (#{columns}) VALUES (#{qmarks});", values)
-          end
-        end
-      end
-
-      private
-      def truncate_table(table_name)
-        connection.execute "DELETE FROM #{table_name} WHERE 1"
-      end
-    end
-  end
-end

data/lib/generators/fias/migration.rb

@@ -1,34 +0,0 @@
-module Fias
-  class MigrationGenerator < Rails::Generators::Base
-    include Rails::Generators::Migration
-
-    class_option :prefix, type: :string, default: :fias, desc: 'Table names prefix'
-    class_option :path, type: :string, default: 'tmp/fias', desc: 'Path to FIAS dbfs'
-    class_option :only, type: :string, default: '', desc: 'Only tables'
-
-    source_root File.expand_path("../templates", __FILE__)
-
-    def generate_migration
-      only = options.only.split(',').map(&:strip)
-      wrapper = Fias::DbfWrapper.new(options.path)
-      importer = Fias::Importer.build(prefix: options.prefix)
-
-      tables = wrapper.tables(only)
-      @schema = importer.schema(tables)
-      @schema.gsub!("\n", "\n    ")
-
-      migration_template 'create_fias_tables.rb', 'db/migrate/create_fias_tables'
-    end
-
-    def usage
-      "Generates FIAS migrations for application"
-    end
-
-    # https://rails.lighthouseapp.com/projects/8994/tickets/3820-make-railsgeneratorsmigrationnext_migration_number-method-a-class-method-so-it-possible-to-use-it-in-custom-generators
-    def self.next_migration_number(dirname)
-      orm = Rails.configuration.generators.options[:rails][:orm]
-      require "rails/generators/#{orm}"
-      "#{orm.to_s.camelize}::Generators::Base".constantize.next_migration_number(dirname)
-    end
-  end
-end

data/lib/generators/fias/templates/create_fias_tables.rb

@@ -1,5 +0,0 @@
-class CreateFiasTables < ActiveRecord::Migration
-  def change
-    <%= @schema %>
-  end
-end

data/tasks/fias.rake

@@ -1,68 +0,0 @@
-require 'fias'
-
-namespace :fias do
-  class << self
-    private
-    # Если гем используется внутри рельсов - стоит загрузить энвайронмент
-    # и подключиться к БД.
-    def may_be_rails(name)
-      defined?(Rails) ? {name => :environment} : name
-    end
-
-    # Открывает DBFы ФИАС, соединяется с базой и передает все это блоку
-    def within_connection(&block)
-      require 'active_record'
-      require 'progress_bar'
-
-      begin
-        ActiveRecord::Base.connection
-      rescue ActiveRecord::ConnectionNotEstablished
-        if ENV['DATABASE_URL'].nil?
-          raise ArgumentError, 'Specify database in DATABASE_URL env variable'
-        end
-
-        ActiveRecord::Base.establish_connection
-      end
-
-      fias_path = ENV['FIAS'] || 'tmp/fias'
-      wrapper = Fias::DbfWrapper.new(fias_path)
-      importer = Fias::Importer.build(prefix: ENV['PREFIX'])
-
-      yield(wrapper, importer)
-    end
-
-    def only
-      only = ENV['ONLY'].to_s.split(',').map(&:strip)
-    end
-  end
-
-  desc 'Create FIAS tables (could specify tables PREFIX, PATH to dbfs and DATABASE_URL, EXCLUDE or ONLY tables)'
-  task may_be_rails(:create_tables) do
-    within_connection do |wrapper, importer|
-      tables = wrapper.tables(only)
-      # TODO: Добавить во враппер tables, это убрать
-      raise "DBF file not found for: #{key}" if tables.keys.any? { |key| key.nil? }
-      ActiveRecord::Schema.define do
-        eval(importer.schema(tables))
-      end
-    end
-  end
-
-  desc 'Import FIAS data'
-  task may_be_rails(:import) do
-    within_connection do |wrapper, importer|
-      tables = wrapper.tables(only)
-
-      total_record_count = tables.sum do |accessor, dbf|
-        dbf.present? ? dbf.record_count : 0
-      end
-
-      puts 'Importing FIAS data...'
-
-      bar = ProgressBar.new(total_record_count)
-      importer.import(tables) do
-        bar.increment!
-      end
-    end
-  end
-end