Few changes related to memory leak #72

erikkastelec · May 10, 2023 · 923acfe · 923acfe
1 parent 6540941
commit 923acfe
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/custom_components/wemportal/coordinator.py b/custom_components/wemportal/coordinator.py
@@ -58,6 +58,7 @@ async def _async_update_data(self):
  self.config_entry.data.get(CONF_PASSWORD),
  self.config_entry.options,
  )
+ del self.api
  self.api = new_api
  except Exception as exc2:
  raise UpdateFailed from exc2

diff --git a/custom_components/wemportal/wemportalapi.py b/custom_components/wemportal/wemportalapi.py
@@ -182,6 +182,8 @@ def api_login(self):
  "AppVersion": "2.0.2",
  "ClientOS": "Android",
  }
+ if self.session is not None:
+ self.session.close()
  self.session = reqs.Session()
  self.session.cookies.clear()
  self.session.headers.update(self.headers)