Contents

require 'csv'
require 'optparse'
require 'sqlite3'

require 'csv_query/database'

module CsvQuery
  class Query
    attr_reader :csv_data, :options

    DEFAULT_OPTIONS = {
      :delimiter => ',',
      :select => '*'
    }

    def initialize(csv_data, outputter, options = {})
      @csv_data = csv_data
      @outputter = outputter
      @options = DEFAULT_OPTIONS.merge(options)
    end

    def run
      results = run_query
      output_results_table(results)
    end

    private

    def build_sql_query
      sql_query = options[:sql_query]
      return sql_query unless sql_query.nil?

      select_statement = "SELECT #{options[:select]}"
      from_statement = "FROM csv"
      where_statement = if options[:where]
        "WHERE #{options[:where]}"
      end

      [
        select_statement,
        from_statement,
        where_statement
      ].join(" ")
    end

    def create_database_with_data_from_csv
      database = CsvQuery::Database.new(csv)
      database.import_data_from_csv(csv)
      database
    end

    def csv
      @parsed_csv ||= parse_csv_data
    end

    def database
      @database ||= create_database_with_data_from_csv
    end

    def headers
      options[:headers] || :first_row
    end

    def output_results_table(results)
      @outputter.output(results)
    end

    def parse_csv_data
      csv_options = {
        :headers => headers,
        :col_sep => options[:delimiter]
      }

      CSV.parse(csv_data, csv_options)
    end

    def run_query
      database.query(sql_query)
    end

    def sql_query
      @sql_query ||= build_sql_query
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
csv_query-1.0.4	lib/csv_query/query.rb