如何为删除文本数组的多个控制器编写方法?

时间:2010-06-22 22:29:39

标签: ruby-on-rails ruby string-substitution

我想写一个简化公司名称的方法。我希望它的工作方式如下:

@clear_company = clear_company(@company.name)

@ company.name =“Company,Inc。”会发生什么@clear_company将是“公司”

如果@ company.name =“Company Corporation”@clear_company将是“公司”

不会有额外的空间。我查看了不同的strip和gsub,但我需要维护一个数组:

clean_array = %w[Inc. Incorporated LLC]

我可以更新它以使其更有效。

我该怎么做?

3 个答案:

答案 0 :(得分:2)

lib / clear_company.rb中的

 module ClearCompany
  BUSINESS_ENTITY = %w[Corporation Inc. Incorporated LLC]

  def clear_company
    strip_business_entity.remove_trailing_punctuation
  end

  def strip_business_entity
    BUSINESS_ENTITY.inject(self) do |company, clean_word|
      company.sub(clean_word, '')
    end
  end

  def remove_trailing_punctuation
    strip.sub(/,$/, '')
  end
end

在config / initializers / string.rb中:

class String
  include ClearCompany
end

如果你喜欢RSpec:

describe String, :clear_company do
  it "removes ', Inc.' from the end" do
    "Company, Inc.".clear_company.should == "Company"
  end

  it "removes ' Corporation' from the end" do
    "Company Corporation".clear_company.should == "Company"
  end
end

答案 1 :(得分:0)

def clear_company(name)
  clean_array = %w[Inc. Incorporated LLC]
  name = name.strip
  word_to_remove = clean_array.find {|x| name[/#{x}$/] }
  name.sub(/#{word_to_remove}$/, '').strip
end

最后.strip很重要,因为没有它,“X Inc.”会成为“X”。

答案 2 :(得分:0)

清理数据并不是控制器真正关心的问题,因此最好将其保留在模型中。最简单的方法是使用before_save过滤器:

class Company < ActiveRecord::Base
  before_save :clean_name

private
  def clean_name
    self.name = name.gsub(/Corporation|LLC|Incorporated|Inc.?/i, "").strip
  end 
end