summaryrefslogtreecommitdiffstats
path: root/plugins/sadf.py
blob: 597b14c3f277a5fa3fd42186f886fc724b24e041 (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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# sadf.py -- nagios plugin uses sadf output for perf data
# Copyright (C) 2014 Red Hat Inc
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
#

from datetime import datetime, timedelta
from glusternagios import utils
import xml.etree.cElementTree as etree

if hasattr(etree, 'ParseError'):
    _etreeExceptions = (etree.ParseError, AttributeError, ValueError)
else:
    _etreeExceptions = (SyntaxError, AttributeError, ValueError)


class SadfException(Exception):
    message = "sadf exception"

    def __init__(self, rc=0, out=(), err=()):
        self.rc = rc
        self.out = out
        self.err = err

    def __str__(self):
        o = '\n'.join(self.out)
        e = '\n'.join(self.err)
        if o and e:
            m = o + '\n' + e
        else:
            m = o or e

        s = self.message
        if m:
            s += '\nerror: ' + m
        if self.rc:
            s += '\nreturn code: %s' % self.rc
        return s


class SadfCmdExecFailedException(SadfException):
    message = "sadf command failed"


class SadfXmlErrorException(SadfException):
    message = "XML error"


def sadfExecCmd(sadfCmd):
    try:
        (rc, out, err) = utils.execCmd(sadfCmd, raw=True)
    except (OSError, ValueError) as e:
        raise SadfCmdExecFailedException(err=[str(e)])

    if rc != 0:
        raise SadfCmdExecFailedException(rc, [out], [err])

    try:
        return etree.fromstring(out)
    except _etreeExceptions:
        raise SadfXmlErrorException(err=[out])


def utcnow():
    return datetime.utcnow()


def getLatestStat(root, interval=1):
    try:
        el = root.findall('host/statistics/timestamp')[-1]
    except (_etreeExceptions + (IndexError,)):
        raise SadfXmlErrorException(err=[etree.tostring(root)])

    d = utils.xml2dict(el)
    statTime = datetime.strptime("%s %s" % (d['timestamp']['date'],
                                            d['timestamp']['time']),
                                 "%Y-%m-%d %H:%M:%S")
    minutes = timedelta(minutes=interval)
    now = utcnow()
    if (now - statTime) <= minutes:
        return d['timestamp']
    else:
        return None


def add_common_args(parser):
    parser.add_argument("-t", "--interval", action="store",
                        type=int,
                        default=1,
                        help="Interval of time(min) older than"
                        " which sadf output in considered stale")