PLC/NodeTags.py

   1 #
   2 # Thierry Parmentelat - INRIA
   3 #
   4 from PLC.Faults import *
   5 from PLC.Parameter import Parameter
   6 from PLC.Filter import Filter
   7 from PLC.Table import Row, Table
   8 from PLC.Nodes import Node, Nodes
   9 from PLC.TagTypes import TagType, TagTypes
  10
  11 class NodeTag(Row):
  12     """
  13     Representation of a row in the node_tag.
  14     To use, instantiate with a dict of values.
  15     """
  16
  17     table_name = 'node_tag'
  18     primary_key = 'node_tag_id'
  19     fields = {
  20         'node_tag_id': Parameter(int, "Node tag identifier"),
  21         'node_id': Node.fields['node_id'],
  22         'hostname' : Node.fields['hostname'],
  23         'tag_type_id': TagType.fields['tag_type_id'],
  24         'value': Parameter(str, "Node tag value"),
  25         'tagname': TagType.fields['tagname'],
  26         'description': TagType.fields['description'],
  27         'category': TagType.fields['category'],
  28         }
  29
  30 class NodeTags(Table):
  31     """
  32     Representation of row(s) from the node_tag table in the
  33     database.
  34     """
  35
  36     def __init__(self, api, node_tag_filter = None, columns = None):
  37         Table.__init__(self, api, NodeTag, columns)
  38
  39         sql = "SELECT %s FROM view_node_tags WHERE True" % \
  40               ", ".join(self.columns)
  41
  42         if node_tag_filter is not None:
  43             if isinstance(node_tag_filter, (list, tuple, set, int, long)):
  44                 node_tag_filter = Filter(NodeTag.fields, {'node_tag_id': node_tag_filter})
  45             elif isinstance(node_tag_filter, dict):
  46                 node_tag_filter = Filter(NodeTag.fields, node_tag_filter)
  47             else:
  48                 raise PLCInvalidArgument, "Wrong node tag filter %r"%node_tag_filter
  49             sql += " AND (%s) %s" % node_tag_filter.sql(api)
  50
  51
  52         self.selectall(sql)