Use get_snmp_session_for_profile() in SNMPHandler

Lets the utility function set up the SNMP session for SNMPHandler based management handlers in PortAdmin.
Uninett · Nov 16, 2023 · d028681 · d028681
1 parent 5368290
commit d028681
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/python/nav/portadmin/snmp/base.py b/python/nav/portadmin/snmp/base.py
@@ -176,10 +176,8 @@ def _get_read_only_handle(self):
             if not profile:
                 raise NoReadOnlyManagementProfileError
 
-            self.read_only_handle = Snmp.Snmp(
+            self.read_only_handle = get_snmp_session_for_profile(profile)(
                 host=self.netbox.ip,
-                community=profile.snmp_community,
-                version=profile.snmp_version,
                 retries=self.retries,
                 timeout=self.timeout,
             )
@@ -206,10 +204,8 @@ def _get_read_write_handle(self):
         """
         if self.read_write_handle is None:
             profile = self.netbox.get_preferred_snmp_management_profile(writeable=True)
-            self.read_write_handle = Snmp.Snmp(
+            self.read_write_handle = get_snmp_session_for_profile(profile)(
                 host=self.netbox.ip,
-                community=profile.snmp_community,
-                version=profile.snmp_version,
                 retries=self.retries,
                 timeout=self.timeout,
             )