civics-cdf-validator 1.66.dev4__tar.gz → 1.66.dev6__tar.gz

{civics_cdf_validator-1.66.dev4/civics_cdf_validator.egg-info → civics_cdf_validator-1.66.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: civics_cdf_validator
-Version: 1.66.dev4
+Version: 1.66.dev6
 Summary: Checks if an election feed follows best practices
 Home-page: https://github.com/google/civics_cdf_validator
 Author: Google Civics

{civics_cdf_validator-1.66.dev4 → civics_cdf_validator-1.66.dev6/civics_cdf_validator.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: civics_cdf_validator
-Version: 1.66.dev4
+Version: 1.66.dev6
 Summary: Checks if an election feed follows best practices
 Home-page: https://github.com/google/civics_cdf_validator
 Author: Google Civics

{civics_cdf_validator-1.66.dev4 → civics_cdf_validator-1.66.dev6}/office_utils.py

@@ -57,6 +57,7 @@ VALID_OFFICE_ROLES = frozenset([
     "criminal court at law",
     "deputy head of government",
     "deputy state executive",
+    "diplomat",
     "district attorney",
     "district clerk",
     "district court",

{civics_cdf_validator-1.66.dev4 → civics_cdf_validator-1.66.dev6}/rules.py

@@ -474,6 +474,8 @@ class ValidIDREF(base.BaseRule):
 
   _REFERENCE_TYPE_OVERRIDES = {
       # ID refs to GpUnits
+      "DiplomatReceivingStateId": "GpUnit",
+      "DiplomatSendingStateId": "GpUnit",
       "ElectoralDistrictId": "GpUnit",
       "ElectionScopeId": "GpUnit",
       "ScopeLevel": "GpUnit",
@@ -2609,6 +2611,31 @@ class OfficesHaveValidOfficeRole(base.BaseRule):
       )
 
 
+class DiplomaticOfficesHaveCorrespondingGpUnitsSet(base.BaseRule):
+  """Offices with the 'diplomat' role must have a sending or receiving state."""
+
+  def elements(self):
+    return ["Office"]
+
+  def check(self, element):
+    office_roles = (
+        role.text.strip()
+        for role in element.findall("Role")
+        if element_has_text(role)
+    )
+    if "diplomat" not in office_roles:
+      return
+
+    if not element_has_text(
+        element.find("DiplomatSendingStateId")
+    ) and not element_has_text(element.find("DiplomatReceivingStateId")):
+      raise loggers.ElectionError.from_message(
+          "Office with role 'diplomat' must have at least one of"
+          " DiplomatSendingStateId or DiplomatReceivingStateId set.",
+          [element],
+      )
+
+
 class ContestHasValidContestStage(base.BaseRule):
   """Each Contest must have a valid contest-stage."""
 
@@ -5524,6 +5551,7 @@ COMMON_RULES = (
     BadCharactersInPersonFullName,
     CandidateImageInContactInformation,
     DeprecatedPartyLeadershipSchema,
+    DiplomaticOfficesHaveCorrespondingGpUnitsSet,
     DuplicateGpUnits,
     DuplicateID,
     DuplicatedGpUnitOcdId,

{civics_cdf_validator-1.66.dev4 → civics_cdf_validator-1.66.dev6}/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.66.dev4'
+__version__ = '1.66.dev6'