Make (un)bypass input available for all firmware versions

unii-security · Jun 20, 2024 · d145fb3 · d145fb3
1 parent 9419c50
commit d145fb3
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/unii/unii.py b/unii/unii.py
@@ -390,15 +390,21 @@ def _handle_equipment_information(self, data: UNiiEquipmentInformation):
 
         self.equipment_information = data
 
+        # Set the supported features of the equipment
+        self.features = []
+
+        # Feature that applies to all versions of the firmware
+        self.features.append(UNiiFeature.BYPASS_INPUT)
+
         # Get capabilities based on firmware version number
-        software_version = (
-            self.equipment_information.software_version.finalize_version()
-        )
-        if software_version.match(">=2.17.0"):
-            # self.features.append(UNiiFeature.ARM_SECTION)
-            self.features.append(UNiiFeature.BYPASS_INPUT)
-            # self.features.append(UNiiFeature.BYPASS_ZONE)
-            # self.features.append(UNiiFeature.SET_OUTPUT)
+        # Library doesn't distinct between versions yet, so disabled for now
+        # software_version = (
+        #     self.equipment_information.software_version.finalize_version()
+        # )
+        # if software_version.match(">=2.17.0"):
+        #     self.features.append(UNiiFeature.ARM_SECTION)
+        #     self.features.append(UNiiFeature.BYPASS_ZONE)
+        #     self.features.append(UNiiFeature.SET_OUTPUT)
 
     def _handle_section_arrangement(self, data: UNiiSectionArrangement):
         for _, section in data.items():