added snmp in __init__.py

This commit is contained in:
alejandro 2023-08-18 10:41:39 +02:00
parent 66394f8ec4
commit a42a87dffe
2 changed files with 21 additions and 1 deletions

View File

@ -6,4 +6,5 @@ from .agents import *
from .modules import * from .modules import *
from .transfer import * from .transfer import *
from .discovery import * from .discovery import *
from .http import * from .http import *
from .snmp import *

View File

@ -20,7 +20,16 @@ _GLOBAL_VARIABLES = {
'remote_port' : 161, 'remote_port' : 161,
} }
####
# Creates an SNMP session based on the global configuration variables.
#########################################################################################
def create_snmp_session(): def create_snmp_session():
"""
Creates an SNMP session based on the global configuration variables.
Returns:
Session: An SNMP session configured based on the global variables.
"""
host = _GLOBAL_VARIABLES['hostname'] host = _GLOBAL_VARIABLES['hostname']
version = _GLOBAL_VARIABLES['version'] version = _GLOBAL_VARIABLES['version']
community = _GLOBAL_VARIABLES['community'] community = _GLOBAL_VARIABLES['community']
@ -67,6 +76,10 @@ def create_snmp_session():
return Session(**session_kwargs) return Session(**session_kwargs)
####
# Performs an SNMP GET operation to retrieve the value of a specified OID.
#########################################################################################
def snmp_get( def snmp_get(
session: Session, session: Session,
oid: str oid: str
@ -83,6 +96,9 @@ def snmp_get(
""" """
return session.get(oid) return session.get(oid)
####
# Performs an SNMP WALK operation to retrieve a list of values from a subtree of the MIB.
#########################################################################################
def snmp_walk( def snmp_walk(
session: Session, session: Session,
oid: str oid: str
@ -99,6 +115,9 @@ def snmp_walk(
""" """
return session.walk(oid) return session.walk(oid)
####
# Sends an SNMP trap to the specified destination IP using the given OID, value, and community.
#########################################################################################
def send_trap( def send_trap(
trap_oid: str, trap_oid: str,
trap_value: str, trap_value: str,