Rails导出csv,包括连接表属性

时间:2013-02-11 17:54:34

标签: ruby-on-rails ruby ruby-on-rails-3 ruby-on-rails-3.2

我的app +关联联接表中有3个模型:

评级集:

class RatingSet < ActiveRecord::Base
  has_many :product_rating_sets, :dependent => :destroy
  has_many :products, :through => :product_rating_sets
end

产品:

class Product < ActiveRecord::Base
  has_many :product_recommendations, :dependent => :destroy
  has_many :recommendations, :through => :product_recommendations
  has_many :product_rating_sets, :dependent => :destroy
  has_many :rating_sets, :through => :product_rating_sets  
end

建议:

class Recommendation < ActiveRecord::Base
  has_many :product_recommendations, :dependent => :destroy
  has_many :products, :through => :product_recommendations
end

我正在尝试向csv函数添加导出。

到目前为止我的方法:

def self.as_csv
    CSV.generate do |csv|
      csv << ["set_id", "time", "item_id", "related_item_id", "rating"]
      set_id = ['1','2','3']
      time = ['10:40am', "11:30pm", "14:40am"]
      item_id = ["123456","123456765", "123456543"]
      related_item_id = ["1234565432345","109876532", "564783920"]
      rating = ["12.3", "13.2", "132.2"]

      all.each do |item|
        csv << item.attributes.values_at(*column_names)#lost here
      end
    end
  end

我打算添加范围来查找set_id,time,item_id,related_item_id和评级,但是对于测试,我使用的是上面的值。如何遍历每个数组,并将值添加到各自标题下的导出的csv文件中?

1 个答案:

答案 0 :(得分:3)

将以下方法添加到lib文件中,并在每个控制器中调用它。基本上必须一次迭代一个模型,并在获得我需要的所有属性后,我附加到csv对象。

def self.as_csv(rating_id)
        CSV.generate do |csv|
          csv << ["set_id","item_id", "related_item_id", "rating", "time"]
          product_ids = RatingSet.find(rating_id).products
          product_ids.each do |product|
           recommendation_ids = Recommendation.find(:all, :joins => :products, :conditions => ["product_id = ?", product.id])
            recommendation_ids.each do |recommendation|
              recommendation.ratings.each do |ratings|
               time_updated = (ratings.updated_at)
                csv <<  [rating_id, product.id, recommendation.id, ratings.rating, time_updated]
              end
            end
          end
        end
      end