civics-cdf-validator 1.65.dev0__tar.gz → 1.65.dev1__tar.gz

{civics_cdf_validator-1.65.dev0/civics_cdf_validator.egg-info → civics_cdf_validator-1.65.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: civics_cdf_validator
-Version: 1.65.dev0
+Version: 1.65.dev1
 Summary: Checks if an election feed follows best practices
 Home-page: https://github.com/google/civics_cdf_validator
 Author: Google Civics
@@ -18,6 +18,7 @@ Requires-Dist: frozendict>=2.4.4
 Requires-Dist: attrs>=25.1.0
 Requires-Dist: python-dateutil>=2.8.1
 Requires-Dist: six
+Requires-Dist: botocore
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: absl-py; extra == "test"

{civics_cdf_validator-1.65.dev0 → civics_cdf_validator-1.65.dev1/civics_cdf_validator.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: civics_cdf_validator
-Version: 1.65.dev0
+Version: 1.65.dev1
 Summary: Checks if an election feed follows best practices
 Home-page: https://github.com/google/civics_cdf_validator
 Author: Google Civics
@@ -18,6 +18,7 @@ Requires-Dist: frozendict>=2.4.4
 Requires-Dist: attrs>=25.1.0
 Requires-Dist: python-dateutil>=2.8.1
 Requires-Dist: six
+Requires-Dist: botocore
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: absl-py; extra == "test"

{civics_cdf_validator-1.65.dev0 → civics_cdf_validator-1.65.dev1}/civics_cdf_validator.egg-info/requires.txt

@@ -7,6 +7,7 @@ frozendict>=2.4.4
 attrs>=25.1.0
 python-dateutil>=2.8.1
 six
+botocore
 
 [test]
 pytest

{civics_cdf_validator-1.65.dev0 → civics_cdf_validator-1.65.dev1}/rules.py

@@ -21,6 +21,7 @@ import enum
 import hashlib
 import re
 
+from botocore import utils as botocore_utils
 from civics_cdf_validator import base
 from civics_cdf_validator import gpunit_rules
 from civics_cdf_validator import loggers
@@ -4565,6 +4566,61 @@ class SourceDirPathsAreUnique(base.BaseRule):
       raise loggers.ElectionError(error_log)
 
 
+_ARN_PARSER = botocore_utils.ArnParser()
+
+
+class SqsQueueNameIsFullyQualifiedArn(base.BaseRule):
+  """SqsQueueName must be a valid fully qualified ARN.
+
+  Only AWS region ARNs are supported.
+  """
+
+  def elements(self):
+    return ["Feed"]
+
+  def check(self, element):
+    sqs_queue_name = element.find("SqsQueueName")
+    if not element_has_text(sqs_queue_name):
+      return
+    queue_name = sqs_queue_name.text.strip()
+    feed_id_element = element.find("FeedId")
+    feed_id = (
+        feed_id_element.text.strip()
+        if element_has_text(feed_id_element)
+        else ""
+    )
+
+    is_valid = False
+    if queue_name.startswith("arn:"):
+      try:
+        arn_dict = _ARN_PARSER.parse_arn(queue_name)
+        is_valid = bool(
+            arn_dict.get("partition") == "aws"
+            and arn_dict.get("service", "").lower() == "sqs"
+            and arn_dict.get("region")
+            and arn_dict.get("account")
+            and arn_dict.get("resource")
+        )
+      except botocore_utils.InvalidArnException:
+        is_valid = False
+
+    if not is_valid:
+      if feed_id:
+        error_message = (
+            f"SqsQueueName '{queue_name}' is not a valid fully qualified ARN"
+            f" for feed {feed_id}."
+        )
+      else:
+        error_message = (
+            f"SqsQueueName '{queue_name}' is not a valid fully qualified ARN."
+        )
+      raise loggers.ElectionError.from_message(
+          error_message,
+          [element],
+          [sqs_queue_name.sourceline],
+      )
+
+
 class SqsQueueNameRequiresS3SourceDirPath(base.BaseRule):
   """If SqsQueueName is set, SourceDirPath must also be set and must be an s3 path."""
 
@@ -5616,6 +5672,7 @@ METADATA_RULES = (
     Schema,
     SourceDirPathMustBeSetAfterInitialDeliveryDate,
     SourceDirPathsAreUnique,
+    SqsQueueNameIsFullyQualifiedArn,
     SqsQueueNameRequiresS3SourceDirPath,
     UniqueLabel,
     # go/keep-sorted end

{civics_cdf_validator-1.65.dev0 → civics_cdf_validator-1.65.dev1}/setup.py

@@ -61,6 +61,7 @@ setup(
         'attrs>=25.1.0',
         'python-dateutil>=2.8.1',
         'six',
+        'botocore',
     ],
     setup_requires=['pytest-runner'],
     extras_require={

{civics_cdf_validator-1.65.dev0 → civics_cdf_validator-1.65.dev1}/version.py

@@ -5,4 +5,4 @@ No dependencies should be added to this module.
 See https://packaging.python.org/guides/single-sourcing-package-version/
 """
 
-__version__ = '1.65.dev0'
+__version__ = '1.65.dev1'