From 4e53a43006e76e8536ca5b88b850106c5372ebb8 Mon Sep 17 00:00:00 2001
From: Tulir Asokan <tulir@maunium.net>
Date: Tue, 1 Aug 2023 14:55:09 +0300
Subject: [PATCH] Don't immediately stop polling on auth errors

---
 mautwitdm/poller.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/mautwitdm/poller.py b/mautwitdm/poller.py
index 18424d8..27572a5 100644
--- a/mautwitdm/poller.py
+++ b/mautwitdm/poller.py
@@ -60,6 +60,7 @@ class TwitterPoller(TwitterDispatcher):
     poll_sleep: int = 3
     error_sleep: int = 5
     max_poll_errors: int = 12
+    max_poll_auth_errors: int = 4
     poll_cursor: str | None
     dispatch_initial_resp: bool
     _poll_task: asyncio.Task | None
@@ -297,10 +298,13 @@ async def _poll_forever(self) -> None:
                     self.log.debug(f"Increased poll sleep to {self.poll_sleep}")
                 await asyncio.sleep(sleep)
                 continue
-            except TwitterAuthError:
-                raise
             except Exception as e:
-                if errors > self.max_poll_errors > 0:
+                max_errors = (
+                    self.max_poll_auth_errors
+                    if isinstance(e, TwitterAuthError)
+                    else self.max_poll_errors
+                )
+                if errors > max_errors > 0:
                     self.log.debug(
                         f"Error count ({errors}) exceeded maximum, raising error as fatal"
                     )