Handle long URIs (lyft#31)

If an escalation policy has over 155 services, this script will create a URL that is too long for urllib3 to handle.

This PR handles that case by splitting up that call into 2 chunks.

Author: Alex Chantavy

pull_alerts.py

@@ -3,6 +3,7 @@
 
 import argparse
 import logging
+import urllib
 from collections import Counter, OrderedDict, defaultdict, namedtuple
 from datetime import datetime, timedelta
 from dateutil import tz
@@ -45,11 +46,24 @@ def pretty_output(self):
 
 def recent_incidents_for_services(services, time_window):
     service_ids = [service.id for service in services]
-    recent_incidents = list(pagerduty_service.incidents.list(
-        service_ids=service_ids,
-        since=datetime.now() - time_window
-    ))
-    return recent_incidents
+    try:
+        recent_incidents = list(pagerduty_service.incidents.list(
+            service_ids=service_ids,
+            since=datetime.now() - time_window
+        ))
+        return recent_incidents
+
+    except urllib.error.HTTPError as e:
+        if e.reason == 'URI Too Long':
+            mid_point = int(len(service_ids)/2)
+            return recent_incidents_for_services(
+               service_ids=service_ids[:mid_point],
+               time_window=time_window
+            ) + recent_incidents_for_services(
+               service_ids=service_ids[mid_point:],
+               time_window=time_window
+            )
+        raise
 
 
 def print_all_incidents(