summaryrefslogtreecommitdiffstats
path: root/func/minion/modules/snmp.py
blob: b8033ee2b9303ea3dce0d4b0aac246fee84c0ff1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# Copyright 2007, Red Hat, Inc
# James Bowes <jbowes@redhat.com>
# Seth Vidal modified command.py to be snmp.py
#
# This software may be freely redistributed under the terms of the GNU
# general public license.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

"""
Abitrary command execution module for func.
"""

import func_module
from func.minion import sub_process
base_snmp_command = '/usr/bin/snmpget -v2c -Ov -OQ'

class Snmp(func_module.FuncModule):

    version = "0.0.1"
    api_version = "0.0.1"
    description = "SNMP related calls through func."

    def get(self, oid, rocommunity, hostname='localhost'):
        """
        Runs an snmpget on a specific oid returns the output of the call.
        """
        command = '%s -c %s %s %s' % (base_snmp_command, rocommunity, hostname, oid)
        
        cmdref = sub_process.Popen(command.split(), stdout=sub_process.PIPE, stderr=sub_process.PIPE, shell=False, close_fds=True)
        data = cmdref.communicate()
        return (cmdref.returncode, data[0], data[1])

    def register_method_args(self):
        """
        Implementing the argument getter
        """
        
        return {
                'get':{
                    'args':{
                        'oid':{
                            'type':'string',
                            'optional':False,
                            'description':'The oid'
                            },
                        'rocommunity':{
                            'type':'string',
                            'optional':False,
                            'description':"The rocommunity"
                            },
                        'hostname':{
                            'type':'string',
                            'optional':True,
                            'default':'localhost',
                            'description':"The hostname tobe apllied on"
                            }
                        },
                    'description':"Runs an snmpget on a specific oid returns the output of the call"
                    }
                }
    #def walk(self, oid, rocommunity):

    #def table(self, oid, rocommunity):