mirror of
https://github.com/internetarchive/brozzler.git
synced 2025-08-08 06:22:23 -04:00
tweak logging
This commit is contained in:
parent
88d3d3b310
commit
d0f5cd7168
4 changed files with 20 additions and 14 deletions
|
@ -17,6 +17,7 @@ See the License for the specific language governing permissions and
|
||||||
limitations under the License.
|
limitations under the License.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
import logging
|
||||||
from pkg_resources import get_distribution as _get_distribution
|
from pkg_resources import get_distribution as _get_distribution
|
||||||
__version__ = _get_distribution('brozzler').version
|
__version__ = _get_distribution('brozzler').version
|
||||||
|
|
||||||
|
@ -57,18 +58,22 @@ class ReachedLimit(Exception):
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return self.__repr__()
|
return self.__repr__()
|
||||||
|
|
||||||
# monkey-patch log level TRACE
|
# monkey-patch log levels TRACE and NOTICE
|
||||||
TRACE = 5
|
logging.TRACE = (logging.NOTSET + logging.DEBUG) // 2
|
||||||
import logging
|
|
||||||
def _logging_trace(msg, *args, **kwargs):
|
|
||||||
logging.root.trace(msg, *args, **kwargs)
|
|
||||||
def _logger_trace(self, msg, *args, **kwargs):
|
def _logger_trace(self, msg, *args, **kwargs):
|
||||||
if self.isEnabledFor(TRACE):
|
if self.isEnabledFor(logging.TRACE):
|
||||||
self._log(TRACE, msg, args, **kwargs)
|
self._log(logging.TRACE, msg, args, **kwargs)
|
||||||
logging.trace = _logging_trace
|
|
||||||
logging.Logger.trace = _logger_trace
|
logging.Logger.trace = _logger_trace
|
||||||
logging._levelToName[TRACE] = 'TRACE'
|
logging.trace = logging.root.trace
|
||||||
logging._nameToLevel['TRACE'] = TRACE
|
logging.addLevelName(logging.TRACE, 'TRACE')
|
||||||
|
|
||||||
|
logging.NOTICE = (logging.INFO + logging.WARN) // 2
|
||||||
|
def _logger_notice(self, msg, *args, **kwargs):
|
||||||
|
if self.isEnabledFor(logging.NOTICE):
|
||||||
|
self._log(logging.NOTICE, msg, args, **kwargs)
|
||||||
|
logging.Logger.notice = _logger_notice
|
||||||
|
logging.notice = logging.root.notice
|
||||||
|
logging.addLevelName(logging.NOTICE, 'NOTICE')
|
||||||
|
|
||||||
# see https://github.com/internetarchive/brozzler/issues/91
|
# see https://github.com/internetarchive/brozzler/issues/91
|
||||||
def _logging_handler_handle(self, record):
|
def _logging_handler_handle(self, record):
|
||||||
|
|
|
@ -317,7 +317,7 @@ class Browser:
|
||||||
kwargs['id'] = msg_id
|
kwargs['id'] = msg_id
|
||||||
msg = json.dumps(kwargs, separators=',:')
|
msg = json.dumps(kwargs, separators=',:')
|
||||||
logging.log(
|
logging.log(
|
||||||
brozzler.TRACE if suppress_logging else logging.DEBUG,
|
logging.TRACE if suppress_logging else logging.DEBUG,
|
||||||
'sending message to %s: %s', self.websock, msg)
|
'sending message to %s: %s', self.websock, msg)
|
||||||
self.websock.send(msg)
|
self.websock.send(msg)
|
||||||
return msg_id
|
return msg_id
|
||||||
|
|
|
@ -43,15 +43,14 @@ def add_common_options(arg_parser, argv=None):
|
||||||
argv = argv or sys.argv
|
argv = argv or sys.argv
|
||||||
arg_parser.add_argument(
|
arg_parser.add_argument(
|
||||||
'-q', '--quiet', dest='log_level', action='store_const',
|
'-q', '--quiet', dest='log_level', action='store_const',
|
||||||
default=logging.INFO, const=logging.WARN, help=(
|
default=logging.INFO, const=logging.NOTICE, help='quiet logging')
|
||||||
'quiet logging, only warnings and errors'))
|
|
||||||
arg_parser.add_argument(
|
arg_parser.add_argument(
|
||||||
'-v', '--verbose', dest='log_level', action='store_const',
|
'-v', '--verbose', dest='log_level', action='store_const',
|
||||||
default=logging.INFO, const=logging.DEBUG, help=(
|
default=logging.INFO, const=logging.DEBUG, help=(
|
||||||
'verbose logging'))
|
'verbose logging'))
|
||||||
arg_parser.add_argument(
|
arg_parser.add_argument(
|
||||||
'--trace', dest='log_level', action='store_const',
|
'--trace', dest='log_level', action='store_const',
|
||||||
default=logging.INFO, const=brozzler.TRACE, help=(
|
default=logging.INFO, const=logging.TRACE, help=(
|
||||||
'very verbose logging'))
|
'very verbose logging'))
|
||||||
# arg_parser.add_argument(
|
# arg_parser.add_argument(
|
||||||
# '-s', '--silent', dest='log_level', action='store_const',
|
# '-s', '--silent', dest='log_level', action='store_const',
|
||||||
|
|
|
@ -94,6 +94,7 @@ class RethinkDbFrontier:
|
||||||
k, expected, result))
|
k, expected, result))
|
||||||
|
|
||||||
def claim_sites(self, n=1):
|
def claim_sites(self, n=1):
|
||||||
|
self.logger.trace('claiming up to %s sites to brozzle', n)
|
||||||
result = (
|
result = (
|
||||||
self.rr.table('sites').get_all(r.args(
|
self.rr.table('sites').get_all(r.args(
|
||||||
r.db(self.rr.dbname).table('sites', read_mode='majority')
|
r.db(self.rr.dbname).table('sites', read_mode='majority')
|
||||||
|
@ -145,6 +146,7 @@ class RethinkDbFrontier:
|
||||||
result["changes"][i]["old_val"]["last_claimed"])
|
result["changes"][i]["old_val"]["last_claimed"])
|
||||||
site = brozzler.Site(self.rr, result["changes"][i]["new_val"])
|
site = brozzler.Site(self.rr, result["changes"][i]["new_val"])
|
||||||
sites.append(site)
|
sites.append(site)
|
||||||
|
self.logger.debug('claimed %s sites', len(sites))
|
||||||
if sites:
|
if sites:
|
||||||
return sites
|
return sites
|
||||||
else:
|
else:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue