feat: allow reset of proxy

zabuldon · Feb 25, 2021 · 2218c57 · 2218c57
1 parent 4b41e12
commit 2218c57
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/teslajsonpy/teslaproxy.py b/teslajsonpy/teslaproxy.py
@@ -130,6 +130,14 @@ async def prepend_relative_urls(self, base_url: URL, html: Text) -> Text:
             html=html,
         )
 
+    async def reset_data(self) -> None:
+        """Reset all stored data.
+
+        A proxy may need to service multiple login requests if the route is not torn down. This function will reset all data between logins.
+        """
+        self.waf_retry = 0
+        await super().reset_data()
+
     async def prepend_i18n_path(self, base_url: URL, html: Text) -> Text:
         """Prepend path for i18n loadPath so it'll reach the proxy.