From bd3a109299a18c2a441ca9a33ddcc2b5208f7f1c Mon Sep 17 00:00:00 2001
From: Pablo Hoffman <pablo@pablohoffman.com>
Date: Tue, 9 Oct 2012 12:41:31 -0200
Subject: [PATCH] Track number of log messages via Scrapy stats

Log messages count are grouped per LEVEL like:

    'log_count/DEBUG': 8,
    'log_count/ERROR': 1,
    'log_count/INFO': 4,

This required passing the Crawler object to scrapy logging facility, so
the scrapy.log module underwent a few changes which should be backwards
compatible (or as backwards compatible as they could be).
---
 scrapy/command.py |  2 +-
 scrapy/log.py     | 27 +++++++++++++++++++++------
 2 files changed, 22 insertions(+), 7 deletions(-)

diff --git a/scrapy/command.py b/scrapy/command.py
index 3bfbe2a24..629d9bd69 100644
--- a/scrapy/command.py
+++ b/scrapy/command.py
@@ -28,7 +28,7 @@ class ScrapyCommand(object):
     @property
     def crawler(self):
         if not log.started:
-            log.start_from_settings(self.settings)
+            log.start_from_crawler(self._crawler)
         self._crawler.configure()
         return self._crawler
 
diff --git a/scrapy/log.py b/scrapy/log.py
index 4d9a34e17..53cb56965 100644
--- a/scrapy/log.py
+++ b/scrapy/log.py
@@ -33,15 +33,28 @@ started = False
 
 class ScrapyFileLogObserver(log.FileLogObserver):
 
-    def __init__(self, f, level=INFO, encoding='utf-8'):
+    def __init__(self, f, level=INFO, encoding='utf-8', crawler=None):
         self.level = level
         self.encoding = encoding
+        if crawler:
+            self.crawler = crawler
+            self.emit = self._emit_with_crawler
+        else:
+            self.emit = self._emit
         log.FileLogObserver.__init__(self, f)
 
-    def emit(self, eventDict):
+    def _emit(self, eventDict):
         ev = _adapt_eventdict(eventDict, self.level, self.encoding)
         if ev is not None:
             log.FileLogObserver.emit(self, ev)
+        return ev
+
+    def _emit_with_crawler(self, eventDict):
+        ev = self._emit(eventDict)
+        if ev:
+            level = ev['logLevel']
+            sname = 'log_count/%s' % level_names.get(level, level)
+            self.crawler.stats.inc_value(sname)
 
 def _adapt_eventdict(eventDict, log_level=INFO, encoding='utf-8', prepend_level=True):
     """Adapt Twisted log eventDict making it suitable for logging with a Scrapy
@@ -99,15 +112,16 @@ def _get_log_level(level_name_or_id):
     else:
         raise ValueError("Unknown log level: %r" % level_name_or_id)
 
-def start(logfile=None, loglevel='INFO', logstdout=True, logencoding='utf-8'):
+def start(logfile=None, loglevel='INFO', logstdout=True, logencoding='utf-8', crawler=None):
     if log.defaultObserver: # check twisted log not already started
         loglevel = _get_log_level(loglevel)
         file = open(logfile, 'a') if logfile else sys.stderr
-        sflo = ScrapyFileLogObserver(file, loglevel, logencoding)
+        sflo = ScrapyFileLogObserver(file, loglevel, logencoding, crawler)
         _oldshowwarning = warnings.showwarning
         log.startLoggingWithObserver(sflo.emit, setStdout=logstdout)
         # restore warnings, wrongly silenced by Twisted
         warnings.showwarning = _oldshowwarning
+        return sflo
 
 def msg(message=None, _level=INFO, **kw):
     kw['logLevel'] = kw.pop('level', _level)
@@ -122,13 +136,14 @@ def err(_stuff=None, _why=None, **kw):
     kw.setdefault('system', 'scrapy')
     log.err(_stuff, _why, **kw)
 
-def start_from_settings(settings):
+def start_from_crawler(crawler):
     global started
+    settings = crawler.settings
     if started or not settings.getbool('LOG_ENABLED'):
         return
     started = True
 
     start(settings['LOG_FILE'], settings['LOG_LEVEL'], settings['LOG_STDOUT'],
-        settings['LOG_ENCODING'])
+        settings['LOG_ENCODING'], crawler)
     msg("Scrapy %s started (bot: %s)" % (scrapy.__version__, \
         settings['BOT_NAME']))