Add connect and disconnect logging

Author: lovelydinosaur

databases/core.py

@@ -1,5 +1,6 @@
 import asyncio
 import functools
+import logging
 import sys
 import typing
 from types import TracebackType
@@ -17,6 +18,9 @@
     from aiocontextvars import ContextVar
 
 
+logger = logging.getLogger("databases")
+
+
 class Database:
     SUPPORTED_BACKENDS = {
         "postgresql": "databases.backends.postgres:PostgresBackend",
@@ -63,6 +67,7 @@ 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))
         self.is_connected = True
 
         if self._force_rollback:
@@ -80,6 +85,7 @@ 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))
         self.is_connected = False
 
     async def __aenter__(self) -> "Database":
@@ -412,14 +418,17 @@ def replace(self, **kwargs: typing.Any) -> "DatabaseURL":
         components = self.components._replace(**kwargs)
         return self.__class__(components.geturl())
 
+    @property
+    def obscure_password(self) -> str:
+        if self.password:
+            return self.replace(password="********")._url
+        return self._url
+
     def __str__(self) -> str:
         return self._url
 
     def __repr__(self) -> str:
-        url = str(self)
-        if self.password:
-            url = str(self.replace(password="********"))
-        return f"{self.__class__.__name__}({repr(url)})"
+        return f"{self.__class__.__name__}({repr(self.obscure_password)})"
 
     def __eq__(self, other: typing.Any) -> bool:
         return str(self) == str(other)