seeder.rb 456 字节
Newer Older
1 2 3
module Gitlab
  class Seeder
    def self.quiet
D
Dmitriy Zaporozhets 已提交
4
      mute_mailer
5 6 7 8 9
      SeedFu.quiet = true
      yield
      SeedFu.quiet = false
      puts "\nOK".green
    end
D
Dmitriy Zaporozhets 已提交
10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27

    def self.by_user(user)
      begin
        Thread.current[:current_user] = user
        yield
      ensure
        Thread.current[:current_user] = nil
      end
    end

    def self.mute_mailer
      code = <<-eos
def Notify.delay
  self
end
      eos
      eval(code)
    end
28 29
  end
end