浏览代码

Add wizard find method

Angus McLeod 6 年之前
父节点
当前提交
bdc699d43f
共有 2 个文件被更改,包括 5 次插入1 次删除
  1. 1 1
      jobs/set_after_time_wizard.rb
  2. 4 0
      lib/wizard.rb

+ 1 - 1
jobs/set_after_time_wizard.rb

@@ -1,7 +1,7 @@
 module Jobs
   class SetAfterTimeWizard < Jobs::Base
     def execute(args)
-      if PluginStoreRow.exists?(plugin_name: 'custom_wizard', key: args[:wizard_id])
+      if CustomWizard::Wizard.find(args[:wizard_id])
         user_ids = []
         User.human_users.each do |u|
           u.custom_fields['redirect_to_wizard'] = args[:wizard_id]

+ 4 - 0
lib/wizard.rb

@@ -167,6 +167,10 @@ class CustomWizard::Wizard
     PluginStore.set('custom_wizard', wizard["id"], wizard)
   end
 
+  def self.find(wizard_id)
+    PluginStoreRow.find_by(plugin_name: 'custom_wizard', key: wizard_id)
+  end
+
   def self.set_redirect(user, wizard_id, url)
     PluginStore.set("#{wizard_id.underscore}_submissions", user.id, [{ redirect_to: url }])
   end