set logger accessor

flyerhzm · Feb 6, 2010 · 39a0d62 · 39a0d62
1 parent 7a7e02c
commit 39a0d62
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/regexp_crawler/crawler.rb b/lib/regexp_crawler/crawler.rb
@@ -1,6 +1,6 @@
 module RegexpCrawler
   class Crawler
-    attr_accessor :start_page, :continue_regexp, :named_captures, :model, :save_method, :headers, :encoding, :need_parse, :logger
+    attr_accessor :start_page, :continue_regexp, :named_captures, :model, :save_method, :headers, :encoding, :need_parse
 
     def initialize(options = {})
       @start_page = options[:start_page]
@@ -19,6 +19,10 @@ def capture_regexp=(regexp)
       @capture_regexp = Regexp.new(regexp.source, regexp.options | Regexp::MULTILINE)
     end
 
+    def logger=(logger)
+      @logger = logger == true ? Logger.new(STDOUT) : logger
+    end
+
     def start
       @results = []
       @captured_pages = []