From 0fde659d15e3fd612acaefce6d2d6130a5e42ad6 Mon Sep 17 00:00:00 2001
From: Mike Lang <mikelang3000@gmail.com>
Date: Mon, 21 Aug 2023 18:44:42 +1000
Subject: [PATCH] tootbot: Reconnect on network connection loss

---
 zulip/tootbot.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/zulip/tootbot.py b/zulip/tootbot.py
index 4458b1f..94e8759 100644
--- a/zulip/tootbot.py
+++ b/zulip/tootbot.py
@@ -2,6 +2,7 @@
 import logging
 
 import argh
+import time
 import yaml
 import mastodon
 from bs4 import BeautifulSoup
@@ -225,8 +226,15 @@ def main(conf_file, stream="bot-spam", post_topic="Toots from Desert Bus", notif
 	mastodon_client = mastodon.Mastodon(api_base_url=mc["url"], access_token=mc["access_token"])
 	listener = Listener(zulip_client, stream, post_topic, notification_topic)
 
-	logging.info("Starting")
-	mastodon_client.stream_user(listener)
+	RETRY_INTERVAL = 1
+
+	while True:
+		logging.info("Starting")
+		try:
+			mastodon_client.stream_user(listener)
+		except mastodon.MastodonNetworkError:
+			logging.warning(f"Lost connection, reconnecting in {RETRY_INTERVAL}s")
+			time.sleep(RETRY_INTERVAL)
 
 
 @cli