Scan full sky query if pixels in query exceed a threshold (#329)

nuztf/ampel_api.py

@@ -25,6 +25,8 @@
 API_ZTF_ARCHIVE_URL = API_BASEURL + "/api/ztf/archive/v3"
 API_CATALOGMATCH_URL = API_BASEURL + "/api/catalogmatch"
 
+MAX_N_PIX = 1000
+
 _, ampel_api_archive_token = load_credentials("ampel_api_archive_token")
 
 
@@ -498,6 +500,18 @@ def ampel_api_skymap(
     # Now we reduce the query size
     regions = utils.deres(nside=nside, ipix=pixels)
 
+    n_pix = 0
+    for reg in regions:
+        n_pix += len(reg["pixels"])
+
+    logger.debug(f"This comprises {n_pix} individual pixels")
+
+    if n_pix > MAX_N_PIX:
+        logger.warning(
+            f"Total number of pixels exceeds threshold ({MAX_N_PIX} pixels). Issuing a query for the full sky instead."
+        )
+        regions = [{"nside": 1, "pixels": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]}]
+
     query = {
         "regions": regions,
         "jd": {

nuztf/skymap_scanner.py

@@ -17,6 +17,7 @@
     ampel_api_acknowledge_chunk,
     ampel_api_lightcurve,
     ampel_api_skymap,
+    ampel_api_timerange,
     get_preprocessed_results,
 )
 from nuztf.base_scanner import BaseScanner

slackbot.py

@@ -59,7 +59,7 @@ def __init__(
             if self.event_type == "nu":
                 self.time_window = 10
             elif self.event_type == "gw":
-                self.time_window = 2
+                self.time_window = 3
         else:
             self.time_window = time_window