karesansui/lib/rrd/interface.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 #
   4 # This file is part of Karesansui Core.
   5 #
   6 # Copyright (C) 2010 HDE, Inc.
   7 #
   8 # This program is free software; you can redistribute it and/or
   9 # modify it under the terms of the GNU Lesser General Public
  10 # License as published by the Free Software Foundation; either
  11 # version 2.1 of the License, or (at your option) any later version.
  12 #
  13
  14 import re
  15 import datetime
  16 import rrdtool
  17 import karesansui
  18 from karesansui.lib.const import GRAPH_COMMON_PARAM, DEFAULT_LANGS
  19 from karesansui.lib.utils import is_readable, generate_phrase, copy_file
  20
  21 def is_interface_file_exist(rrd_dir, dev):
  22     ret = True
  23
  24     rrd_filepath = ("%s/interface/if_packets-%s.rrd" % (rrd_dir, dev),
  25                     "%s/interface/if_octets-%s.rrd" % (rrd_dir, dev),
  26                     "%s/interface/if_errors-%s.rrd" % (rrd_dir, dev),
  27                     )
  28
  29     for filepath in rrd_filepath:
  30         if is_readable(filepath) is False:
  31             ret = False
  32
  33     return ret
  34
  35 def create_interface_graph(_, lang, graph_dir, rrd_dir, start, end, dev, type):
  36     graph_filename = "%s.png" % (generate_phrase(12,'abcdefghijklmnopqrstuvwxyz'))
  37     graph_filepath = "%s/%s" % (graph_dir, graph_filename)
  38
  39     rrd_filepath = ("%s/interface/if_%s-%s.rrd" % (rrd_dir, type, dev),
  40                     )
  41
  42     graph_title = {
  43         "packets":"%s - packets",
  44         "octets":"%s - traffic",
  45         "errors":"%s - errors",
  46         }
  47
  48     graph_label = {
  49         "packets":"Packets / sec",
  50         "octets":"Octet / sec",
  51         "errors":"Packets / sec",
  52         }
  53
  54     for filepath in rrd_filepath:
  55         if is_readable(filepath) is False:
  56             return ""
  57
  58     legend_header_label = {"min":_('Min'),
  59                            "max":_('Max'),
  60                            "ave":_('Ave'),
  61                            "last":_('Last'),
  62                            }
  63
  64     for key in legend_header_label.keys():
  65         if re.search(u"[^a-zA-Z0-9]", legend_header_label[key]):
  66             legend_header_label[key] = "</tt>%s<tt>" % (legend_header_label[key].encode("utf-8"))
  67         else:
  68             legend_header_label[key] = "%s" % (legend_header_label[key].encode("utf-8"))
  69
  70     legend_header = "<tt>             %s        %s        %s        %s</tt>" % (legend_header_label['min'],
  71                                                                                 legend_header_label['max'],
  72                                                                                 legend_header_label['ave'],
  73                                                                                 legend_header_label['last']
  74                                                                                 )
  75
  76     legend_label = {"rx":_('RX'),
  77                     "tx":_('TX'),
  78                     }
  79     for key in legend_label.keys():
  80         if re.search(u"[^a-zA-Z0-9]", legend_label[key]):
  81             legend_label[key] = "</tt>%s<tt>" % (legend_label[key].encode("utf-8"))
  82         else:
  83             legend_label[key] = "%s" % (legend_label[key].encode("utf-8"))
  84
  85     title = "<tt>%s</tt>" % (graph_title[type] % (dev))
  86
  87     created_label = _('Graph created')
  88     if re.search(u"[^a-zA-Z0-9 ]", created_label):
  89         created_label = "</tt>%s<tt>" % (created_label.encode("utf-8"))
  90     else:
  91         created_label = "%s" % (created_label.encode("utf-8"))
  92
  93     created_time = "%s" % (datetime.datetime.today().strftime(DEFAULT_LANGS[lang]['DATE_FORMAT'][1]))
  94     created_time = re.sub(r':', '\:', created_time)
  95
  96     legend_footer = "<tt>%s \: %s</tt>" % (created_label, created_time)
  97
  98     data = rrdtool.graph(graph_filepath,
  99                          GRAPH_COMMON_PARAM,
 100                          "--title", title,
 101                          "--vertical-label", graph_label[type],
 102                          "--start", start,
 103                          "--end",  end,
 104                          "--legend-direction", "bottomup",
 105                          "DEF:rx=%s:rx:AVERAGE" % (rrd_filepath[0]),
 106                          "DEF:tx=%s:tx:AVERAGE" % (rrd_filepath[0]),
 107                          "COMMENT:%s\\r" % legend_footer,
 108                          "COMMENT:<tt>---------------------------------------------------------------------------</tt>\\n",
 109                          "AREA:rx#E7EF00:<tt>%s  </tt>" % (legend_label["rx"]),
 110                          "GPRINT:rx:MIN:<tt>%8.2lf%s</tt>",
 111                          "GPRINT:rx:MAX:<tt>%8.2lf%s</tt>",
 112                          "GPRINT:rx:AVERAGE:<tt>%8.2lf%s</tt>",
 113                          "GPRINT:rx:LAST:<tt>%8.2lf%s</tt>\\n",
 114                          "STACK:tx#80AA00:<tt>%s  </tt>" % (legend_label["tx"]),
 115                          "GPRINT:tx:MIN:<tt>%8.2lf%s</tt>",
 116                          "GPRINT:tx:MAX:<tt>%8.2lf%s</tt>",
 117                          "GPRINT:tx:AVERAGE:<tt>%8.2lf%s</tt>",
 118                          "GPRINT:tx:LAST:<tt>%8.2lf%s</tt>\\n",
 119                          "COMMENT:%s\\n" % (legend_header),
 120                          "COMMENT: \\n",
 121                          "LINE1:rx#E7EF00",
 122                          "STACK:tx#80AA00",
 123                          )
 124
 125     return graph_filepath