From 42f44d302019f29a56d27cd05b04f45472a544c8 Mon Sep 17 00:00:00 2001
From: andvikt <39195436+andvikt@users.noreply.github.com>
Date: Wed, 20 Jan 2021 23:33:53 +0300
Subject: [PATCH] Fix offline bugs

---
 custom_components/mega/hub.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/custom_components/mega/hub.py b/custom_components/mega/hub.py
index fed025a..40e3dc4 100644
--- a/custom_components/mega/hub.py
+++ b/custom_components/mega/hub.py
@@ -98,10 +98,10 @@ class MegaD:
             else:
                 await self.get_port(self.port_to_scan)
 
-            await asyncio.sleep(5)
+            await asyncio.sleep(1)
             if (datetime.now() - self.last_update).total_seconds() > (self.poll_interval + 10):
                 await self.get_port(self.port_to_scan)
-                await asyncio.sleep(5)
+                await asyncio.sleep(1)
                 if (datetime.now() - self.last_update).total_seconds() > (self.poll_interval + 10):
                     self.lg.warning('mega is offline')
                     self.hass.states.async_set(
@@ -168,7 +168,7 @@ class MegaD:
         ftr = asyncio.get_event_loop().create_future()
 
         def cb(msg):
-
+            self.last_update = datetime.now()
             try:
                 if '"value":NA' in msg.payload.decode():
                     if not ftr.done():