Add scopes for video type, Refs #33

Techbay · May 27, 2016 · a835761 · a835761
1 parent 96f5ab0
commit a835761
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/app/controllers/videos_controller.rb b/app/controllers/videos_controller.rb
@@ -9,9 +9,9 @@ class VideosController < ApplicationController
   def index
 
     if I18n.locale.to_s == "en"
-      @videos = Video.where(video_type: "Youtube")
+      @videos = Video.video_by_type("Youtube")
     elsif I18n.locale.to_s == "zh-CN"
-      @videos = Video.where(video_type: "Youku")
+      @videos = Video.video_by_type("Youku")
     end
   end
 

diff --git a/app/models/video.rb b/app/models/video.rb
@@ -31,6 +31,12 @@ class Video < ApplicationRecord
   has_and_belongs_to_many :users
   has_and_belongs_to_many :tags
   # this is done before save or update
+
+  # scope :by_youtube, -> {where(video_type: "Youtube")}
+  # scope :by_youku, -> {where(video_type: "Youku")}
+  scope :video_by_type, ->(video_type) {where(video_type: video_type)}
+
+
   before_save do
     if self.summary_md_changed?
       puts "convert md to html"