module LookyLu
  class States

    def self.insert_data(country_name, object_name)

      states = from_country country_name
      begin
        class_object = eval(object_name.camelize)

        states.each do |s|
          class_object.where(s).first_or_create
        end
      rescue => e
        raise "Could not find object for #{object_name.camelize}"
      end

    end


    def self.from_country(country_name)

      yml = YAML.load_file(File.dirname(__FILE__) + "/states.yml")

      if country_name == 'all'
        states = []

        yml.each do |k,v|
          states += v
        end

      else
        states = yml[country_name]
      end

      states
    end

  end
end