t5: Guard against CIDR too broad errors

supertassu · supertassu · commit f46f8a78ce9f · 2024-07-27T15:41:40.000+03:00
Change-Id: I8e4a850121d73f92df4bf0b1ffee383c6835ab82
diff --git a/majavahbot/tasks/task_5_steward_request_bot.py b/majavahbot/tasks/task_5_steward_request_bot.py
@@ -5,6 +5,7 @@
 from typing import Dict, List
 
 import mwparserfromhell
+import pywikibot
 from mwparserfromhell.wikicode import Wikicode
 from pywikibot.data.api import QueryGenerator
 
@@ -77,11 +78,17 @@ def __init__(self, number, name, site, family):
         self.supports_manual_run = True
 
     def get_steward_who_gblocked_ip(self, api: MediawikiApi, ip_or_range):
-        data = QueryGenerator(
-            site=api.get_site(),
-            list="globalblocks",
-            bgip=ip_or_range,
-        ).request.submit()["query"]["globalblocks"]
+        try:
+            data = QueryGenerator(
+                site=api.get_site(),
+                list="globalblocks",
+                bgip=ip_or_range,
+            ).request.submit()["query"]["globalblocks"]
+        except pywikibot.exceptions.APIError as e:
+            if e.code == "cidrtoobroad":
+                return None
+            raise
+
         if len(data) == 0:
             return None
 
