From a44ed46641143b67fd0976d5627d60fafb4b3fa6 Mon Sep 17 00:00:00 2001
From: Nikita Yadav <nikita.yadav@candelatech.com>
Date: Wed, 12 Jan 2022 10:16:13 +0530
Subject: [PATCH] added set mesh services method

Signed-off-by: Nikita Yadav <nikita.yadav@candelatech.com>
---
 libs/controller/controller_2x/controller.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/libs/controller/controller_2x/controller.py b/libs/controller/controller_2x/controller.py
index 98ce5307c..eba22b5c5 100644
--- a/libs/controller/controller_2x/controller.py
+++ b/libs/controller/controller_2x/controller.py
@@ -384,6 +384,26 @@ class UProfileUtility:
         }
         self.mode = None
 
+    def set_mesh_services(self):
+        del self.base_profile_config['metrics']['wifi-frames']
+        del self.base_profile_config['metrics']['dhcp-snooping']
+        var = {
+            "filters": ["probe",
+                        "auth"]
+                }
+        self.base_profile_config["metrics"]['wifi-frames'] = var
+        del self.base_profile_config['services']
+        var2 = {
+            "lldp":{
+                "describe": "uCentral",
+                "location": "universe"
+            },
+            "ssh" : {
+                "port" : 22
+            }
+        }
+        self.base_profile_config['services'] = var2
+
     def set_express_wifi(self, open_flow=None):
         if self.mode == "NAT":
             self.base_profile_config["interfaces"][0]["services"] = ["lldp", "ssh"]