diff options
-rw-r--r-- | crocoite/browser.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/crocoite/browser.py b/crocoite/browser.py index 06f2b22..23b251f 100644 --- a/crocoite/browser.py +++ b/crocoite/browser.py @@ -91,12 +91,14 @@ class SiteLoader: tab.Network.responseReceived = self._responseReceived tab.Network.loadingFinished = self._loadingFinished tab.Network.loadingFailed = self._loadingFailed + tab.Log.entryAdded = self._entryAdded #tab.Page.loadEventFired = loadEventFired # start the tab tab.start() # enable events + tab.Log.enable () tab.Network.enable() tab.Page.enable () tab.Network.clearBrowserCache () @@ -137,6 +139,7 @@ class SiteLoader: tab.Page.stopLoading () tab.Network.disable () tab.Page.disable () + tab.Log.disable () # XXX: we can’t drain the event queue directly, so insert (yet another) wait tab.wait (1) tab.Network.requestWillBeSent = None @@ -144,6 +147,7 @@ class SiteLoader: tab.Network.loadingFinished = None tab.Network.loadingFailed = None tab.Page.loadEventFired = None + tab.Log.entryAdded = None def __exit__ (self, exc_type, exc_value, traceback): self.tab.stop () @@ -224,6 +228,14 @@ class SiteLoader: item = self.requests.pop (reqId, None) self.loadingFailed (item) + def _entryAdded (self, **kwargs): + """ Log entry added """ + entry = kwargs['entry'] + level = {'verbose': logging.DEBUG, 'info': logging.INFO, + 'warning': logging.WARNING, + 'error': logging.ERROR}[entry['level']] + self.logger.log (level, 'console: {}: {}'.format (entry['source'], entry['text']), extra=entry) + class AccountingSiteLoader (SiteLoader): """ SiteLoader that keeps basic statistics about retrieved pages. |