Add check for operation ID to Detect v1 list_results sample

Chronicle Team · copybara-github · commit e65ea67182ad · 2021-05-12T15:00:04.000-07:00
PiperOrigin-RevId: 373458576
diff --git a/detect/v1/list_results.py b/detect/v1/list_results.py
@@ -23,9 +23,10 @@
 import re
 from typing import Any, Mapping, Optional, Sequence, Tuple
 
-from common import chronicle_auth
 from google.auth.transport import requests
 
+from common import chronicle_auth
+
 CHRONICLE_API_BASE_URL = "https://backstory.googleapis.com"
 
 OPERATION_ID_PATTERN = re.compile(
@@ -79,6 +80,10 @@ def list_results(
     except binascii.Error:
       raise ValueError(f"Invalid page token: '{page_token}'.")
 
+  if not OPERATION_ID_PATTERN.fullmatch(operation_id):
+    raise ValueError(f"Invalid detection operation ID: '{operation_id}' != " +
+                     "'rulejob_jo_<UUID>'.")
+
   url = (f"{CHRONICLE_API_BASE_URL}/v1/rules_results?name=operations/" +
          operation_id)
   if page_size > 0:
