Peer.py: allow overriding the log level

Not a best solution, but with minimal code changes.
This commit is contained in:
Vadim Ushakov 2020-10-28 14:53:56 +07:00
parent f1d91989d5
commit 112c778c28

View file

@ -32,6 +32,8 @@ class Peer(object):
self.site = site self.site = site
self.key = "%s:%s" % (ip, port) self.key = "%s:%s" % (ip, port)
self.log_level = logging.DEBUG
self.connection = None self.connection = None
self.connection_server = connection_server self.connection_server = connection_server
self.has_hashfield = False # Lazy hashfield object not created yet self.has_hashfield = False # Lazy hashfield object not created yet
@ -59,12 +61,18 @@ class Peer(object):
return getattr(self, key) return getattr(self, key)
def log(self, text): def log(self, text):
if self.log_level <= logging.DEBUG:
if not config.verbose: if not config.verbose:
return # Only log if we are in debug mode return # Only log if we are in debug mode
logger = None
if self.site: if self.site:
self.site.log.debug("%s:%s %s" % (self.ip, self.port, text)) logger = self.site.log
else: else:
logging.debug("%s:%s %s" % (self.ip, self.port, text)) logger = logging.getLogger()
logger.log(self.log_level, "%s:%s %s" % (self.ip, self.port, text))
# Connect to host # Connect to host
def connect(self, connection=None): def connect(self, connection=None):