Colourize connect/disconnect messages

Author: lovelydinosaur

databases/core.py

@@ -23,8 +23,16 @@
     LOG_EXTRA = {
         "color_message": "Query: " + click.style("%s", bold=True) + " Args: %s"
     }
+    CONNECT_EXTRA = {
+        "color_message": "Connected to database " + click.style("%s", bold=True)
+    }
+    DISCONNECT_EXTRA = {
+        "color_message": "Disconnected from database " + click.style("%s", bold=True)
+    }
 except ImportError:  # pragma: no cover
     LOG_EXTRA = {}
+    CONNECT_EXTRA = {}
+    DISCONNECT_EXTRA = {}
 
 
 logger = logging.getLogger("databases")
@@ -76,7 +84,11 @@ async def connect(self) -> None:
         assert not self.is_connected, "Already connected."
 
         await self._backend.connect()
-        logger.info("Connected to database %s" % repr(self.url.obscure_password))
+        logger.info(
+            "Connected to database %s",
+            repr(self.url.obscure_password),
+            extra=CONNECT_EXTRA,
+        )
         self.is_connected = True
 
         if self._force_rollback:
@@ -94,7 +106,11 @@ async def disconnect(self) -> None:
             await self._global_transaction.__aexit__()
 
         await self._backend.disconnect()
-        logger.info("Disconnected from database %s" % repr(self.url.obscure_password))
+        logger.info(
+            "Disconnected from database %s",
+            repr(self.url.obscure_password),
+            extra=DISCONNECT_EXTRA,
+        )
         self.is_connected = False
 
     async def __aenter__(self) -> "Database":