process_prmacs in separate method

theengs · Dec 18, 2023 · bf89de1 · bf89de1
1 parent 0faf830
commit bf89de1
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/TheengsGateway/ble_gateway.py b/TheengsGateway/ble_gateway.py
@@ -461,11 +461,9 @@ def decode_advertisement(
             decoded_json = json.loads(decoded_json)
 
             if "prmac" in decoded_json:
-                address = decoded_json.get("id")
-                if address in self.configuration["identities"]:
-                    decoded_json.pop("prmac", None)
-                else:
-                    decoded_json["type"] = "RMAC"
+                decoded_json = self.process_prmacs(
+                    decoded_json,
+                )
 
             # Only process if the device is not a random mac address
             if decoded_json["type"] != "RMAC":
@@ -501,6 +499,19 @@ def decode_advertisement(
             add_manufacturer(data_json, company_id)
             self.publish_json(data_json, decoded=False)
 
+    def process_prmacs(
+        self,
+        decoded_json: DataJSONType,
+    ) -> DataJSONType:
+        """Process potential RMACs."""
+        address = decoded_json.get("id")
+        if address in self.configuration["identities"]:
+            decoded_json.pop("prmac", None)
+        else:
+            decoded_json["type"] = "RMAC"
+
+        return decoded_json
+
     def publish_json(
         self,
         data_json: DataJSONType,