From d559525bf6bb6c6f945103384c999bf0eaae10dd Mon Sep 17 00:00:00 2001
From: Luke Murphy <lukewm@riseup.net>
Date: Wed, 13 Jan 2021 23:19:06 +0100
Subject: [PATCH] Support logging

Closes https://git.autonomic.zone/decentral1se/xbotlib/issues/2.
---
 xbotlib.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/xbotlib.py b/xbotlib.py
index 210af62..5d442c3 100644
--- a/xbotlib.py
+++ b/xbotlib.py
@@ -3,6 +3,7 @@
 from argparse import ArgumentParser, BooleanOptionalAction
 from configparser import ConfigParser
 from getpass import getpass
+from logging import DEBUG, INFO, basicConfig
 from os import environ
 from os.path import exists
 from pathlib import Path
@@ -46,6 +47,7 @@ class Bot(ClientXMPP):
 
     def __init__(self):
         self.parse_arguments()
+        self.setup_logging()
         self.read_config()
         self.init_bot()
         self.register_xmpp_event_handlers()
@@ -61,8 +63,23 @@ class Bot(ClientXMPP):
             action=BooleanOptionalAction,
             default=True,
         )
+        self.parser.add_argument(
+            "-d",
+            "--debug",
+            help="Set logging to DEBUG",
+            action="store_const",
+            dest="log_level",
+            const=DEBUG,
+            default=INFO,
+        )
         self.args = self.parser.parse_args()
 
+    def setup_logging(self):
+        """Arrange logging for the bot."""
+        basicConfig(
+            level=self.args.log_level, format="%(levelname)-8s %(message)s"
+        )
+
     def read_config(self):
         """Read configuration for running bot."""
         self.config = ConfigParser()