[uss_qualifier/mock_uss/riddp/observation] Propagate 'eu_classificati…

…on' field
interuss · Dec 17, 2024 · 11407cf · 11407cf
1 parent 1667619
commit 11407cf
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/monitoring/mock_uss/riddp/routes_observation.py b/monitoring/mock_uss/riddp/routes_observation.py
@@ -213,6 +213,7 @@ def riddp_flight_details(flight_id: str) -> Tuple[str, int]:
         ),
         uas=observation_api.UAS(
             id=details.arbitrary_uas_id,
+            eu_classification=details.eu_classification,
         ),
     )
     if details.operator_location is not None:

diff --git a/monitoring/monitorlib/fetch/rid.py b/monitoring/monitorlib/fetch/rid.py
@@ -503,6 +503,22 @@ def arbitrary_uas_id(self) -> Optional[str]:
                 f"Cannot retrieve plain_uas_id using RID version {self.rid_version}"
             )
 
+    @property
+    def eu_classification(
+        self,
+    ) -> Optional[v22a.api.UAClassificationEU]:
+        if self.rid_version == RIDVersion.f3411_19:
+            return None
+        elif self.rid_version == RIDVersion.f3411_22a:
+            if self.v22a_value.has_field_with_value("eu_classification"):
+                return self.v22a_value.eu_classification
+            else:
+                return None
+        else:
+            raise NotImplementedError(
+                f"Cannot retrieve UA classification using RID version {self.rid_version}"
+            )
+
     @property
     def operator_location(
         self,