From a2ca08905f2614a3da2dc97aee911c8828924e55 Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <balloob@gmail.com>
Date: Sat, 2 Jan 2021 01:34:10 +0100
Subject: [PATCH] Guard unbound var for DSMR (#44673)

---
 homeassistant/components/dsmr/sensor.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/homeassistant/components/dsmr/sensor.py b/homeassistant/components/dsmr/sensor.py
index 0c53fbd6079..78cd317bb3e 100644
--- a/homeassistant/components/dsmr/sensor.py
+++ b/homeassistant/components/dsmr/sensor.py
@@ -197,6 +197,10 @@ async def async_setup_entry(
 
     async def connect_and_reconnect():
         """Connect to DSMR and keep reconnecting until Home Assistant stops."""
+        stop_listener = None
+        transport = None
+        protocol = None
+
         while hass.state != CoreState.stopping:
             # Start DSMR asyncio.Protocol reader
             try:
@@ -211,10 +215,9 @@ async def async_setup_entry(
                     # Wait for reader to close
                     await protocol.wait_closed()
 
-                # Unexpected disconnect
-                if transport:
-                    # remove listener
-                    stop_listener()
+                    # Unexpected disconnect
+                    if not hass.is_stopping:
+                        stop_listener()
 
                 transport = None
                 protocol = None
@@ -234,7 +237,7 @@ async def async_setup_entry(
                 protocol = None
             except CancelledError:
                 if stop_listener:
-                    stop_listener()
+                    stop_listener()  # pylint: disable=not-callable
 
                 if transport:
                     transport.close()