Contents

require 'fastercsv'

##
# Imports contacts from a CSV file

class Graybook::Importer::Csv < Graybook::Importer::Base

  DEFAULT_COLUMNS = [:name,:email,:misc]
  DEFAULT_PATTERN = /,/

  ##
  # Matches this importer to a file that contains CSV values

  def =~(options)
    options && options[:file].respond_to?(:open) ? true : false
  end

  ##
  # fetch_contacts! implementation for this importer

  def fetch_contacts!
    lines   = IO.readlines(options[:file].path)
    columns = to_columns(lines.first)
    lines.shift if columns.first == :name
    columns = DEFAULT_COLUMNS.dup unless columns.first == :name

    contacts = Array.new
    lines.each do |l|
      vals = l.split(pattern)
      next if vals.empty?
      vals.map! {|x| x.gsub( /\A"/m, "" ).strip.gsub( /"\Z/m, "" ) }
      contacts << to_hash(columns, vals)
    end

    contacts
  end

  def to_hash(cols, vals) # :nodoc:
    h = Hash.new
    cols.each do |c|
      h[c] = (c == cols.last) ? vals.join(',') : vals.shift
    end
    h
  end

  def to_columns(line) # :nodoc:
    columns = Array.new
    if line.match(pattern)
      tags = line.split(pattern)
    else
      tags = line.split(DEFAULT_PATTERN)
    end
    # deal with "Name,E-mail..." oddity up front
    if tags.first =~ /^name$/i
      tags.shift
      columns << :name
      if tags.first =~ /^e.?mail/i # E-mail or Email
        tags.shift
        columns << :email
      end
    end
    tags.each do |v|
      v = v.to_s.gsub(/\0/,"")
      columns << v.strip.to_sym unless v.blank?
    end
    columns
  end
  
  def pattern
    @pattern ||= (options[:pattern] || DEFAULT_PATTERN) rescue DEFAULT_PATTERN
  end

  Graybook.register(:csv, self)
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
graybook-1.0.24	lib/graybook/importer/csv.rb
graybook-1.0.23	lib/graybook/importer/csv.rb
graybook-1.0.22	lib/graybook/importer/csv.rb